BZOJ1036 (其实这只是一份板子)
生活随笔
收集整理的這篇文章主要介紹了
BZOJ1036 (其实这只是一份板子)
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
我說我是不是完蛋了啊... ...昨天考試線段樹寫錯,調(diào)了好久才調(diào)回來;今天做這道樹鏈剖分辣雞操作題,TM寫錯了4個地方!先是建樹為了省常數(shù)打了一個build結(jié)果初值賦錯了,然后又是線段樹!getmax和getsum錯的還不是一個地方!最后我發(fā)現(xiàn)update賦的不是dfn[x]而是x!再這樣下去我是不是有一天連for循環(huán)都不會打了啊!!!
好吧我是用Windows的Devcpp寫的,丑了一點啊。
#include <iostream> #include <cstdio> #include <cstdlib> #include <algorithm> #define ls (x<<1) #define rs (x<<1|1) using namespace std; const int N = 30010; struct Node{int to,next;}E[N<<1]; int n,m,fa[N],top[N],val[N],T[N<<4],T2[N<<4],son[N],size[N],rec[N],dfn[N],tot,dep[N],num,head[N]; inline int gi() {int x=0,res=1;char ch=getchar();while(ch<'0' || ch>'9'){if(ch=='-')res=-res;ch=getchar();}while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();return x*res; } inline void link(int u,int v) {E[++tot]=(Node){v,head[u]};head[u]=tot;E[++tot]=(Node){u,head[v]};head[v]=tot; } inline void dfs1(int x,int fat) {dep[x]=dep[fat]+1;fa[x]=fat;size[x]++;for(int e=head[x];e;e=E[e].next){if(E[e].to==fat)continue;dfs1(E[e].to,x);size[x]+=size[E[e].to];if(size[E[e].to]>size[son[x]])son[x]=E[e].to;} } inline void dfs2(int x,int tp) {top[x]=tp;dfn[x]=++num;rec[num]=x;if(!son[x])return;dfs2(son[x],tp);for(int e=head[x];e;e=E[e].next)if(E[e].to!=fa[x]&&E[e].to!=son[x])dfs2(E[e].to,E[e].to); } inline void build(int x,int l,int r) {if(l==r){T2[x]=T[x]=val[rec[l]];return;}int mid=(l+r)>>1;build(ls,l,mid);build(rs,mid+1,r);T2[x]=T2[ls]+T2[rs];T[x]=max(T[ls],T[rs]); } inline int gc() {char ch=getchar();while(ch<'A'||ch>'Z')ch=getchar();if(ch=='C')return 1;if(ch=='Q')ch=getchar();if(ch=='M')return 2;if(ch=='S')return 3; } inline void update(int x,int l,int r,int xp,int v) {if(l==r){T[x]=T2[x]=v;return;}int mid=(l+r)>>1;if(xp<=mid)update(ls,l,mid,xp,v);else update(rs,mid+1,r,xp,v);T2[x]=T2[ls]+T2[rs];T[x]=max(T[ls],T[rs]); } inline int getmax(int x,int l,int r,int xl,int xr) {if(xl<=l && r<=xr)return T[x];int mid=(l+r)>>1;if(xr<=mid)return getmax(ls,l,mid,xl,xr);else if(xl>mid)return getmax(rs,mid+1,r,xl,xr);return max(getmax(ls,l,mid,xl,mid),getmax(rs,mid+1,r,mid+1,xr)); } inline int gmax(int u,int v) {int Ans=-N*N;while(top[u]^top[v]){if(dep[top[u]]<dep[top[v]])swap(u,v);Ans=max(Ans,getmax(1,1,n,dfn[top[u]],dfn[u]));u=fa[top[u]];}if(dep[u]<dep[v])swap(u,v);Ans=max(Ans,getmax(1,1,n,dfn[v],dfn[u]));return Ans; } inline int getsum(int x,int l,int r,int xl,int xr) {if(xl<=l && r<=xr)return T2[x];int mid=(l+r)>>1;if(xr<=mid)return getsum(ls,l,mid,xl,xr);else if(xl>mid)return getsum(rs,mid+1,r,xl,xr);else return getsum(ls,l,mid,xl,mid)+getsum(rs,mid+1,r,mid+1,xr); } inline int gsum(int u,int v) {int Ans=0;while(top[u]^top[v]){if(dep[top[u]]<dep[top[v]])swap(u,v);Ans+=getsum(1,1,n,dfn[top[u]],dfn[u]);u=fa[top[u]];}if(dep[u]<dep[v])swap(u,v);Ans+=getsum(1,1,n,dfn[v],dfn[u]);return Ans; } int main() {n=gi();for(int i=1;i<n;++i)link(gi(),gi());for(int i=1;i<=n;++i)val[i]=gi();dfs1(1,1);dfs2(1,1);build(1,1,n);m=gi();while(m--){int type=gc(),x=gi(),y=gi();switch(type){case 1:{update(1,1,n,dfn[x],y);break;}case 2:{printf("%d\n",gmax(x,y));break;}case 3:{printf("%d\n",gsum(x,y));break;}default:{break;}}} }
轉(zhuǎn)載于:https://www.cnblogs.com/fenghaoran/p/6623012.html
總結(jié)
以上是生活随笔為你收集整理的BZOJ1036 (其实这只是一份板子)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 两节点的最小公共祖先LCA
- 下一篇: 脚本检测CDN节点资源是否与源站资源一致