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

点击打开链接

题意: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;
}