[Coderforces600E] Lomsat gelral

SD_le發表於2017-06-18

大意是每個點有一個顏色,詢問以每個點為根的子樹中出現次數最多的顏色是誰,如果有多個輸出編號和。

這個東西用資料結構是不太好做的,考慮優化暴力。

首先最裸的的暴力就是在dfs序上跑莫隊,但因為是樹的結構所以有更優美的暴力。

參考的部落格:http://www.cnblogs.com/candy99/p/dsuontree.html

類似樹上啟發式合併,叫做dsu on tree。

0.首先輕重鏈剖分。

1.然後先處理輕兒子,處理完把貢獻刪除。

2.處理重兒子,保留貢獻。

3.把輕兒子的貢獻再加上。

4.算這個點的答案。

5.如果這個點是輕兒子就把陣列裡的東西清空。

因為每個點到根只有$log$條輕鏈,所以在執行3和5時每個點最多加入刪除$log$次。

複雜度$nlogn$

#include<bits/stdc++.h>
#define ll long long
#define N 100005
using namespace std;
int v[N],head[N],ver[N*2],nxt[N*2],tot;
void add(int a,int b)
{
	tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;return;
}
int n,size[N],son[N];
void dfs(int x,int f)
{
	size[x]=1;
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f)continue;
		dfs(ver[i],x);
		size[x]+=size[ver[i]];
		if(size[ver[i]]>size[son[x]])son[x]=ver[i];
	}
}
int cnt[N];
ll sum[N],ans[N];
int mx;
void ds(int x,int f,int z)
{
	sum[cnt[v[x]]]-=v[x];
	cnt[v[x]]+=z;
	sum[cnt[v[x]]]+=v[x];
	if(cnt[v[x]]>mx)mx=cnt[v[x]];
	else if(sum[mx]==0)mx--;
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f)continue;
		ds(ver[i],x,z);
	}
}
void dffs(int x,int f,int op)
{
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f||ver[i]==son[x])continue;
		dffs(ver[i],x,0);
	}
	if(son[x])dffs(son[x],x,1);
	sum[cnt[v[x]]]-=v[x];
	cnt[v[x]]++;
	sum[cnt[v[x]]]+=v[x];
	if(cnt[v[x]]>mx)mx=cnt[v[x]];
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f||ver[i]==son[x])continue;
		ds(ver[i],x,1);
	}
	ans[x]=sum[mx];
	if(!op)ds(x,f,-1);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&v[i]);
	int t1,t2;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&t1,&t2);
		add(t1,t2);add(t2,t1);
	}
	dfs(1,-1);
	dffs(1,-1,0);
	for(int i=1;i<=n;i++)
	{
		printf("%I64d ",ans[i]);
	}
	return 0;
}