【BZOJ】【3757】苹果树

树分块

  orz HZWER 

  http://hzwer.com/5259.html

  不知为何我原本写的倍增求LCA给WA了……学习了HZWER的倍增新姿势~

  树上分块的转移看vfk博客的讲解吧……(其实是先指向hzwer博客,再跳转vfk和KoribohG……)

  vfk讲的很详细,重点就在于转移的时候无视lca,只有在计算答案的时候临时加进来lca,算完答案再把lca去掉。

技术分享
  1 /**************************************************************
  2     Problem: 3757
  3     User: Tunix
  4     Language: C++
  5     Result: Accepted
  6     Time:17180 ms
  7     Memory:17716 kb
  8 ****************************************************************/
  9  
 10 //BZOJ 3757
 11 #include<cmath>
 12 #include<cstdio>
 13 #include<cstring>
 14 #include<cstdlib>
 15 #include<iostream>
 16 #include<algorithm>
 17 #define rep(i,n) for(int i=0;i<n;++i)
 18 #define F(i,j,n) for(int i=j;i<=n;++i)
 19 #define D(i,j,n) for(int i=j;i>=n;--i)
 20 using namespace std;
 21 void read(int &v){
 22     v=0; int sign=1; char ch=getchar();
 23     while(ch<0||ch>9){ if (ch==-) sign=-1; ch=getchar();}
 24     while(ch>=0&&ch<=9){ v=v*10+ch-0; ch=getchar();}
 25     v*=sign;
 26 }
 27 #define debug
 28 /******************tamplate*********************/
 29 const int N=100010;
 30 int head[N],to[N],next[N],cnt;
 31 void add(int x,int y){
 32     to[++cnt]=y; next[cnt]=head[x]; head[x]=cnt;
 33     to[++cnt]=x; next[cnt]=head[y]; head[y]=cnt;
 34 }
 35 /*******************edge************************/
 36 int n,m,color[N],root;
 37 int B,belong[N],st[N],top,tot,deep[N];
 38 int fa[N][20],bin[25];
 39 void dfs(int x){
 40     int bottom=top;
 41     for(int i=1;i<=16;i++)
 42         if(deep[x]>=bin[i])
 43             fa[x][i]=fa[fa[x][i-1]][i-1];
 44         else break;
 45     for(int i=head[x];i;i=next[i])
 46         if (to[i]!=fa[x][0]){
 47             fa[to[i]][0]=x;
 48             deep[to[i]]=deep[x]+1;
 49             dfs(to[i]);
 50             if (top-bottom>=B){
 51                 ++tot;
 52                 while(top!=bottom)
 53                     belong[st[top--]]=tot;
 54             }
 55         }
 56     st[++top]=x;
 57 }
 58 int LCA(int x,int y){
 59     if (deep[x]<deep[y]) swap(x,y);
 60     int t=deep[x]-deep[y];
 61     for(int i=0;bin[i]<=t;i++)
 62         if (t&bin[i]) x=fa[x][i];
 63     D(i,16,0)
 64         if(fa[x][i]!=fa[y][i])
 65             x=fa[x][i],y=fa[y][i];
 66     if(x==y) return x;
 67     return fa[x][0];
 68 }
 69 /*******************dfs&&LCA********************/
 70 struct ques{
 71     int x,y,a,b,num;
 72     bool operator < (const ques &now)const{
 73         if (belong[x]==belong[now.x]) return belong[y]<belong[now.y];
 74         else return belong[x]<belong[now.x];
 75     }
 76 }Q[N];
 77  
 78 int num[N],ans=0,ANS[N];
 79 bool used[N];
 80 inline void work(int x){
 81     if(!used[x]){
 82         num[color[x]]++;used[x]=1;
 83         if(num[color[x]]==1) ans++;
 84     }
 85     else{
 86         num[color[x]]--;used[x]=0;
 87         if(num[color[x]]==0) ans--;
 88     }
 89 }
 90  
 91 void Xor(int x,int y){
 92     int lca=LCA(x,y);
 93     while(x!=lca) {work(x); x=fa[x][0];}
 94     while(y!=lca) {work(y); y=fa[y][0];}
 95 }
 96  
 97 int main(){
 98     bin[0]=1; F(i,1,20) bin[i]=bin[i-1]<<1;
 99  
100     read(n); read(m);
101     B=sqrt(n);
102     F(i,1,n) read(color[i]);
103     int x,y;
104     F(i,1,n){
105         read(x); read(y);
106         if (x==0) root=y;
107         if (y==0) root=x;
108         add(x,y);
109     }
110     dfs(root);
111     tot++;
112     while(top)belong[st[top--]]=tot;
113  
114     int a,b;
115     F(i,1,m){
116         read(x); read(y); read(a); read(b);
117         Q[i]=(ques){x,y,a,b,i};
118     }
119     sort(Q+1,Q+m+1);
120     //转移的时候不考虑LCA,查答案的时候临时算进来,计算完答案后再把LCA删掉
121     int lca=(LCA(Q[1].x,Q[1].y));
122     Xor(Q[1].x,Q[1].y);
123     work(lca);
124     ANS[Q[1].num]=ans;
125     if (num[Q[1].a]!=0 && num[Q[1].b]!=0 && Q[1].a!=Q[1].b) ANS[Q[1].num]--;
126     work(lca);
127     F(i,2,m){
128         Xor(Q[i-1].x,Q[i].x);
129         Xor(Q[i-1].y,Q[i].y);
130         lca=LCA(Q[i].x,Q[i].y);
131         work(lca);
132         ANS[Q[i].num]=ans;
133         if(num[Q[i].a]!=0 && num[Q[i].b]!=0 && Q[i].a!=Q[i].b) ANS[Q[i].num]--;
134         work(lca);
135     }
136     F(i,1,m) printf("%d\n",ANS[i]);
137     return 0;
138 }
View Code

 

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。