AGC012B Splatter Painting

传送门

题意:
给你一个n个节点的简单无向图(无重边、自环),含有m条边。现在有q次操作。初始的时候所有点的颜色都为0。
每一次操作的描述如下:选择一个节点v[i],将与这个节点的距离小于等于d[i]的点全部都染成颜色c[i]。
现在要求你回答,完成这q次操作之后,每一个点的最终颜色是什么。

题解:
这道题d很小(≤10),可以暴力。但是直接暴力有可能会T(举个例子:菊花图)。
考虑将操作倒过来,那么就成了:将与这个节点的距离小于等于d[i]且还没有染色的点全部都染成颜色c[i]。
维护一个maxd[u]为u点操作到的最大的距离,那么当当前距离小于等于maxd[u]时,说明这次操作会被后面的操作抵消,可以忽略。
那么因为每个点最多只会被访问10次,所以时间复杂度为 O ( n d ) O(nd) O(nd)
注意这里不仅仅是起点要判断,访问到的点也要判断剩余距离和maxd的关系。

代码:

#include
#include
#define INF 0x3f3f3f3f
#define maxn 100005
int n,m,q,s[maxn],d[maxn],x[maxn],maxd[maxn],c[maxn],que[maxn],fr,bk;
struct node { int v; node *nxt; } edge[maxn*2],*head[maxn],*ncnt;
void addedge(int u,int v)
{
	ncnt++;
	ncnt->v=v,ncnt->nxt=head[u];
	head[u]=ncnt;
}
void dfs(int u,int d,int x)
{
	if(maxd[u]>=d) return;
	if(!c[u]) c[u]=x;
	if(!d) return;
	maxd[u]=d;
	for(node *p=head[u];p;p=p->nxt) dfs(p->v,d-1,x);
}
int main()
{
	scanf("%d%d",&n,&m);
	ncnt=&edge[0];
	for(int i=1;i<=m;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		addedge(u,v); addedge(v,u);
	}
	memset(maxd,-1,sizeof(maxd));
	scanf("%d",&q);
	for(int i=1;i<=q;i++) scanf("%d%d%d",&s[i],&d[i],&x[i]);
	for(int i=q;i;i--) dfs(s[i],d[i],x[i]); 
	for(int i=1;i<=n;i++) printf("%d\n",c[i]);
}

你可能感兴趣的:(懵逼了半天终于AC,搜索,暴力)