题意:
给你一个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<cstdio>
#include<cstring>
#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]);
}