SPOJ COT2 Count on a tree II (树上莫队)

题意：

思路：

     树上分块方法<-由这个题，我们可以得出一种分块方法，他能保证每个块中内部移动次数，同时也能保证块的大小
     这个是讲述怎么转移的因为转移的时候，lca非常麻烦，要各种分类讨论，这个博客告诉我们，只要不管lca进行转移，最后计算结果的时候临时加上lca即可

错误及反思：

第一个树上莫队，所以跑得有点慢，写的也有点长，以后写的好了再改改

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 101000;

map<int,int> mp;
vector<int> son[N];
stack<int> s;

int blocks,block[N],cnt[N],ans[N],to[N][20],tot=0,tid=0,depth[N],now=0,n,m,val[N];
bool vis[N];

struct Q{
    int l,r,id;
}q[N];

bool cmp(Q a,Q b){
    if(block[a.l]!=block[b.l])
        return block[a.l]<block[b.l];
    return block[a.r]<block[b.r];
}

void dfs(int now,int fa,int dep){
    int si=s.size();
    to[now][0]=fa;
    depth[now]=dep;
    for(int i=0;i<son[now].size();i++){
        if(son[now][i]!=fa){
            dfs(son[now][i],now,dep+1);
           if(s.size()-si>=blocks){
                while(s.size()!=si){
                    block[s.top()]=tot;
                    s.pop();
                }
                tot++;
            }
        }
    }
    s.push(now);
}


void getlca(){
    for(int i=1;i<=18;i++)
        for(int j=1;j<=n;j++)
            to[j][i]=to[to[j][i-1]][i-1];
}

int lca(int a,int b){
    if(depth[a]>depth[b])
        swap(a,b);
    for(int i=18;i>=0;i--)
        if(depth[to[b][i]]>=depth[a])
            b=to[b][i];
    if(a==b) return a;
    for(int i=18;i>=0;i--){
        if(to[a][i]!=to[b][i]){
            a=to[a][i];
            b=to[b][i];
        }
    }
    return to[a][0];
}

void change(int u,int v){
    int zx=lca(u,v);
    while(u!=zx){
        if(vis[u]){
            cnt[val[u]]--;
            if(cnt[val[u]]==0)
                now--;
        }
        else{
            cnt[val[u]]++;
            if(cnt[val[u]]==1)
                now++;
        }
        vis[u]=!vis[u];
        u=to[u][0];
    }
    while(v!=zx){
        if(vis[v]){
            cnt[val[v]]--;
            if(cnt[val[v]]==0)
                now--;
        }
        else{
            cnt[val[v]]++;
            if(cnt[val[v]]==1)
                now++;
        }
        vis[v]=!vis[v];
        v=to[v][0];
    }

}
int main(){
    scanf("%d%d",&n,&m);
    blocks=sqrt(n);
    for(int i=1;i<=n;i++){
        scanf("%d",&val[i]);
        if(!mp[val[i]]) mp[val[i]]=val[i]=++tid;
        else val[i]=mp[val[i]];
    }
    for(int i=1;i<n;i++){
        int ta,tb;
        scanf("%d%d",&ta,&tb);
        son[ta].push_back(tb);
        son[tb].push_back(ta);
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    dfs(1,1,1);
    while(s.empty()){
        block[s.top()]=tot-1;
        s.pop();
    }
    getlca();
    sort(q+1,q+m+1,cmp);
    for(int i=1,l=1,r=1;i<=m;i++){
        if(l!=q[i].l) change(l,q[i].l);
        if(r!=q[i].r) change(r,q[i].r);
        int zx=lca(q[i].l,q[i].r);
        if(cnt[val[zx]]==0)
            ans[q[i].id]=now+1;
        else ans[q[i].id]=now;
        l=q[i].l;
        r=q[i].r;
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}