Atcoder AGC B Splatter Painting 记忆化+逆向模拟

来源：互联网发布：算法设计与分析沙特编辑：程序博客网时间：2024/06/04 19:02

点击打开链接

题意:n点,m边,q个操作 vi,di,ci 每次将与v距离d以内的点涂成c,问q次操作后n个定点颜色?
n,m,q<=1e5,d<=10 首先操作i,j,i<j 若操作i有部分与j重叠,则重叠部分无效
//逆向考虑,从编号大的操作开始,每个点u最终颜色由最后一次对u上色的操作决定 if(!a[u]) a[u]=c,记忆化操作fun(u,d),如果fun(u,d)被执行,编号小的fun(u,d)就不用再执行了,
fun(u,d)=fun(v,d-1)(v是u的邻点).时间复杂度O(Q+M*d)

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=2e5+20;int n,m,a[N];vector<int> e[N];int vis[N];int f[N][15];struct node{int v,d,c;}q[N];void fun(int u,int d,int c){if(f[u][d]!=-1)return;if(!a[u])//已经被编号大的上过色 a[u]=c;if(d==0)//边界 return;for(int i=0;i<e[u].size();i++){int v=e[u][i];fun(v,d-1,c);}f[u][d]=1;}int main(){while(cin>>n>>m){int u,v;for(int i=1;i<=n;i++)e[i].clear();memset(a,0,sizeof(a));memset(f,-1,sizeof(f));for(int i=0;i<m;i++){scanf("%d%d",&u,&v);e[u].push_back(v);e[v].push_back(u);}int Q;cin>>Q;for(int i=0;i<Q;i++)scanf("%d%d%d",&q[i].v,&q[i].d,&q[i].c);for(int i=Q-1;i>=0;i--){fun(q[i].v,q[i].d,q[i].c);}for(int i=1;i<=n;i++)cout<<a[i]<<endl;}return 0;}

0 0