樹上莫隊操作
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;
}