BZOJ5379: Tree

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;
}

 

posted @   符拉迪沃斯托克  阅读(203)  评论(0编辑  收藏  举报
编辑推荐:
· 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)
· 程序员常用高效实用工具推荐,办公效率提升利器!
Live2D
欢迎阅读『BZOJ5379: Tree』
点击右上角即可分享
微信分享提示