【AtCoder】【思維】【圖論】Splatter Painting(AGC012)

題意:

有一個含有n個點的無向圖,所有的點最初顏色均爲0。有q次操作,每次操作將v[i]周圍的距離小於等於d[i]的點全部都染成顏色c[i]。最後輸出每個點的最終的顏色。

數據範圍:

1<=n,m,q<=10^5
0<=d[i]<=10
1<=c[i]<=10^5

思路:

看見1<=d[i]<=10,這個條件,第一反應當然是暴力啦。但是如果從一個點總是能夠訪問所有的節點,那麼這就變成O(n^2)了。那麼我們應當考慮時間複雜度更加穩定的算法。
然後開始考慮如何優化。倒着掃操作是很容易想到的。然後可以對於每一個點維護一個對於當前已經掃完的操作的最大值。假如說當前在點v,然後當前的d爲d[i],假如說d[i]<=maxd[v],那麼就說明在後面的操作中將當前這一次操作所產生的效果抵消了。於是就可以直接返回了。
經過上述的優化之後,我們發現能夠進入一個點併成功進行拓展的條件是d[i]>maxd[v],那麼因爲1<=d[i]<=10,所以說就算d[i]從1~10依次排列,也只會對於v點訪問最多10次。這樣子時間複雜度就變爲了穩定的O(10*n+m),從而穩當了不少。

代碼:

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 100000
using namespace std;
struct node
{
	int to;
	node *nxt;
}edges[MAXN*2+5];
node *ncnt=&edges[0],*Adj[MAXN+5];
int n,m,q,col[MAXN+5],maxd[MAXN+5];
int V[MAXN+5],D[MAXN+5],C[MAXN+5];
void Init()
{
	memset(maxd,-1,sizeof(maxd));
}
void AddEdge(int u,int v)
{
	node *p=++ncnt;
	p->to=v;
	p->nxt=Adj[u];
	Adj[u]=p;
	
	node *q=++ncnt;
	q->to=u;
	q->nxt=Adj[v];
	Adj[v]=q;
}
void DFS(int u,int d,int c)
{
	if(col[u]==0)//沒有賦過值才賦值
		col[u]=c;
	if(maxd[u]>=d)//判斷當前操作是否被後面的操作覆蓋了
		return;
	if(d==0)//到達能夠賦值的邊界了
		return;
	maxd[u]=d;
	for(node *p=Adj[u];p!=NULL;p=p->nxt)
	{
		int v=p->to;
		DFS(v,d-1,c);//d--,繼續賦值
	}
}
int main()
{
	Init();
	scanf("%d %d",&n,&m);
	int u,v;
	for(int i=1;i<=m;i++)
	{
		scanf("%d %d",&u,&v);
		AddEdge(u,v);
	}
	scanf("%d",&q);
	for(int i=1;i<=q;i++)
		scanf("%d %d %d",&V[i],&D[i],&C[i]);
	for(int i=q;i>=1;i--)//倒着處理
		DFS(V[i],D[i],C[i]);
	for(int i=1;i<=n;i++)
		printf("%d\n",col[i]);
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章