bzoj3757 苹果树

　　莫队算法，先处理dfs序，然后将询问进行一下处理，假设询问x,y两点，x的的dfs序区间是[xl,xr],y的dfs序是[yl,yr]，那么这两个区间如果包含，那么询问区间[min(xl,yl),max(xl,yl)]，否则询问区间[min(xr,yr),max(xl,yl)]。对于算法当前处理的一个区间的删除插入操作，假设要判断一个点x是否出现在当前区间维护的路径中，如果其dfs区间被当前区间包含或者其记录的出现次数为0，说明该点其实是不存在的，否则存在。

　　代码

　　

#include<cstdio>
#include<algorithm>
#define N 2000010
using namespace std;
int dp,p[N],pre[N],tt[N],l[N],r[N],tot,o[N];
int L,R,ans[N],sum[N],cnt,flag1[N],flag2[N],flag3[N],colsum[N];
int n,m,i,col[N],a,b;
struct g{
    int l,r,ca,cb,id;
}q[N];
void link(int x,int y)
{
    dp++;pre[dp]=p[x];p[x]=dp;tt[dp]=y;
}
void dfs(int x,int fa)
{
    int i=p[x];
    l[x]=++tot;
    o[tot]=x;
    while (i)
    {
        if (tt[i]!=fa)
        {
            dfs(tt[i],x);
            o[++tot]=x;
        }
        i=pre[i];
    }
    if (tot==l[x]) o[++tot]=x;
    r[x]=tot;
}
void cc(int x,int w)
{
    int t=o[x];
    if (t==0) return;
    sum[t]+=w;
    if (x==l[t]) if (w==-1) flag1[t]=0;else flag1[t]=1;
    if (x==r[t]) if (w==-1) flag2[t]=0;else flag2[t]=1;
    if ((sum[t])&&((flag1[t]&&flag2[t])==0))
    {
        if (!flag3[t])
        {
            if (colsum[col[t]]==0) cnt++;
            colsum[col[t]]++;
        }
        flag3[t]=1;
    }
    else
    {
        if (flag3[t])
        {
            if (colsum[col[t]]==1) cnt--;
            colsum[col[t]]--;
        }
        flag3[t]=0;
    }
}
bool cmp(g a,g b)
{
    if (a.l/500==b.l/500)
    return a.r<b.r;
    return a.l/500<b.l/500;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++)
        scanf("%d",&col[i]);
    for (i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        link(a,b);link(b,a);
    }
    dfs(0,-1);
    for (i=1;i<=m;i++)
    {
        scanf("%d%d%d%d",&q[i].l,&q[i].r,&q[i].ca,&q[i].cb);q[i].id=i;
        int ta=q[i].l;int tb=q[i].r;
        if (r[ta]<l[tb]) q[i].l=r[ta],q[i].r=l[tb];
        else
        if (r[tb]<l[ta]) q[i].l=r[tb],q[i].r=l[ta];
        else
        if (l[ta]<l[tb]) q[i].l=l[ta],q[i].r=l[tb];
        else q[i].l=l[tb],q[i].r=l[ta];
    }

    sort(q+1,q+1+m,cmp);
    L=1;R=0;
    for (i=1;i<=m;i++)
    {
        while (R<q[i].r)R++,cc(R,1);
        while (R>q[i].r)cc(R,-1),R--;
        while (L<q[i].l)cc(L,-1),L++;
        while (L>q[i].l)L--,cc(L,1);
        ans[q[i].id]=cnt;
        if (colsum[q[i].ca]&&colsum[q[i].cb]&&(q[i].ca!=q[i].cb)) ans[q[i].id]--;
    }
    for (i=1;i<=m;i++)
        printf("%d
",ans[i]);
}