cf#221-div1-D - Tree and Queries-dfs序+莫队算法

http://codeforces.com/contest/375/problem/D

给一棵有根树，每个节点有个颜色，每次询问 v,k，即v节点为根的子树中，有多少种颜色，出现的次数超过k。

对于查询v的子树，先转为dfs序。

对于出现次数超过k的答案，我们另外用一个数组维护，即当color1个数增加时，那么num[ color1_num ]++, 

减少时则 num[ color1_num ]--

这样我们就能直接得到  num[k]为出现次数超过k次的颜色种类。

这样的话，可以发现对于区间【l,r】的转移符合使用莫队算法的条件，则可以用莫队算法，分块，区间排序。

复杂度 （q+n）sqrtn  跑了233ms

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;
#define N 100100
typedef long long ll;
int a
, cnt
;
ll ans
, res;
int L, R;
vector< int > mp
;
int in
;
int out
;
int id=0;
int col
;
int who
;
int vis
;
int num
;
void dfs(int x)
{
int i;
in[x]=++id;
who[id]=x;
vis[x]=1;
for (i=0; i<mp[x].size(); i++)
{
int v=mp[x][i];
if (!vis[v])
dfs(v);
}
out[x]=id;
}
struct node
{
int x, y, l, p,k;
} q
;
bool cmp(const node &x, const node &y)
{
if (x.l == y.l) return x.y < y.y;
return x.l < y.l;
}
void query(int x, int y, int flag )
{
if (flag)
{
for (int i=x; i<L; i++)
{
cnt[ col[who[i]] ]++;
num[cnt[ col[who[i]] ]]++;
}
for (int i=L; i<x; i++)
{
num[cnt[ col[who[i]] ]]--;
cnt[ col[who[i]] ]--;
}
for (int i=y+1; i<=R; i++)
{
num[cnt[ col[who[i]] ]]--;
cnt[ col[who[i]] ]--;
}
for (int i=R+1; i<=y; i++)
{
cnt[ col[who[i]] ]++;
num[cnt[ col[who[i]] ]]++;
}
}
else
{
for (int i=x; i<=y; i++)
{
cnt[ col[who[i]] ]++;
num[cnt[ col[who[i]] ]]++;
}
}
L = x, R = y;
}
int main()
{
int n, m;

scanf("%d%d", &n, &m);
int x,y;
for (int i=1; i<=n ; i++)
scanf("%d",&col[i]);
for (int i=1; i<=n-1; i++)
{
scanf("%d%d", &x,&y);
mp[x].push_back(y);
mp[y].push_back(x);
}
dfs(1);

int block_size = sqrt(1.0*n);

for (int i=0; i<m; i++)
{
scanf("%d%d", &x,&y);
q[i].x=in[x];
q[i].y=out[x];
q[i].l = q[i].x / block_size;
q[i].p = i;
q[i].k=y;
}
sort(q, q+m, cmp);

res = 0;
for (int i=0; i<m; i++)
{
query(q[i].x, q[i].y, i);
ans[q[i].p] = num[q[i].k];
}

for (int i=0; i<m; i++) printf("%I64d\n", ans[i]);

return 0;
}