0%

颜色统计

给出一棵 n 个结点的树,每个结点有一个颜色c_i。 询问 q 次,每次询问以 v 结点为根的子树中,出现次数 ≥k 的颜色有多少种。树的根节点是1。

输入描述:

一行两个整数n和m。

第二行n个整数c_i,表示每个节点的颜色。

接下来n-1行,每行两个整数,表示一条边。

接下来m行,每行两个整数v和k,表示一个询问。

1<=n,m,ci<=10^5

输出描述:

对于每个询问,输出一个结果。

示例1
输入

8 5
1 2 2 3 3 2 3 3
1 2
1 5
2 3
2 4
5 6
5 7
5 8
1 2
1 3
1 4
2 3
5 3

输出

2
2
1
0
1

yansetongji.cppview raw
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#include<stdio.h>
#include<iostream>
#include<vector>
#include<algorithm>
#define M 100005
using namespace std;
struct Query{
int id,c;
};
vector<Query>q[M];
vector<int>edge[M];
int col[M],cnt[M],Res[M];
int sz[M],Big[M],ans[M];
int Lt[M],Rt[M],dfsLine[M],tid=0;
void Rd(int &res){
char c;res=0;
while(c=getchar(),c<'0');
do res=(res<<3)+(res<<1)+(c^48);
while(c=getchar(),c>='0');
}
void dfs_pre(int x,int f){
sz[x]=1;Big[x]=0;
Lt[x]=++tid;dfsLine[tid]=x;
for(int i=0;i<edge[x].size();i++){
int y=edge[x][i];
if(y==f)continue;
dfs_pre(y,x);
sz[x]+=sz[y];
}
Rt[x]=tid;
}
void Del(int x){
for(int i=Lt[x];i<=Rt[x];i++){
int t=dfsLine[i];
--Res[cnt[col[t]]--];
}

}
void Add(int x){
for(int i=Lt[x];i<=Rt[x];i++) {
int t=dfsLine[i];
++Res[++cnt[col[t]]];
}
}
void solve(int x,int f){
int big=-1;
for(int i=0;i<edge[x].size();i++){
int y=edge[x][i];
if(y==f)continue;
if(big==-1||sz[y]>sz[big])big=y;
}
for(int i=0;i<edge[x].size();i++){
int y=edge[x][i];
if(y!=f&&y!=big){
solve(y,x);Del(y);
}
}
if(big!=-1)solve(big,x);

for(int i=0;i<edge[x].size();i++){
int y=edge[x][i];
if(y!=f&&y!=big)Add(y);
}
++Res[++cnt[col[x]]];

for(int i=0;i<q[x].size();i++)
ans[q[x][i].id]=Res[q[x][i].c];
}
void Prt(int x){
if(x==0)return;
Prt(x/10); putchar(x%10^48);
}
int main(){
int n,m,a,b;
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++){
edge[i].clear();
scanf("%d",&col[i]);
}
for(int i=1;i<n;i++){
scanf("%d %d",&a,&b);
edge[a].push_back(b);
edge[b].push_back(a);
}
for(int i=1;i<=m;i++){
scanf("%d %d",&a,&b);
q[a].push_back((Query){i,b});
}
dfs_pre(1,0);
solve(1,0);
for(int i=1;i<=m;i++){
if(ans[i]==0)putchar('0');
Prt(ans[i]);
putchar('\n');
}
return 0;
}