//树状数组区间更新
//树形结构->线性结构DFS处理
//求区间有K个相同的有多少数字
//每个节点表示到i的区间有几个K相同值
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=100005;
int B[maxn];
int val[maxn];
int vx[maxn];
int L[maxn];
int map[maxn];
int R[maxn];
int ans[maxn];
int n,m;
int tot;
void update(int id,int val){
for(int i=id;i<maxn;i+=(i&-i))
B[i]+=val;
}
int sum(int id){
int ret=0;
for(int i=id;i>0;i-=(i&-i))
ret+=B[i];
return ret;
}
vector<int >edge[maxn];
vector<int >len[maxn];
struct Q{
int id;
int r,l;
bool operator < (const Q &t )const {
return r<t.r;
}
}p[maxn];
void dfs(int u,int f){
int sz=edge[u].size();
map[tot]=u;
L[u]=tot++;
for(int i=0;i<sz;i++)
{
int v=edge[u][i];
if(v==f)continue;
dfs(v,u);
}
R[u]=tot;
}
int main(){
int T,q;
int i,j;
int u,v;
scanf("%d",&T);
for(int cas=1;cas<=T;cas++){
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++){
scanf("%d",&val[i]);
vx[i-1]=val[i];
}
sort(vx,vx+n);
int cnt=unique(vx,vx+n)-vx;
for(i=1;i<=n;i++) val[i]=lower_bound(vx,vx+cnt,val[i])-vx;
for(i=0;i<=n;i++)edge[i].clear(),len[i].clear();
for(i=1;i<n;i++){
scanf("%d%d",&u,&v);
edge[u].push_back(v);
edge[v].push_back(u);
}
tot=1;
dfs(1,-1);
scanf("%d",&q);
for(i=0;i<q;i++){
scanf("%d",&u);
p[i].id=i;
p[i].r=R[u];
p[i].l=L[u];
}
for(i=0;i<cnt;i++)
len[i].push_back(0);
memset(B,0,sizeof(B));
sort(p,p+q);
for(i=1,j=0;i<tot;i++){
int t=val[map[i]];
len[t].push_back(i);
int sz=len[t].size()-1;
if(sz>=m){
update(len[t][sz-m]+1,1);
update(len[t][sz-m+1]+1,-1);
}
if(sz>m){
update(len[t][sz-m-1]+1,-1);
update(len[t][sz-m]+1,1);
}
while(j<q && p[j].r-1==i){
ans[p[j].id]=sum(p[j].l);
j++;
}
}
if(cas!=1)puts("");
printf("Case #%d:\n",cas);
for(i=0;i<q;i++)
printf("%d\n",ans[i]);
}
return 0;
}
HDU Boring counting 树状数组
最新推荐文章于 2019-09-03 10:55:09 发布