Boring counting HDU - 4358 dfs把树在区间内展开+莫队或者启发式合并 memset会RE

题目链接：https://vjudge.net/problem/HDU-4358
题意：给一棵树及其节点权重，每次询问以某个节点为根的子树内有多少权值正好出现了k次。
莫队
转自：https://blog.csdn.net/gatevin/article/details/47264181
用dfs把树转化为区间询问，然后莫队。
一个问题就是输入时树的节点编号必须是连续并且从小到大的，否则可能dfs序的区间和输入权值对应不起来。如：
10
7 1
1 1 1 2 3 1 1
1 3
2 6
2 7
3 4
3 5
2 1
2
2
3

#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=1e5+5;
vector <int> v,g[maxn];
int a[maxn],vis[maxn],L[maxn],R[maxn],time,sum[maxn],n,k,ans,res[maxn];
struct NODE
{
    int l,r,block,id;
    NODE(int l=0,int r=0,int id=0,int block=0):l(l),r(r),id(id),block(block)
    {

    }
    bool operator <(NODE s)const
    {
        return block==s.block?r<s.r:block<s.block;
    }
} node[maxn];
void dfs(int cur,int fa)//dfs把树展开
{
    time++;
    L[cur]=time;
    vis[cur]=1;
    for(int i=0; i<g[cur].size(); i++)
    {
        if(vis[g[cur][i]])
            continue;
        dfs(g[cur][i],cur);
    }
    R[cur]=time;
}
void work(int x,int y)
{
    if(sum[x]==k)
        ans--;
    sum[x]+=y;
    if(sum[x]==k)
        ans++;
}
int main()
{
    int T,kase=0;
    scanf("%d",&T);
    while(T--)
    {
        time=0;
        v.clear();
        memset(vis,0,sizeof(vis));
        memset(sum,0,sizeof(sum));
        if(kase++)
            printf("\n");
        scanf("%d%d",&n,&k);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            v.push_back(a[i]);
        }
        sort(v.begin(),v.end());
        for(int i=1; i<=n; i++)//离散化
        {
            a[i]=lower_bound(v.begin(),v.end(),a[i])-v.begin();
        }
        for(int i=1; i<=n; i++)
        {
            g[i].clear();
        }
        for(int i=1; i<n; i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            g[y].push_back(x);
        }
        dfs(1,-1);
        int q;
        scanf("%d",&q);
        int u=sqrt(n);
        for(int i=1; i<=q; i++)
        {
            int x;
            scanf("%d",&x);
            node[i]=NODE(L[x],R[x],i,L[x]/u);
        }
        ans=0;
        sort(node+1,node+1+q);
        int l=1,r=0;
        for(int i=1; i<=q; i++)
        {
            while(r<node[i].r)
            {
                work(a[++r],1);
            }
            while(r>node[i].r)
            {
                work(a[r--],-1);
            }
            while(l<node[i].l)
            {
                work(a[l++],-1);
            }
            while(l>node[i].l)
            {
                work(a[--l],1);
            }
            res[node[i].id]=ans;
        }
        printf("Case #%d:\n",kase);
        for(int i=1; i<=q; i++)
        {
            printf("%d\n",res[i]);
        }
    }
}

树上启发式合并
没有莫队输入方式的限制。
注意初始化时不能用memset，会RE。可能是缓冲区爆了。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=1e5+5;
int n,k,col[maxn],son[maxn],sz[maxn],cnt[maxn],SON,q[maxn];
int sum=0,ans[maxn];
vector <int> v[maxn],tep;
void dfs1(int x,int fa)//树链剖分
{
    sz[x]=1;
    for(int i=0; i<v[x].size(); i++)
    {
        int to=v[x][i];
        if(fa==to)
            continue;
        dfs1(to,x);
        sz[x]+=sz[to];
        if(sz[to]>sz[son[x]])
            son[x]=to;
    }
}
void add(int x,int fa,int val)//统计贡献
{
    if(cnt[col[x]]==k)
        sum--;
    cnt[col[x]]+=val;
    if(cnt[col[x]]==k)
    {
        sum++;
    }
    for(int i=0; i<v[x].size(); i++)
    {
        int to=v[x][i];
        if(to==fa||to==SON)
            continue;
        add(to,x,val);
    }
}
void dfs2(int x,int fa,int o)//o=0代表清空 o=1不需要清空
{
    for(int i=0; i<v[x].size(); i++) //先遍历轻儿子
    {
        int to=v[x][i];
        if(to==fa)
            continue;
        if(to!=son[x])
            dfs2(to,x,0);
    }
    if(son[x])//遍历重儿子
    {
        dfs2(son[x],x,1);
        SON=son[x];
    }
    add(x,fa,1);
    SON=0;
    ans[x]=sum;
    if(!o)//清空数组
    {
        add(x,fa,-1);
    }
}
void init()
{
    sum=0;
    tep.clear();
    for(int i=0; i<=n; i++)
        v[i].clear();
    for(int i=0; i<=n; i++)
        cnt[i]=son[i]=0;
}
int main()
{
    int T;
    scanf("%d",&T);
    int kase=0;
    while(T--)
    {
        if(kase++)
            printf("\n");
        scanf("%d%d",&n,&k);
        init();
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&col[i]);
            tep.push_back(col[i]);
        }
        sort(tep.begin(),tep.end());
        for(int i=1; i<=n; i++)
        {
            col[i]=lower_bound(tep.begin(),tep.end(),col[i])-tep.begin()+1;
        }
        for(int i=1; i<n; i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            v[x].push_back(y);
            v[y].push_back(x);
        }
        int x;
        scanf("%d",&x);
        for(int i=1; i<=x; i++)
        {
            scanf("%d",&q[i]);
        }
        dfs1(1,0);
        dfs2(1,0,0);
        printf("Case #%d:\n",kase);
        for(int i=1; i<=x; i++)
        {
            printf("%d\n",ans[q[i]]);
        }
    }
    return 0;
}