树上莫队

树上莫队操作
dfs求出欧拉序
blc=sqrt(2n)。
注意开个pos[i]数组优化，代表i/blc。
注意是2n

夯拙ヽ(￣ω￣(￣ω￣〃)ゝ

将题目所给区间转化为欧拉序区间

    for(i=0;i<m;i++){
        scanf("%d%d",&x,&y);
        if(in[x]>in[y])
            swap(x,y);
        int pre=LCA(x,y);
        if(pre==x){
            a[i].l=in[x];
            a[i].r=in[y];
            a[i].lca=0;
        }
        else{
            a[i].l=out[x];
            a[i].r=in[y];
            a[i].lca=pre;
        }
        //cout<<a[i].l<<" "<<a[i].r<<" hi\n";
        a[i].num=i;
    }

莫队模板

for(i=0;i<m;i++){
        int l=a[i].l,r=a[i].r;
        while(nowr<r){
            nowr++;
            cal(s[nowr]);
        }
        while(nowr>r){
            cal(s[nowr]);
            nowr--;
        }
        while(nowl<l){
            cal(s[nowl]);
            nowl++;
        }
        while(nowl>l){
            nowl--;
            cal(s[nowl]);
        }
   ans[a[i].num]=num;
    }

树上莫队模板题
Count on the tree II

#include<iostream>
#include<cstdio>
#include<vector>
#include<cmath>
#include<set>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAX_N=40100;
const int MAX_M=101000;
vector<int>v[MAX_N];
int blc;
int tt;
int dp[MAX_N][20],dis[MAX_N],n;
int in[MAX_N],out[MAX_N];
int b[MAX_N],s[2*MAX_N];
int tot;
int cnt;
int ver[2*MAX_N],Next[2*MAX_N],head[MAX_N];
void Add(int x,int y){
    ver[++cnt]=y;
    Next[cnt]=head[x];
    head[x]=cnt;
}
void dfs(int now,int fa){
    int i;
    in[now]=++tot;
    s[tot]=now;
    //cout<<tot<<" "<<s[tot]<<" hi\n";
    for(i=head[now];i;i=Next[i]){
        int to=ver[i];
        if(to==fa)
            continue;
        dp[to][0]=now;
        dis[to]=dis[now]+1;
        dfs(to,now);
    }
    out[now]=++tot;
    s[tot]=now;
    //cout<<tot<<" "<<s[tot]<<" hi\n";
}
int LCA(int a,int b){
    int k=tt,i;
    if(dis[a]<dis[b]){
        int c=a;
        a=b;
        b=c;
    }
    while(dis[a]!=dis[b]){
        for(i=k;i>=0;i--){
            if(dis[dp[a][i]]>=dis[b])
            a=dp[a][i];
        }
    }
    if(a==b){
        return b;
    }
    for(i=k;i>=0;i--){
        if(dp[a][i]!=dp[b][i]){
            a=dp[a][i];
            b=dp[b][i];
        }
    }
    return dp[a][0];
}
struct skt{
    int l,r,num,lca;
}a[MAX_M];
int pos[2*MAX_N];
bool cmp(skt a,skt b){
    if(pos[a.l]!=pos[b.l])
        return pos[a.l]<pos[b.l];
    return a.r<b.r;
}
int ans[MAX_M];
int num;
bool vis[MAX_N];
int sub_a[MAX_N],t[MAX_N];
int sum[MAX_N];
void cal(int k){
    if(vis[k]){
        vis[k]=false;
        sum[b[k]]--;
        if(!sum[b[k]]){
            num--;
        }
    }
    else{
        vis[k]=true;
        if(!sum[b[k]]){
            num++;
        }
        sum[b[k]]++;
    }
}
int main(void){
    int m,i,j,x,y;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++){
        scanf("%d",&t[i]);
        sub_a[i-1]=t[i];
    }
    sort(sub_a,sub_a+n);
    int nn=unique(sub_a,sub_a+n)-sub_a;
    for(i=1;i<=n;i++){
        b[i]=lower_bound(sub_a,sub_a+nn,t[i])-sub_a+1;
    }
    for(i=0;i<n-1;i++){
        scanf("%d%d",&x,&y);
        Add(x,y);
        Add(y,x);
    }
    tot=0;
    dis[1]=1;
    dfs(1,0);
    tt=(int)(log(n)/log(2))+1;
    for(i=1;i<=tt;i++){
        for(j=1;j<=n;j++){
            dp[j][i]=dp[dp[j][i-1]][i-1];
        }
    }
    blc=sqrt(2.0*n);
    for(i=1;i<2*n;i++){
        pos[i]=i/blc;
    }
    for(i=0;i<m;i++){
        scanf("%d%d",&x,&y);
        if(in[x]>in[y])
            swap(x,y);
        int pre=LCA(x,y);
        if(pre==x){
            a[i].l=in[x];
            a[i].r=in[y];
            a[i].lca=0;
        }
        else{
            a[i].l=out[x];
            a[i].r=in[y];
            a[i].lca=pre;
        }
        //cout<<a[i].l<<" "<<a[i].r<<" hi\n";
        a[i].num=i;
    }
    sort(a,a+m,cmp);
    int nowl=1,nowr=0;
    for(i=0;i<m;i++){
        int l=a[i].l,r=a[i].r;
        while(nowr<r){
            nowr++;
            cal(s[nowr]);
        }
        while(nowr>r){
            cal(s[nowr]);
            nowr--;
        }
        while(nowl<l){
            cal(s[nowl]);
            nowl++;
        }
        while(nowl>l){
            nowl--;
            cal(s[nowl]);
        }
        int op=0;
        if(a[i].lca){
            if(b[a[i].lca]!=0&&!sum[b[a[i].lca]]){
                op=1;
                //cout<<"hi\n";
            }
        }
        ans[a[i].num]=num+op;
    }
    for(i=0;i<m;i++)
        printf("%d\n",ans[i]);
    return 0;
}