[codeforces375D]Tree and Queries（dsu on the tree+bit）

题目：

我是超链接

题意：

统计以v为顶点的子树中颜色出现大于等于k次的种类数

题解：

dsu on the tree！
先把询问记录下来—-这里的询问记录很多都用虚树呢
这里因为是至少k次，用bit就很好了，这里对于bit的处理十分巧妙，记录出现多少次的颜色种类

add操作
if (cnt[a[x]]) change(cnt[a[x]],-1);
cnt[a[x]]+=vv;
if (cnt[a[x]]) change(cnt[a[x]],1);

没有严格的父子关系就不要连单向边啊

代码：

#include 
#include 
#include 
#define LL long long
#define N 100005 
#define inf 1e5
using namespace std;
int tot,nxt[N*2],point[N],v[N*2],size[N],tot1,nxt1[N*2],point1[N],v1[N*2],id[N],h[N],son[N];
int cnt[N],Son,c[N],a[N],ans[N];
void addline(int x,int y){
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void addline1(int x,int y,int i){++tot1; nxt1[tot1]=point1[x]; point1[x]=tot1; v1[tot1]=y; id[tot1]=i;}
void getson(int x,int fa)
{
    size[x]=1;
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa)
      {
        getson(v[i],x);
        if (size[v[i]]>size[son[x]]) son[x]=v[i];
        size[x]+=size[v[i]];
      }
}
void change(int loc,int vv)
{
    for (int i=loc;i<=inf;i+=i&(-i))
      c[i]+=vv;
}
int ask(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
      ans+=c[i];
    return ans;
}
void add(int x,int fa,int vv)
{
    if (cnt[a[x]]) change(cnt[a[x]],-1);
    cnt[a[x]]+=vv;
    if (cnt[a[x]]) change(cnt[a[x]],1);
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa && v[i]!=Son) add(v[i],x,vv);
}
void dfs(int x,int fa,int k)
{
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa && v[i]!=son[x]) dfs(v[i],x,0); 
    if (son[x]) dfs(son[x],x,1),Son=son[x];
    add(x,fa,1);   Son=0;
    for (int i=point1[x];i;i=nxt1[i])
    {
        int xx=ask(inf),yy=ask(v1[i]-1);
        ans[id[i]]=xx-yy; 
    }
    if (!k) add(x,fa,-1);
}
int main()
{
    int n,i,m;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++) scanf("%d",&a[i]);
    for (i=1;iint x,y;
        scanf("%d%d",&x,&y);
        addline(x,y);
    }
    getson(1,0);
    for (i=1;i<=m;i++)
    {
        int vv,hh;
        scanf("%d%d",&vv,&hh); 
        addline1(vv,hh,i);
    }
    dfs(1,0,0);
    for (i=1;i<=m;i++) printf("%d\n",ans[i]);
}