洛谷-Tree and Queries-(树上启发式+树状数组)

DIV1-D

题意：
就是给你一个树，然后每个节点有一个颜色。给你m次询问，每次问你以a点的子树中，出现次数>=k的颜色有多少种。

思考：
不管别的，先考虑一下，怎么快速查询>=k的颜色有多少种。这是以前我遇到过的问题，一个数组，每次可以让一个点加1，一个点减1，最后问你整个数组元素>0的个数。其实，可以这样做，不直接去维护。而是先用一个数组，看看这个颜色出现了多少次，树状数组维护这个次数就行了。但是，对于刚才提的问题，如果出现负值就不行了，因为不能维护负值。然后呢，对于comb也就是合并的地方，只会进行log次，也就是再加上一个log也没事。所以没炸时间。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;

int T,n,m,k;
int va[N];
int siz[N],son[N];
int cnt[N],tr[N],R = 1e5+5;
int anw[N];

vector<PII > v[N];
vector<int > e[N];

void get(int now,int p)
{
	siz[now] = 1;
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		get(spot,now);
		siz[now] += siz[spot];
		if(siz[spot]>siz[son[now]]) son[now] = spot;
	}
}

int bit(int x)
{
	return x&(-x);
}

void update(int x,int value)
{
	while(x)
	{
		tr[x] += value;
		x -= bit(x);
	}	
}

int query(int x)
{
	int sum = 0;
	while(x<=R)
	{
		sum += tr[x];
		x += bit(x);
	}
	return sum;
}

void add(int now,int p,int value,int wson)
{
	update(cnt[va[now]],-1);
	cnt[va[now]] += value;
	update(cnt[va[now]],1);
	for(auto spot:e[now])
	{
		if(spot==p||spot==wson) continue;
		add(spot,now,value,wson);
	}
}

void dfs(int now,int p,int del)
{
	for(auto spot:e[now])
	{
		if(spot==p||spot==son[now]) continue;
		dfs(spot,now,1);
	}
	if(son[now]) dfs(son[now],now,0);
	add(now,p,1,son[now]);
	for(auto t:v[now]) anw[t.se] = query(t.fi);
	if(del) add(now,p,-1,0);
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	for(int i=1;i<=m;i++)
	{
		int a,b;
		cin>>a>>b;
		v[a].pb({b,i});
	}
	get(1,0);dfs(1,0,1);
	for(int i=1;i<=m;i++) cout<<anw[i]<<"\n";
	return 0;
}

总结：
多多思考，一个算法的本质，一个算法学了一段时间之后一定要多思考思考。