QWQ(玄关求调)
  • 板块学术版
  • 楼主wuzhirui
  • 当前回复0
  • 已保存回复0
  • 发布时间2024/1/23 15:34
  • 上次更新2024/1/23 18:01:58
查看原帖
QWQ(玄关求调)
1046964
wuzhirui楼主2024/1/23 15:34
#include<bits/stdc++.h>
using namespace std;
#define int long long 
const int maxn=100000+100;
const int inf=-0x3f3f3f3f3f3f;
int head[maxn],k;
struct node{
	int to,next,id,val;
}e[maxn*2];
void add(int u,int v,int val,int id){
	e[++k].to=v;
	e[k].next=head[u];
	head[u]=k;
	e[k].id=id;
	e[k].val=val;
}
int w[maxn],son[maxn],siz[maxn],f[maxn],dep[maxn],g[maxn];
void dfs1(int u,int fa,int d){
	dep[u]=d;siz[u]=1;f[u]=fa;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(fa==v) continue;
		w[v]=e[i].val;g[v]=e[i].id;
		dfs1(v,u,d+1);
		if(son[u]==0 || siz[son[u]]<siz[v]) son[u]=v;
	}
}
int pos[maxn],a[maxn],b[maxn],top[maxn],cnt;
void dfs2(int u,int tp){
	pos[u]=++cnt;top[u]=tp;a[cnt]=w[u];b[g[u]]=cnt;
	if(son[u]==0) return;
	dfs2(son[u],tp);
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(v!=f[u] && v!=son[u]) dfs2(v,v);
	}
}
struct tree{
	int l,r,ma,cov,add;
}tr[maxn*4];
void build(int id,int l,int r){
	tr[id].l=l;tr[id].r=r;tr[id].cov=inf;tr[id].ma=inf;
	if(l==r){
		tr[id].ma=a[l];
		return ;
	}
	int mid=(l+r)/2;
	build(id*2,l,mid);build(id*2+1,mid+1,r);
	tr[id].ma=max(tr[id*2].ma,tr[id*2+1].ma);
}
void down(int id){
	if(tr[id].cov!=inf){
		tr[id*2].ma=tr[id].cov;
		tr[id*2].cov=tr[id].cov;
		tr[id*2].add=0;
		tr[id*2+1].ma=tr[id].cov;
		tr[id*2+1].cov=tr[id].cov;
		tr[id*2+1].add=0;
		tr[id].cov=inf;
	}
	if(tr[id].add!=0){
		tr[id*2].ma+=tr[id].add;
		tr[id*2+1].ma+=tr[id].add;
		if(tr[id*2].cov!=inf)tr[id*2].cov+=tr[id].add;
		else tr[id*2].add+=tr[id].add;
		if(tr[id*2+1].cov!=inf)tr[id*2+1].cov+=tr[id].add;
		else tr[id*2+1].add+=tr[id].add;
		tr[id].add=0;
	}
}
void update1(int id,int l,int r,int v){
	if(tr[id].r<l || tr[id].l>r) return;
	if(tr[id].l>=l && tr[id].r<=r){
		tr[id].ma=v;
		tr[id].cov=v;
		tr[id].add=0;
		return ;
	}
	if(tr[id].add!=0 || tr[id].cov!=inf) down(id);
	int mid=(tr[id].l+tr[id].r)/2;
	if(mid>=l)update1(id*2,l,r,v);
	if(mid<=r)update1(id*2+1,l,r,v);
	tr[id].ma=max(tr[id*2].ma,tr[id*2+1].ma);
}
void update2(int id,int l,int r,int v){
	if(tr[id].r<l || tr[id].l>r) return;
	if(tr[id].l>=l && tr[id].r<=r){
		tr[id].ma+=v;
		if(tr[id].cov!=-inf) tr[id].cov+=v;
		else tr[id].add+=v;
		return ;
	}
	if(tr[id].add!=0 || tr[id].cov!=inf) down(id);
	int mid=(tr[id].l+tr[id].r)/2;
	if(mid>=l)update2(id*2,l,r,v);
	if(mid<=r)update2(id*2+1,l,r,v);
	tr[id].ma=max(tr[id*2].ma,tr[id*2+1].ma);
}
int select(int id,int l,int r){
	if(tr[id].l>r || tr[id].r<l) return inf;
	if(tr[id].l>=l && tr[id].r<=r) return tr[id].ma; 
	if(tr[id].cov!=inf || tr[id].add!=0) down(id);
	return max(select(id*2,l,r),select(id*2+1,l,r));
}
void change1(int x,int y,int z){
	while(top[x]!=top[y]){
	if(dep[top[x]]<dep[top[y]]) swap(x,y);
		update1(1,pos[top[x]],pos[x],z);
		x=f[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	update1(1,pos[x]+1,pos[y],z);
}
void change2(int x,int y,int z){
	while(top[x]!=top[y]){
	if(dep[top[x]]<dep[top[y]]) swap(x,y);
		update2(1,pos[top[x]],pos[x],z);
		x=f[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	update2(1,pos[x]+1,pos[y],z);
}
int ask(int x,int y){
	long long ans=inf;
	while(top[x]!=top[y]) {
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		ans=max(select(1,pos[top[x]],pos[x]),ans);
		x=f[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	if(x!=y) ans=max(ans,select(1,pos[x]+1,pos[y]));
	return ans;
}
signed main(){
	int n;
	cin>>n;
	for(int i=1;i<=n-1;i++){
		int u,v,wd;
		cin>>u>>v>>wd;
		add(u,v,wd,i);add(v,u,wd,i);
	}
	dfs1(1,1,1);
	k=0,cnt=0;
	dfs2(1,1);
	build(1,1,n);
	for(int i=1;i;i++){
		string opt;
		cin>>opt;
		if(opt=="Change"){
			int s,wk;
			cin>>s>>wk;
			update1(1,b[s],b[s],wk);
		}
		if(opt=="Cover"){
			int u,v,k;
			cin>>u>>v>>k;
			change1(u,v,k);
		}
		if(opt=="Add"){
			int u,v,k;
			cin>>u>>v>>k;
			change2(u,v,k);
		}
		if(opt=="Max"){
			int u,v;
			cin>>u>>v;
			cout<<ask(u,v)<<endl;
		}
		if(opt=="Stop"){
			break;
		}
	}
	return 0;
}

样例过,10分(P4315)

PS:线段树update,select那几排写法不同,但是对的

2024/1/23 15:34
加载中...