BZOJ3757 : 苹果树

树上莫队，有几个坑：

1. a可能等于b

2. 数据范围是100000

#include<cstdio>

#include<algorithm>

#include<cmath>

#define N 100010

#define K 17

using namespace std;

inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}

struct P{int l,r,z,a,b,id;}Q[N];

int lim,pos[N<<1],l,r,c[N],g[N],v[N<<1],nxt[N<<1],ed,n,m,i,j,x,y,z,loc[N<<1],dfn,st[N],en[N],d[N],f[N][18],ans[N],cnt[N],sum;bool vis[N];char ch;

inline bool cmp(P a,P b){return pos[a.l]<pos[b.l]||pos[a.l]==pos[b.l]&&a.r<b.r;}

inline void add(int x,int y){v[++ed]=y;nxt[ed]=g[x];g[x]=ed;}

void dfs(int x){

  for(int i=vis[loc[st[x]=++dfn]=x]=1;i<=K;i++)f[x][i]=f[f[x][i-1]][i-1];

  for(int i=g[x];i;i=nxt[i])if(!vis[v[i]])d[v[i]]=d[f[v[i]][0]=x]+1,dfs(v[i]);

  loc[en[x]=++dfn]=x;

}

inline int lca(int x,int y){

  if(x==y)return x;

  if(d[x]<d[y])swap(x,y);

  for(int i=K;~i;i--)if(d[f[x][i]]>=d[y])x=f[x][i];

  if(x==y)return x;

  for(int i=K;~i;i--)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];

  return f[x][0];

}

inline void deal(int x){

  if(!vis[x]){if(!(--cnt[c[x]]))sum--;}else if(!(cnt[c[x]]++))sum++;

  vis[x]^=1;

}

int main(){

  for(read(n),read(m),i=1;i<=n;i++)read(c[i]);

  for(i=1;i<=n;i++){

    read(x),read(y);

    if(x*y)add(x,y),add(y,x);

  }

  for(dfs(d[1]=1),lim=(int)sqrt(n*2+0.5),i=1;i<=dfn;i++)pos[i]=(i-1)/lim+1;

  for(i=1;i<=m;i++){

    read(x),read(y),read(Q[i].a),read(Q[i].b);Q[i].id=i;

    if(st[x]>st[y])swap(x,y);

    z=lca(x,y);

    if(z==x)Q[i].l=st[x],Q[i].r=st[y];else Q[i].l=en[x],Q[i].r=st[y],Q[i].z=z;

  }

  for(sort(Q+1,Q+m+1,cmp),i=1,l=1,r=0;i<=m;i++){

    if(r<Q[i].r){for(r++;r<=Q[i].r;r++)deal(loc[r]);r--;}

    if(r>Q[i].r)for(;r>Q[i].r;r--)deal(loc[r]);

    if(l<Q[i].l)for(;l<Q[i].l;l++)deal(loc[l]);

    else if(l>Q[i].l){for(l--;l>=Q[i].l;l--)deal(loc[l]);l++;}

    if(Q[i].z)deal(Q[i].z);

    ans[Q[i].id]=sum-(Q[i].a!=Q[i].b&&cnt[Q[i].a]&&cnt[Q[i].b]);

    if(Q[i].z)deal(Q[i].z);

  }

  for(i=1;i<=m;i++)printf("%d\n",ans[i]);

  return 0;

}