SP10707 COT2 - Count on a tree II

链接

https://vjudge.net/problem/SPOJ-COT2
https://www.luogu.org/problemnew/show/SP10707

思路

dfs欧拉序转化为普通莫队（并不算树上莫队，不过也可做)
好神仙啊，原来欧拉序是可以求任意两点的点，不过要加lca。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+7;
int read() {
    int x=0,f=1;char s=getchar();
    for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
    return x*f;
}
int n,m,w[N],now_ans,top;
vector<int> G[N];
int st[N][30],dep[N],belong[N<<1],ans[N];
int lsh[N],a[N<<1],be[N],en[N<<1],T[N];
int js[N];
struct node {
    int l,r,other,id;
    bool operator < (const node &b) const {
        return belong[l]==belong[b.l] ? belong[l]&1 ? r<b.r : r>b.r :belong[l]<belong[b.l];
    }
}Q[N];
void dfs(int u,int f) {
    a[++top]=u;
    be[u]=top;
    for(vector<int>::iterator it=G[u].begin();it!=G[u].end();++it) {
        if(*it==f) continue;
        st[*it][0]=u;
        dep[*it]=dep[u]+1;
        dfs(*it,u);
    }
    a[++top]=u;
    en[u]=top;
}
int lca(int x,int y) {
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=25;i>=0;--i)
        if(dep[st[x][i]]>=dep[y]) x=st[x][i];
    if(x==y) return x;
    for(int i=25;i>=0;--i)
        if(st[x][i]!=st[y][i]) x=st[x][i],y=st[y][i];
    return st[x][0];
}
void add(int x) {
    if(!T[x]) now_ans++;
    T[x]++;
}
void del(int x) {
    T[x]--;
    if(!T[x]) now_ans--;
}
int main() {
    //freopen("a.in","r",stdin);
    n=read(),m=read();
    for(int i=1;i<=n;++i) lsh[i]=w[i]=read();
    sort(lsh+1,lsh+1+n);
    int len=unique(lsh+1,lsh+1+n)-lsh-1;
    for(int i=1;i<=n;++i) w[i]=lower_bound(lsh+1,lsh+1+len,w[i])-lsh;
    for(int i=1;i<n;++i) {
        int x=read(),y=read();
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dep[1]=1;
    dfs(1,0);
    for(int j=1;j<=25;++j) {
        for(int i=1;i<=n;++i) {
            st[i][j]=st[st[i][j-1]][j-1];
        }
    }
    for(int i=1;i<=m;++i) {
        int x=read(),y=read(),z=lca(x,y);
        if(be[x]>be[y]) swap(x,y);
        if(z==x||z==y) {
            Q[i].l=be[x];
            Q[i].r=be[y];
            Q[i].other=0;
        } else {
            Q[i].l=en[x];
            Q[i].r=be[y];
            Q[i].other=z;
        }
        Q[i].id=i;
    }
    int k=sqrt(2*n);
    for(int i=1;i<=2*n;++i) belong[i]=(i-1)/k+1;
    sort(Q+1,Q+1+m);
    for(int i=1,l=1,r=0;i<=m;++i) {
        while(r<Q[i].r) {//add(++r)
            ++r;
            if(js[a[r]])
                del(w[a[r]]);
            else
                add(w[a[r]]);
            js[a[r]]^=1;
        }
        while(l>Q[i].l) {//add(--l)
            --l;
            if(js[a[l]])
                del(w[a[l]]);
            else
                add(w[a[l]]);

            js[a[l]]^=1;
        }
        while(l<Q[i].l) {//del(l++)
            if(js[a[l]])
                del(w[a[l]]);
            else
                add(w[a[l]]);
            js[a[l]]^=1;

            l++;
        }
        while(r>Q[i].r) {//del(r--)
            if(js[a[r]])
                del(w[a[r]]);
            else
                add(w[a[r]]);
            js[a[r]]^=1;

            r--;
        }
        if(Q[i].other)
            add(w[Q[i].other]);
        ans[Q[i].id]=now_ans;
        if(Q[i].other)
            del(w[Q[i].other]);
//      cout<<Q[i].l<<" "<<Q[i].r<<"\n";
    }
    for(int i=1;i<=m;++i) printf("%d\n",ans[i]);
    return 0;
}