BZOJ5379: Tree
Description
题目大意就是:1. 换根。2. 对LCA(u,v)LCA(u,v)的子树修改。3. 求x子树的权值和。
如果没有操作一,这就是一道沙茶题对吧。。。
于是考虑换根的影响。
具体可以看这题:链接。
但是那个换根后的LCA(u,v)怎么解?
其实换完根的LCA(u,v)就是以1为根的LCA(u,v),LCA(u,root),LCA(v,root)中深度最大的那个。
这题就愉快地解决了。。。
原来写的那个板子有点BUG,导致我这题WA了无数发。。。
药丸。。。
附代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 | #include<iostream> #include<algorithm> #include<cstdio> #define LSON rt<<1 #define RSON rt<<1|1 #define DATA(x) b[x].data #define SIGN(x) b[x].c #define LSIDE(x) b[x].l #define RSIDE(x) b[x].r #define WIDTH(x) (RSIDE(x)-LSIDE(x)+1) #define MAXN 300010 using namespace std; int n,m,c=1,d=1,root=1; int val[MAXN],head[MAXN],deep[MAXN],son[MAXN],size[MAXN],fa[MAXN],id[MAXN],pos[MAXN],top[MAXN]; struct Tree{ int next,to; }a[MAXN<<1]; struct Segment_Tree{ long long data,c; int l,r; }b[MAXN<<2]; inline int read(){ int date=0,w=1; char c=0; while (c< '0' ||c> '9' ){ if (c== '-' )w=-1;c= getchar ();} while (c>= '0' &&c<= '9' ){date=date*10+c- '0' ;c= getchar ();} return date*w; } inline void add( int x, int y){ a[c].to=y;a[c].next=head[x];head[x]=c++; a[c].to=x;a[c].next=head[y];head[y]=c++; } void dfs1( int rt){ son[rt]=0;size[rt]=1; for ( int i=head[rt];i;i=a[i].next){ int will=a[i].to; if (!deep[will]){ deep[will]=deep[rt]+1; fa[will]=rt; dfs1(will); size[rt]+=size[will]; if (size[son[rt]]<size[will])son[rt]=will; } } } void dfs2( int rt, int f){ id[rt]=d++;pos[id[rt]]=rt;top[rt]=f; if (son[rt])dfs2(son[rt],f); for ( int i=head[rt];i;i=a[i].next){ int will=a[i].to; if (will!=fa[rt]&&will!=son[rt])dfs2(will,will); } } inline void pushup( int rt){ DATA(rt)=DATA(LSON)+DATA(RSON); } inline void pushdown( int rt){ if (!SIGN(rt)||LSIDE(rt)==RSIDE(rt)) return ; SIGN(LSON)+=SIGN(rt); DATA(LSON)+=SIGN(rt)*WIDTH(LSON); SIGN(RSON)+=SIGN(rt); DATA(RSON)+=SIGN(rt)*WIDTH(RSON); SIGN(rt)=0; } void buildtree( int l, int r, int rt){ LSIDE(rt)=l;RSIDE(rt)=r;SIGN(rt)=0; if (l==r){ DATA(rt)=val[pos[l]]; return ; } int mid=l+r>>1; buildtree(l,mid,LSON); buildtree(mid+1,r,RSON); pushup(rt); } void update( int l, int r, long long c, int rt){ if (l<=LSIDE(rt)&&RSIDE(rt)<=r){ SIGN(rt)+=c; DATA(rt)+=c*WIDTH(rt); return ; } pushdown(rt); int mid=LSIDE(rt)+RSIDE(rt)>>1; if (l<=mid)update(l,r,c,LSON); if (mid<r)update(l,r,c,RSON); pushup(rt); } long long query( int l, int r, int rt){ long long ans=0; if (l<=LSIDE(rt)&&RSIDE(rt)<=r) return DATA(rt); pushdown(rt); int mid=LSIDE(rt)+RSIDE(rt)>>1; if (l<=mid)ans+=query(l,r,LSON); if (mid<r)ans+=query(l,r,RSON); return ans; } int LCA( int x, int y){ while (top[x]!=top[y]){ if (deep[top[x]]<deep[top[y]])swap(x,y); x=fa[top[x]]; } if (deep[x]>deep[y])swap(x,y); return x; } int check( int x){ if (x==root) return -1; if (id[x]<=id[root]&&id[root]<=id[x]+size[x]-1){ int y=root; while (deep[y]>deep[x]){ if (fa[top[y]]==x) return top[y]; y=fa[top[y]]; } return son[x]; } return 0; } void update_subtree( int x, int k){ int y=check(x); if (y==-1)update(1,n,k,1); else if (y==0)update(id[x],id[x]+size[x]-1,k,1); else { update(1,n,k,1); update(id[y],id[y]+size[y]-1,-k,1); } } void query_subtree( int x){ long long s; int y=check(x); if (y==-1)s=query(1,n,1); else if (y==0)s=query(id[x],id[x]+size[x]-1,1); else s=query(1,n,1)-query(id[y],id[y]+size[y]-1,1); printf ( "%lld\n" ,s); } void work(){ int f,x,y,k; while (m--){ f=read();x=read(); if (f==1)root=x; else if (f==2){ y=read();k=read(); int lca=LCA(x,y),lca1=LCA(x,root),lca2=LCA(y,root); if (deep[lca1]>deep[lca])lca=lca1; if (deep[lca2]>deep[lca])lca=lca2; update_subtree(lca,k); } else query_subtree(x); } } void init(){ int x,y; n=read();m=read(); for ( int i=1;i<=n;i++)val[i]=read(); for ( int i=1;i<n;i++){ x=read();y=read(); add(x,y); } deep[1]=1; dfs1(1); dfs2(1,1); buildtree(1,n,1); } int main(){ init(); work(); return 0; } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
· 手把手教你在本地部署DeepSeek R1,搭建web-ui ,建议收藏!
· Spring AI + Ollama 实现 deepseek-r1 的API服务和调用
· 数据库服务器 SQL Server 版本升级公告
· C#/.NET/.NET Core技术前沿周刊 | 第 23 期(2025年1.20-1.26)
· 程序员常用高效实用工具推荐,办公效率提升利器!