CF375D-Tree and Queries(莫队+按次数分块)

题目
题意：给出一棵 n 个结点的树，每个结点有一个颜色 c i 。 询问 q 次，每次询问以 v 结点为根的子树中，出现次数 ≥k 的颜色有多少种。树的根节点是1。
数据范围: 2<=n<=1e5 1<=m<=1e5 1<=a[i]<=1e5;

思路：每次询问k值不一样，必须除了莫队还要有一个数据结构维护次数，也要用一个cnt[]数组维护。每次修改很多，查询只有m次。故用分块。
写完20分钟 调试2小时 T_T 一定要思路清晰!!! 要自杀了

#include
#include
#include
#include
#define m(a,b) memset(a,b,sizeof a)
#define foru(i,a,b) for(int i=a;i<=b;++i)
#define en '\n'
using namespace std;
typedef long long ll;
const int N=1e5+5,M=1e5+5;
template<class T>void rd(T &x)
{
    x=0;int f=0;char ch=getchar();
    while(ch<'0'||ch>'9')  {f|=(ch=='-');ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
    x=f?-x:x;
    return;
}
struct Edge{int to,nex;}edge[N<<1];
int head[N],tot,sz,in[N],out[N],vis[N],fp[N];
void add(int from,int to)
{
    edge[++tot]=(Edge){to,head[from]};head[from]=tot;
    edge[++tot]=(Edge){from,head[to]};head[to]=tot;
}
void dfs(int x)
{
    vis[x]=1,in[x]=sz++,fp[in[x]]=x;
    for(int i=head[x];i;i=edge[i].nex)
        if(!vis[edge[i].to]) dfs(edge[i].to);
    out[x]=sz-1;
}
int tag[N],sum[320],bl[N],block,n;
void update(int x,int val)
{
    tag[x]+=val;
    sum[bl[x]]+=val;
}
int query(int x)
{
    int ans=0;
    for(int i=x;i<=min(bl[x]*block,n);i++)
        if(tag[i]) ans+=tag[i];//我是脑子进水了吗???我为什么要ans++,明明是ans+=tag[i].
    for(int i=bl[x]+1;i<=bl[n];i++)
        ans+=sum[i];
    return ans;
}
struct Query{int id,l,r,v;}q[M];
int dexl,dexr,ans[M],cnt[N];
int a[N];
int cmp(Query x,Query y)
{
    if(x.l/block==y.l/block)
        return x.r<y.r;
    return x.l/block<y.l/block;
}
void add(int x){
    x=a[fp[x]];//x是颜色,从dfs序中转化到实际序列里.
    if(cnt[x])
        update(cnt[x],-1);//将该次数在按次数的分块里修改
    ++cnt[x];
    update(cnt[x],1);
}
void del(int x){
    x=a[fp[x]];
    update(cnt[x],-1);
    --cnt[x];
    if(cnt[x])
        update(cnt[x],1);
}
int main()
{
    int m;rd(n),rd(m);block=sqrt(n);
    foru(i,1,n)
        rd(a[i]),bl[i]=(i-1)/block+1;
    foru(i,1,n-1)
    {
        int x,y;rd(x),rd(y);
        add(x,y);
    }
    sz=1,dfs(1),--sz;
    foru(i,1,m)
    {
        int x,v;rd(x),rd(v);
        q[i]=(Query){i,in[x],out[x],v};
    }
    sort(q+1,q+m+1,cmp);
    dexl=1,dexr=0;
    foru(i,1,m)
    {
        while(dexr<q[i].r) add(++dexr);
        while(dexl>q[i].l) add(--dexl);
        while(dexr>q[i].r) del(dexr--);
        while(dexl<q[i].l) del(dexl++);
        if(q[i].v>n)//一定一定要注意!!!因为x=q[i].v>n时:bl[x]此时无值,虽说不会RE，但是多循环了for(int i=bl[x]+1;i<=bl[n];i++) 导致结果错误。
            ans[q[i].id]=0;
        else
            ans[q[i].id]=query(q[i].v);
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}