- linzhixing2024's blog
这辈子再也不乱封装了
- 2025-8-29 22:38:01 @
哪个傻逼天才玩树剖把树给封装了啊!!
这辈子再也不乱封装了呜呜呜~~~
P3979 遥远的国度
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn=1e5+5;
#define lson (u<<1)
#define rson (u<<1|1)
#define mid ((l+r)>>1)
const int INF=2147483648;
int a[maxn];
int m;
struct tree{
int root;
int n,tot;
int dep[maxn],fa[maxn],sz[maxn],wc[maxn],top[maxn],dfn[maxn],rdfn[maxn];
vector<int> e[maxn];
void dfs1(int u,int p){
sz[u]=1,fa[u]=p,dep[u]=dep[p]+1;
for(auto v:e[u]){
if(v==p)continue;
dfs1(v,u);
sz[u]+=sz[v];
if(sz[v]>sz[wc[u]])wc[u]=v;
}
}
void dfs2(int u,int tp){
top[u]=tp;
dfn[u]=++tot;
rdfn[tot]=u;
if(wc[u])dfs2(wc[u],tp);
for(auto v:e[u]){
if(v!=fa[u]&&v!=wc[u]){
dfs2(v,v);
}
}
}
}tr;
struct segtree{
struct treenode{
int l,r;
int mn,tag;
}tre[maxn<<2];
void pushup(int u){
tre[u].mn=min(tre[lson].mn,tre[rson].mn);
}
void build(int u,int l,int r){
tre[u].l=l,tre[u].r=r;
if(l==r){
tre[u].mn=a[tr.rdfn[l]];
return;
}
build(lson,l,mid);
build(rson,mid+1,r);
pushup(u);
}
void pushdown(int u){
if(tre[u].tag){
tre[lson].tag=tre[rson].tag=tre[u].tag;
tre[lson].mn=tre[rson].mn=tre[u].tag;
tre[u].tag=0;
}
}
int query(int u,int L,int R){
int l=tre[u].l,r=tre[u].r;
if(L>r||R<l)return INF;
if(L<=l&&r<=R)return tre[u].mn;
pushdown(u);
return min(query(lson,L,R),query(rson,L,R));
}
void update(int u,int L,int R,int x){
int l=tre[u].l,r=tre[u].r;
if(L>r||R<l)return;
if(L<=l&&r<=R){tre[u].mn=tre[u].tag=x;return;}
pushdown(u);
update(lson,L,R,x);
update(rson,L,R,x);
pushup(u);
}
}seg;
void trupdate(int u,int v,int x){
while(tr.top[u]!=tr.top[v]){
if(tr.dep[tr.top[u]]<tr.dep[tr.top[v]])swap(u,v);
seg.update(1,tr.dfn[tr.top[u]],tr.dfn[u],x);
u=tr.fa[tr.top[u]];
}
if(tr.dep[u]>tr.dep[v])swap(u,v);
seg.update(1,tr.dfn[u],tr.dfn[v],x);
}
int getp(int u){
if(u==tr.root)return -1;
if(tr.dfn[u]>=tr.dfn[tr.root]||tr.dfn[u]+tr.sz[u]-1<tr.dfn[tr.root])return 0;
int now=tr.root;
while(tr.top[now]!=tr.top[u]){
if(tr.fa[tr.top[now]]==u)return tr.top[now];
now=tr.fa[tr.top[now]];
}
return tr.wc[u];
}
int trquery(int u){
int pl=getp(u);
if(pl==-1)return seg.tre[1].mn;
if(pl==0)return seg.query(1,tr.dfn[u],tr.dfn[u]+tr.sz[u]-1);
else{
int ans=seg.query(1,1,tr.dfn[pl]-1);
if(tr.dfn[pl]+tr.sz[pl]-1!=tr.n)ans=min(ans,seg.query(1,tr.dfn[pl]+tr.sz[pl],tr.n));
return ans;
}
}
signed main(){
cin>>tr.n>>m;
for(int i=1,u,v;i<tr.n;i++){
cin>>u>>v;
tr.e[u].push_back(v);
tr.e[v].push_back(u);
}
for(int i=1;i<=tr.n;i++){
cin>>a[i];
}
tr.dfs1(1,0);
tr.dfs2(1,1);
seg.build(1,1,tr.n);
cin>>tr.root;
for(int i=1,op;i<=m;i++){
cin>>op;
if(op==1)cin>>tr.root;
if(op==2){
int u,v,x;
cin>>u>>v>>x;
trupdate(u,v,x);
}if(op==3){
int u;
cin>>u;
cout<<trquery(u)<<endl;
}
}
return 0;
}