bzoj2243: [SDOI2011]染色--线段树+树链剖分

Posted 2020-07-14

此题代码量较大。。但是打起来很爽

原本不用lca做一直wa不知道为什么。。

后来改lca重打了一遍= =结果一遍就AC了orz

题目比较裸，也挺容易打，主要是因为思路可以比较清晰

另：加读入优化比没加快了1.3s。。

 

  1 #include<stdio.h>
  2 #include<string.h>
  3 #include<algorithm>
  4 using namespace std;
  5 const int maxn = 100010;
  6 struct node{
  7     int l,r,lc,rc,sum,lz;
  8 }t[maxn*4];
  9 struct edge{
 10     int to,next;
 11 }e[maxn*2];
 12 int n,m,u,v,tot,cnt,logn=0;
 13 int head[maxn],son[maxn],pre[maxn],tree[maxn],top[maxn],dep[maxn],fa[maxn][30],size[maxn],col[maxn];
 14 
 15 inline void read(int &x){
 16     char cc=getchar(); x=0; int f=1;
 17     while (cc<‘0‘ || cc>‘9‘){if (cc==‘-‘) f=-1; cc=getchar();}
 18     while (cc>=‘0‘ && cc<=‘9‘) x=x*10+cc-‘0‘,cc=getchar(); x*=f;
 19 }
 20 
 21 inline void insert(int u, int v){
 22     e[++tot].to=v; e[tot].next=head[u]; head[u]=tot;
 23 }
 24 
 25 inline void dfs1(int u, int f, int d){
 26     size[u]=1; fa[u][0]=f; dep[u]=d;
 27     for (int i=1; i<=logn; i++) fa[u][i]=fa[fa[u][i-1]][i-1];
 28     for (int i=head[u]; i!=-1; i=e[i].next){
 29         int v=e[i].to;
 30         if (v==f) continue;
 31         dfs1(v,u,d+1);
 32         size[u]+=size[v];
 33         if (!son[u] || size[v]>size[son[u]]) son[u]=v;
 34     }
 35 }
 36 
 37 inline void dfs2(int u, int num){
 38     top[u]=num; tree[u]=++cnt;
 39     pre[cnt]=u;
 40     if (!son[u]) return;
 41     dfs2(son[u],num);
 42     for (int i=head[u]; i!=-1; i=e[i].next)
 43         if (e[i].to!=fa[u][0] && e[i].to!=son[u])
 44             dfs2(e[i].to,e[i].to);
 45 }
 46 
 47 inline int lca(int u, int v){
 48     if (dep[u]>dep[v]) swap(u,v);
 49     while (dep[u]<dep[v]){
 50         for (int i=logn; i>=0; i--)
 51             if (dep[u]<dep[fa[v][i]])
 52                 v=fa[v][i];
 53         v=fa[v][0];
 54     }
 55     if (u==v) return u;
 56     for (int i=logn; i>=0; i--)
 57         if (fa[u][i]!=fa[v][i]){
 58             u=fa[u][i];
 59             v=fa[v][i];
 60         }
 61     u=fa[u][0];
 62     return u;
 63 }
 64 
 65 inline void pushdown(int x){
 66     if (t[x].lz){
 67         t[x<<1].lz=t[x<<1|1].lz=t[x].lz;
 68         t[x<<1].lc=t[x<<1].rc=t[x<<1|1].lc=t[x<<1|1].rc=t[x].lz;
 69         t[x<<1].sum=t[x<<1|1].sum=1;
 70         t[x].lz=0;
 71     }
 72 }
 73 
 74 inline void pushup(int x){
 75     t[x].lc=t[x<<1].lc; t[x].rc=t[x<<1|1].rc;
 76     t[x].sum=t[x<<1].sum+t[x<<1|1].sum-(t[x<<1].rc==t[x<<1|1].lc);
 77 }
 78 
 79 inline int query(int a, int b, int x){
 80     int l=t[x].l, r=t[x].r;
 81     if (a==l && r==b) return t[x].sum;
 82     int mid=l+r>>1;
 83     pushdown(x);
 84     if (b<=mid) return query(a,b,x<<1);
 85     else if (a>mid) return query(a,b,x<<1|1);
 86     else return query(a,mid,x<<1)+query(mid+1,b,x<<1|1)-(t[x<<1].rc==t[x<<1|1].lc);
 87 }
 88 
 89 inline void update(int a, int b, int c, int x){
 90     int l=t[x].l, r=t[x].r;
 91     if (a==l && r==b){
 92         t[x].lc=t[x].rc=t[x].lz=c;
 93         t[x].sum=1;
 94         return;
 95     }
 96     int mid=l+r>>1;
 97     pushdown(x);
 98     if (b<=mid) update(a,b,c,x<<1);
 99     else if (a>mid) update(a,b,c,x<<1|1);
100     else{
101         update(a,mid,c,x<<1);
102         update(mid+1,b,c,x<<1|1);
103     }
104     pushup(x);
105 }
106 
107 inline int get_col(int a, int x){
108     int l=t[x].l, r=t[x].r;
109     if (l==r) return t[x].lc;
110     pushdown(x);
111     int mid=l+r>>1;
112     if (a<=mid) return get_col(a,x<<1);
113     else return get_col(a,x<<1|1);
114 }
115 
116 inline void build(int l, int r, int x){
117     t[x].l=l; t[x].r=r;
118     if (l==r){
119         t[x].lc=t[x].rc=col[pre[l]];
120         t[x].sum=1;
121         return;
122     }
123     int mid=l+r>>1;
124     if (l<=mid) build(l,mid,x<<1);
125     if (r>mid) build(mid+1,r,x<<1|1);
126     pushup(x);
127 }
128 
129 inline void change(int x, int f, int c){
130     while (top[x]!=top[f]){
131         update(tree[top[x]],tree[x],c,1);
132         x=fa[top[x]][0];
133     }
134     update(tree[f],tree[x],c,1);
135 }
136 
137 inline int get_sum(int x, int f){
138     int res=0;
139     while (top[x]!=top[f]){
140         res+=query(tree[top[x]],tree[x],1);
141         if (get_col(tree[top[x]],1)==get_col(tree[fa[top[x]][0]],1)) res--;
142         x=fa[top[x]][0];
143     }
144     res+=query(tree[f],tree[x],1);
145     return res;
146 }
147 
148 int main(){
149     read(n); read(m);
150     while ((1<<logn)<n) logn++;
151     for (int i=1; i<=n; i++) read(col[i]),col[i]++;
152     tot=-1; memset(head,-1,sizeof(head));
153     for (int i=1; i<n; i++){
154         read(u); read(v);
155         insert(u,v); insert(v,u);
156     }
157     cnt=0;
158     dfs1(1,0,1); dfs2(1,0);
159     build(1,n,1);
160     char s[5];
161     while (m--){
162         scanf("%s", s);
163         if (s[0]==‘Q‘){
164             read(u); read(v);
165             int t=lca(u,v);
166             printf("%d\n", get_sum(u,t)+get_sum(v,t)-1);
167         }
168         else{
169             int color;
170             read(u); read(v); read(color);
171             int t=lca(u,v);
172             color++;
173             change(u,t,color);
174             change(v,t,color);
175         }
176     }
177     return 0;
178 }