AC日记——Count on a tree II spoj

Count on a tree II

思路：

　　树上莫队；

　　先分块，然后，就好办了；

来，上代码：

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define maxn 40005
#define maxm 100005

struct QueryType {
    int u,v,id;
};
struct QueryType qu[maxm];

int n,m,ti[maxn],num[maxn],Hash[maxn],siz;
int bel[maxn],f[maxn],deep[maxn],ans[maxm];
int E[maxn<<],V[maxn<<],head[maxn],cnt=;
int top[maxn],size[maxn],dis[maxn],sizee;

bool if_[maxn];

inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'') Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}

void pre(int now,int fa)
{
    f[now]=fa,deep[now]=deep[fa]+;
    bel[now]=(++cnt+)/siz,size[now]=;
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==fa) continue;
        pre(V[i],now),size[now]+=size[V[i]];
    }
}

void dfs(int now,int chain)
{
    top[now]=chain;int pos=;
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==f[now]) continue;
        if(size[V[i]]>size[pos]) pos=V[i];
    }
    if(pos==) return ;
    dfs(pos,chain);
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==f[now]||V[i]==pos) continue;
        dfs(V[i],V[i]);
    }
}

int solve_lca(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(deep[top[x]]<deep[top[y]]) swap(x,y);
        x=f[top[x]];
    }
    if(deep[x]>deep[y]) swap(x,y);
    return x;
}

bool cmp(QueryType aa,QueryType bb)
{
    if(bel[aa.u]==bel[bb.u]) return bel[aa.v]<bel[bb.v];
    else return bel[aa.u]<bel[bb.u];
}

inline void updata(int to)
{
    if(if_[to])
    {
        ti[num[to]]--;
        if(ti[num[to]]==) cnt--;
    }
    else
    {
        ti[num[to]]++;
        if(ti[num[to]]==) cnt++;
    }
    if_[to]=!if_[to];
}

int main()
{
    in(n),in(m);siz=sqrt(n);
    for(int i=;i<=n;i++) in(num[i]),Hash[i]=num[i];
    sort(Hash+,Hash+n+);
    sizee=unique(Hash+,Hash+n+)-Hash-;int u,v;
    for(int i=;i<=n;i++) num[i]=lower_bound(Hash+,Hash+sizee+,num[i])-Hash;
    for(int i=;i<n;i++)
    {
        in(u),in(v);
        E[++cnt]=head[u],V[cnt]=v,head[u]=cnt;
        E[++cnt]=head[v],V[cnt]=u,head[v]=cnt;
    }
    cnt=,pre(,),dfs(,);
    for(int i=;i<=m;i++)
    {
        in(qu[i].u),in(qu[i].v),qu[i].id=i;
        if(bel[qu[i].u]>bel[qu[i].v]) swap(qu[i].u,qu[i].v);
    }
    sort(qu+,qu+m+,cmp);u=,v=,cnt=;
    for(int no=;no<=m;no++)
    {
        int lca=solve_lca(u,qu[no].u);
        while(u!=f[lca]) updata(u),u=f[u];u=qu[no].u;
        while(u!=f[lca]) updata(u),u=f[u];u=qu[no].u;
        lca=solve_lca(v,qu[no].v);
        while(v!=f[lca]) updata(v),v=f[v];v=qu[no].v;
        while(v!=f[lca]) updata(v),v=f[v];v=qu[no].v;
        lca=solve_lca(u,v);
        updata(lca),ans[qu[no].id]=cnt,updata(lca);
    }
    for(int i=;i<=m;i++) printf("%d\n",ans[i]);
    return ;
}