求助大佬,调了2h依旧全WA,崩溃了
查看原帖
求助大佬,调了2h依旧全WA,崩溃了
315388
dy_iy楼主2023/1/25 13:27
#include<bits/stdc++.h>
using namespace std;
inline int read(){
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
	return x*f;
}
const int maxn=100010;
int n;
int dep[maxn],siz[maxn],fa[maxn],son[maxn],w[maxn];
int cnt,id[maxn],wt[maxn],top[maxn];
int tot,head[maxn<<1];
long long tree[maxn<<2],lzy[maxn<<2],tag[maxn<<2];
struct node{
	int to,next,length;
}bian[maxn<<1];
void add(int u,int v,int w){
	bian[++tot].to=v;
	bian[tot].length=w;
	bian[tot].next=head[u];
	head[u]=tot;
}
void dfs1(int f,int u,int deep){
	dep[u]=deep;
	siz[u]=1;
	fa[u]=f;
	int maximum=-1;
	for(int i=head[u];i;i=bian[i].next){
		int v=bian[i].to;
		if(v==f)continue;
		w[v]=bian[i].length;
		dfs1(u,v,deep+1);
		siz[u]+=siz[v];
		if(siz[v]>maximum){
			son[u]=v;
			maximum=siz[v];
		}
	}
}
void dfs2(int u,int topf){
	id[u]=++cnt;
	wt[cnt]=w[u];
	top[u]=topf;
	if(!son[u])return;
	dfs2(son[u],topf);
	for(int i=head[u];i;i=bian[i].next){
		int v=bian[i].to;
		if(v==fa[u]||v==son[u])continue;
		dfs2(v,v);
	}
}
void pushup(int x){
	tree[x]=max(tree[x<<1],tree[x<<1|1]);
}
void pushdown(int x){
	if(tag[x]>=0){
		lzy[x<<1]=lzy[x<<1|1]=0;
		tree[x<<1]=tree[x<<1|1]=tag[x<<1]=tag[x<<1|1]=tag[x];
		tag[x]=-1;
	}
	if(lzy[x]){
		tree[x<<1]+=lzy[x];
		tree[x<<1|1]+=lzy[x];
		lzy[x<<1]+=lzy[x];
		lzy[x<<1|1]+=lzy[x];
		lzy[x]=0;
	}
}
void build(int x,int l,int r){
	tag[x]=-1;
	if(l==r){
		tree[x]=wt[l];
		return;
	}
	int mid=(l+r)>>1;
	build(x<<1,l,mid);
	build(x<<1|1,mid+1,r);
	return;
}
void change(int x,int l,int r,int ql,int qr,int k){
	if(l>=ql&&r<=qr){
		tree[x]=tag[x]=k;
		lzy[x]=0;
		return;
	}
	pushdown(x);
	int mid=(l+r)>>1;
	if(mid>=ql)change(x<<1,l,mid,ql,qr,k);
	if(mid<qr)change(x<<1|1,mid+1,r,ql,qr,k);
	pushup(x);
}
void modify(int x,int l,int r,int ql,int qr,int k){
	if(l>=ql&&r<=qr){
		tree[x]+=k;
		lzy[x]+=k;
		return;
	}
	pushdown(x);
	int mid=(l+r)>>1;
	if(mid>=ql)modify(x<<1,l,mid,ql,qr,k);
	if(mid<qr)modify(x<<1|1,mid+1,r,ql,qr,k);
	pushup(x);
}
int query(int x,int l,int r,int ql,int qr){
	if(l>=ql&&r<=qr){
		return tree[x];
	}
	int res=0;
	int mid=(l+r)>>1;
	pushdown(x);
	if(mid>=ql)res=max(res,query(x<<1,l,mid,ql,qr));
	if(mid<qr)res=max(res,query(x<<1|1,mid+1,r,ql,qr));
	return res;
}
void cover(int x,int y,int k){
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		change(1,1,n,id[top[x]],id[x],k);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	change(1,1,n,id[x]+1,id[y],k);
}
void addrange(int x,int y,int k){
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		modify(1,1,n,id[top[x]],id[x],k);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	modify(1,1,n,id[x]+1,id[y],k);
}
int querymax(int x,int y){
	int ans=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		ans=max(ans,query(1,1,n,id[top[x]],id[x]));
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	ans=max(ans,query(1,1,n,id[x]+1,id[y]));
	return ans;
}
int main(){
	n=read();
	for(int i=1;i<n;i++){
		int u,v,w;
		u=read();v=read();w=read();
		add(u,v,w);
		add(v,u,w);
	}
	dfs1(0,1,1);
	dfs2(1,1);
	build(1,1,n);
	string s;
	while(cin>>s&&s[0]!='S'){
		int x,y,k;
		if(s[0]=='M'){
			x=read();y=read();
			printf("%d\n",querymax(x,y));
		}else if(s[0]=='A'){
			x=read();y=read();k=read();
			addrange(x,y,k);
		}else if(s[1]=='h'){
			x=read();k=read();
			if(dep[bian[x*2-1].to]<dep[bian[x*2].to])x=bian[x<<1].to;
			else x=bian[x*2-1].to;
			change(1,1,n,id[x],id[x],k);
		}else if(s[1]=='o'){
			x=read();y=read();k=read();
			cover(x,y,k);
		}
	}
	return 0;
}
2023/1/25 13:27
加载中...