【bzoj2243】染色 树链剖分+线段树
2017-02-15 16:48
260 查看
AC通道:http://www.lydsy.com/JudgeOnline/problem.php?id=2243
【题解】
神坑题目,今天一天都在调这道题了。
首先asksum函数中忘记判断区间合并时出现相同颜色的情况,导致wa不断。
然后要到了数据,查出了这个错误。
然而忘记了deep[1]=1,导致在求lca的过程中访问到0号结点,然后又开始RE
然后我的一整天都在二分出错位置了。。。。。。。
【题解】
神坑题目,今天一天都在调这道题了。
首先asksum函数中忘记判断区间合并时出现相同颜色的情况,导致wa不断。
然后要到了数据,查出了这个错误。
然而忘记了deep[1]=1,导致在求lca的过程中访问到0号结点,然后又开始RE
然后我的一整天都在二分出错位置了。。。。。。。
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<ctime> #include<cmath> #include<algorithm> using namespace std; #define FILE "read" #define MAXN 100010 #define up(i,j,n) for(int i=j;i<=n;++i) #define dn(i,j,n) for(int i=j;i>=n;--i) #define cmax(a,b) a=max(a,b) #define cmin(a,b) a=min(a,b) namespace INIT{ char buf[1<<15],*fs,*ft; inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;} inline int read(){ int x=0,f=1; char ch=getchar(); while(!isdigit(ch)) {if(ch=='-') f=-1; ch=getchar();} while(isdigit(ch)) {x=x*10+ch-'0'; ch=getchar();} return x*f; } }using namespace INIT; struct node{int y,next;}e[MAXN*2]; int n,m,len,z,a[MAXN],Link[MAXN],size[MAXN],son[MAXN],deep[MAXN],f[MAXN],top[MAXN],rank[MAXN],w[MAXN],cl[MAXN*8],cr[MAXN*8],v[MAXN*8],tag[MAXN*8],anc[MAXN][25]; void insert(int x,int y){e[++len].next=Link[x];Link[x]=len;e[len].y=y;} void updata(int p){cl[p]=cl[p<<1];cr[p]=cr[p<<1|1];v[p]=v[p<<1]+v[p<<1|1]-(cr[p<<1]==cl[p<<1|1]);} void dfs(int x){ size[x]=1; son[x]=0; anc[x][0]=f[x]; up(i,1,20) anc[x][i]=anc[anc[x][i-1]][i-1]; for(int i=Link[x];i;i=e[i].next){ if(e[i].y==f[x]) continue; deep[e[i].y]=deep[x]+1; f[e[i].y]=x; dfs(e[i].y); size[x]+=size[e[i].y]; if(size[e[i].y]>size[son[x]]) son[x]=e[i].y; } } void dfs(int x,int temp){ w[x]=++z; rank[z]=x; top[x]=temp; if(son[x]) dfs(son[x],temp); for(int i=Link[x];i;i=e[i].next) if(e[i].y!=f[x]&&e[i].y!=son[x]) dfs(e[i].y,e[i].y); } void build(int p,int l,int r){ if(l>r) return; if(l==r) {cl[p]=cr[p]=a[rank[l]]; v[p]=1; return;} int mid=(l+r)>>1; build(p<<1,l,mid); build(p<<1|1,mid+1,r); updata(p); } void pushdown(int p){ if(tag[p]==-1) return; int temp=tag[p]; tag[p]=-1; tag[p<<1]=temp; tag[p<<1|1]=temp; v[p<<1]=v[p<<1|1]=1; cl[p<<1]=cr[p<<1]=cl[p<<1|1]=cr[p<<1|1]=temp; } void change(int p,int l,int r,int x,int y,int val){ pushdown(p); if(x>r||y<l) return; if(x<=l&&y>=r) {tag[p]=cl[p]=cr[p]=val;v[p]=1;return;} int mid=(l+r)>>1; change(p<<1,l,mid,x,y,val); change(p<<1|1,mid+1,r,x,y,val); updata(p); } void solvechange(int x,int y,int val){ while(top[x]!=top[y]){ change(1,1,z,w[top[x]],w[x],val); x=f[top[x]]; } change(1,1,z,w[y],w[x],val); } int find(int p,int l,int r,int x){ pushdown(p); if(x>r||x<l) return 0; if(l==r) return cl[p]; int mid=(l+r)>>1; return find(p<<1,l,mid,x)+find(p<<1|1,mid+1,r,x); } int asksum(int p,int l,int r,int x,int y){ pushdown(p); if(l==x&&r==y)return v[p]; int mid=(l+r)>>1; if(mid>=y)return asksum(p<<1,l,mid,x,y); else if(mid<x)return asksum(p<<1|1,mid+1,r,x,y); else return asksum(p<<1,l,mid,x,mid)+asksum(p<<1|1,mid+1,r,mid+1,y)-(cr[p<<1]==cl[p<<1|1]); } int query(int x,int y){ int ans=0; while(top[x]!=top[y]){ ans+=asksum(1,1,z,w[top[x]],w[x]); if(find(1,1,z,w[top[x]])==find(1,1,z,w[f[top[x]]])) --ans; x=f[top[x]]; } ans+=asksum(1,1,z,w[y],w[x]); return ans; } int lca(int x,int y){ if(deep[x]<deep[y]) swap(x,y); dn(i,20,0) if(deep[anc[x][i]]>=deep[y]) x=anc[x][i]; if(x==y) return x; dn(i,20,0) if(anc[x][i]!=anc[y][i]) x=anc[x][i],y=anc[y][i]; return f[x]; } int main(){ freopen(FILE".in","r",stdin); freopen(FILE".out","w",stdout); n=read(); m=read(); memset(tag,-1,sizeof(tag)); up(i,1,n) a[i]=read(); deep[1]=1; //坑点:一定要有deep[1]=1,否则会在求lca的时候出现RE,博主因此检查了一个下午 up(i,1,n-1) {int x=read(),y=read(); insert(x,y);insert(y,x);} dfs(1); dfs(1,1); build(1,1,z); //up(i,1,n) change(1,1,z,w[i],w[i],a[i]); //build建树常数更小,比直接修改要快1400ms up(i,1,m){ char ch[5]; scanf("%s",ch+1); if(ch[1]=='C') { int x=read(),y=read(),val=read(),t=lca(x,y); solvechange(x,t,val); solvechange(y,t,val); } else { int x=read(),y=read(),t=lca(x,y); printf("%d\n",query(x,t)+query(y,t)-1); } } return 0; }
相关文章推荐
- Bzoj 2243: [SDOI2011]染色(树链剖分+线段树)
- [BZOJ]2243 染色 树链剖分+线段树
- [SDOI2011]染色 BZOJ2243 树链剖分+线段树
- 【BZOJ2243】【SDOI2011】染色(树链剖分+线段树)
- 【bzoj2243】[SDOI2011]染色 树链剖分+线段树
- BZOJ 2243 染色(树链剖分+线段树)
- bzoj 2243: [SDOI2011]染色 (树链剖分+线段树)
- 【BZOJ2243】【SDOI2011】染色(树链剖分+线段树)
- bzoj 2243: [SDOI2011]染色 树链剖分+线段树
- bzoj2243[SDOI2011]染色 树链剖分+线段树
- BZOJ 2243: [SDOI2011]染色 树链剖分+线段树
- [bzoj2243][SDOI2011]染色——树链剖分+线段树
- [BZOJ2243][SDOI2011]染色(树链剖分+线段树)
- 【BZOJ 2243】染色 - 树链剖分+线段树
- BZOJ2243: [SDOI2011]染色
- BZOJ2243: [SDOI2011]染色
- BZOJ 2243: [SDOI2011]染色 (树链剖分,点权,线段树)
- Bzoj 2243: [SDOI2011]染色 树链剖分,LCT,动态树
- bzoj2243: [SDOI2011]染色
- 【BZOJ 2243】 [SDOI2011]染色