社区讨论

样例过了,RE求调

P4315月下“毛景树”参与者 1已保存回复 0

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
0 条
当前快照
1 份
快照标识符
@lr7xi7w2
此快照首次捕获于
2024/01/10 23:21
2 年前
此快照最后确认于
2024/01/10 23:35
2 年前
查看原帖
CPP
%:include<bits/stdc++.h>
%:define int long long
%:define double long double
%:define INF INT_MAX
using namespace std;
const int maxn=2e5+5;
int n,q,cnt;
int a[maxn];
int f[maxn],deap[maxn],siz[maxn],son[maxn];
int id[maxn],p[maxn],top[maxn];
vector<pair<int,int> > g[maxn];
int s1[maxn],s2[maxn];
void dfs1(int dq,int fa,int sum){
	f[dq]=fa,deap[dq]=sum,siz[dq]=1;
	int maxson=-1;
	for(int i=0;i<g[dq].size();i++){
		if(g[dq][i].first==fa) continue;
		a[g[dq][i].first]=g[dq][i].second;
		dfs1(g[dq][i].first,dq,sum+1);
		siz[dq]+=siz[g[dq][i].first];
		if(maxson==-1 || siz[g[dq][i].first]>siz[maxson]) maxson=g[dq][i].first,son[dq]=g[dq][i].first;
	}
	return;
}
void dfs2(int dq,int fa){
	id[dq]=++cnt,p[cnt]=a[dq],top[dq]=fa;
	if(!son[dq]) return;
	dfs2(son[dq],fa);
	for(int i=0;i<g[dq].size();i++){
		if(g[dq][i].first==f[dq] || g[dq][i].first==son[dq]) continue;
		dfs2(g[dq][i].first,g[dq][i].first);
	}
	return;
}
struct Tree{
	int val,add,p;
	%:define sum(dq) t[dq].val
	%:define add(dq) t[dq].add
	%:define p(dq) t[dq].p
}t[maxn<<4];
void up_date(int dq){sum(dq)=sum(dq*2)+sum(dq*2+1);}
void build(int dq,int l,int r){
	p(dq)=INF;
	if(l==r){
		sum(dq)=p[l];
		return;
	}
	int mid=(l+r)/2;
	build(dq*2,l,mid);
	build(dq*2+1,mid+1,r);
	up_date(dq);
}
void spread(int dq,int l,int r){
	if(p(dq)!=INF){
		sum(dq*2)=p(dq*2)=p(dq);
		sum(dq*2+1)=p(dq*2+1)=p(dq);
		p(dq)=INF;
		add(dq*2)=add(dq*2+1)=0;
	}
	if(add(dq)){
		sum(dq*2)+=add(dq);
		sum(dq*2+1)+=add(dq);
		add(dq*2)+=add(dq);
		add(dq*2+1)+=add(dq);
		add(dq)=0;
	}
}

void change_cover(int dq,int ll,int rr,int l,int r,int d){
	if(l<=ll && rr<=r){
		sum(dq)=p(dq)=d;
		add(dq)=0;
		return;
	}
	int mid=(ll+rr)/2;
	spread(dq,ll,rr);
	if(l<=mid) change_cover(dq*2,ll,mid,l,r,d);
	if(r>mid) change_cover(dq*2+1,mid+1,rr,l,r,d);
	up_date(dq);
}
void changee_cover(int x,int y,int d){
	while(top[x]!=top[y]){
		if(deap[top[x]]<deap[top[y]]){
			change_cover(1,1,n,id[top[y]],id[y],d);
			y=f[top[y]];
		}
		else{
			change_cover(1,1,n,id[top[x]],id[x],d);
			x=f[top[x]];
		}
	}
	if(deap[x]<deap[y]) change_cover(1,1,n,id[x]+1,id[y],d);
	else change_cover(1,1,n,id[y]+1,id[x],d);
}

void change_add(int dq,int ll,int rr,int l,int r,int d){
	if(l<=ll && rr<=r){
		sum(dq)+=d;
		add(dq)+=d;
		return;
	}
	int mid=(ll+rr)/2;
	spread(dq,ll,rr);
	if(l<=mid) change_add(dq*2,ll,mid,l,r,d);
	if(r>mid) change_add(dq*2+1,mid+1,rr,l,r,d);
	up_date(dq);
}
void changee_add(int x,int y,int d){
	while(top[x]!=top[y]){
		if(deap[top[x]]<deap[top[y]]){
			change_add(1,1,n,id[top[y]],id[y],d);
			y=f[top[y]];
		}
		else{
			change_add(1,1,n,id[top[x]],id[x],d);
			x=f[top[x]];
		}
	}
	if(deap[x]<deap[y]) change_add(1,1,n,id[x]+1,id[y],d);
	else change_add(1,1,n,id[y]+1,id[x],d);
}

int ask(int dq,int ll,int rr,int l,int r){
	if(l<=ll && rr<=r) return sum(dq);
	int res=-INF;
	spread(dq,ll,rr);
	int mid=(ll+rr)/2;
	if(l<=mid) res=max(res,ask(dq*2,ll,mid,l,r));
	if(r>mid) res=max(res,ask(dq*2+1,mid+1,rr,l,r));
	return res;
}
int askk(int x,int y){
	int res=-INF;
	while(top[x]!=top[y]){
		if(deap[top[x]]<deap[top[y]]){
			res=max(res,ask(1,1,n,id[top[y]],id[y]));
			y=f[top[y]];
		}
		else{
			res=max(res,ask(1,1,n,id[top[x]],id[x]));
			x=f[top[x]];
		}
	}
	if(deap[x]<deap[y]) res=max(res,ask(1,1,n,id[x]+1,id[y]));
	else res=max(res,ask(1,1,n,id[y]+1,id[x]));
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	cin>>n;
	for(int i=1;i<n;i++){
		int u,v,w;
		cin>>u>>v>>w;
		g[u].push_back(make_pair(v,w));
		g[v].push_back(make_pair(u,w));
		s1[i]=u,s2[i]=v;
	}
	dfs1(1,0,1);
	dfs2(1,1);
	build(1,1,n);
	while(true){
	    string op;
		int x,y,w;
		cin>>op;
		if(op=="Stop") break;
		if(op=="Change"){
			cin>>x>>w;
			if(deap[s1[x]]<deap[s2[x]]) change_cover(1,1,n,id[s2[x]],id[s2[x]],w);
			else change_cover(1,1,n,id[s1[x]],id[s1[x]],w);
		}
		if(op=="Cover"){
			cin>>x>>y>>w;
			changee_cover(x,y,w);
		}
		if(op=="Add"){
			cin>>x>>y>>w;
			changee_add(x,y,w);
		}
		if(op=="Max"){
			cin>>x>>y;
			cout<<askk(x,y)<<endl;
		}
	}
	return 0;
}

回复

0 条回复,欢迎继续交流。

正在加载回复...