链接
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<<"
";
}
for(int i=1;i<=m;++i) printf("%d
",ans[i]);
return 0;
}