MnZn刚学OI一秒,WA0分求调
查看原帖
MnZn刚学OI一秒,WA0分求调
239458
cmach_socket楼主2022/10/26 20:43

感觉挺对的

但是就是不对

#include<stdio.h>
#include<algorithm>
#include<list>
#define MAXN 100005
#define lk k<<1
#define rk k<<1|1
#define PPP puts("FUCK");
using namespace std;
class node{
public:
	int l,r,w,la1,la2;
}t[MAXN<<2];
class nd{
public:
	int y,w,t;
	nd(int yy,int ww){
		y=yy,w=ww;
	}
};
int n,mo,tx,ty,tw,tot,xx[MAXN],yy[MAXN],a[MAXN],rak[MAXN],siz[MAXN],f[MAXN],id[MAXN],d[MAXN],son[MAXN],top[MAXN];
list<nd>g[MAXN];
char c[100];
inline void link(int x,int y,int w){
	g[x].emplace_back(nd(y,w));

}
inline void update(int k){
	t[k].w=max(t[lk].w,t[rk].w);
}
inline void pushdown(int k){
	if(t[k].la2>=0){
		t[lk].w=t[rk].w=t[k].la2;
		t[lk].la2=t[rk].la2=t[k].la2;
		t[lk].la1=t[rk].la1=0;
		t[k].la2=-1;
	}
	if(t[k].la1){
		t[lk].w+=t[k].la1;
		t[rk].w+=t[k].la1;
		t[lk].w+=t[k].la1;
		t[rk].w+=t[k].la1;
		t[k].la1=0;
	}
}
 void build(int k,int l,int r){
	t[k].l=l,t[k].r=r;
	t[k].w=t[k].la1=0;
	t[k].la2=-1;
	if(l==r){
		t[k].w=a[rak[l]];
		return ;
	}
	int mid=(l+r)>>1;
	build(lk,l,mid);
	build(rk,mid+1,r);
	update(k);
}
 void qch(int k,int l,int r,int y){
	if(l<=t[k].l and r>=t[k].r){
		if(mo==1){
			t[k].w=y;
			t[k].la1=0;
			t[k].la2=y;
		}
		else{
			t[k].w+=y;
			t[k].la1+=y;
		}
		return ;
	}
	pushdown(k);
	int mid=(t[k].l+t[k].r)>>1;
	if(l<=mid)qch(lk,l,r,y);
	if(r>mid)qch(rk,l,r,y);	
	update(k);
}
 int qask(int k,int l,int r){
	if(l<=t[k].l and r>=t[k].r){
		return t[k].w;
	}
	pushdown(k);
	int mid=(t[k].l+t[k].r)>>1,ans=0;
	if(l<=mid)ans=max(ans,qask(lk,l,r));
	if(r>mid)ans=max(ans,qask(rk,l,r));
	return ans;
}
 void dfs1(int x,int de,int fa){
	d[x]=de,f[x]=fa,siz[x]=1;
	for(auto it:g[x]){
		if(it.y!=fa){
			a[it.y]=it.w;
			dfs1(it.y,de+1,x);
			siz[x]+=siz[it.y];
			if(siz[son[x]]<siz[it.y]){
				son[x]=it.y;
			}
		}
	}
}
 void dfs2(int x,int t){
	top[x]=t;
	id[x]=++tot;
	rak[tot]=x;
	if(!son[x]){
		return ;
	}
	dfs2(son[x],t);
	for(auto it:g[x]){
		if(it.y!=son[x] and it.y!=f[x]){
			dfs2(it.y,it.y);
		}
	}
}
 int get_ans(int x,int y){
	int ans=0;
	while(top[x]!=top[y]){
		if(d[top[x]]<d[top[y]])swap(x,y);
		ans=max(ans,qask(1,id[top[x]],id[x]));
		x=f[top[x]];
		//printf("%d %d %d %d %d d\n",id[top[x]],id[x],top[x],x,ans);
	}
	if(x==y)return ans;
	if(d[x]<d[y])swap(x,y);
	ans=max(ans,qask(1,id[y]+1,id[x]));
	//printf("%d df\n",ans);
	return ans;
}
 void ch_ans(int x,int y,int w){
	while(top[x]!=top[y]){
		if(d[top[x]]<d[top[y]])swap(x,y);
		qch(1,id[top[x]],id[x],w);
		x=f[top[x]];
	}
	if(x==y)return ;
	if(d[x]<d[y])swap(x,y);
	qch(1,id[y]+1,id[x],w);
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		scanf("%d%d%d",&tx,&ty,&tw);
		xx[i]=tx;
		yy[i]=ty;
		link(tx,ty,tw);
		link(ty,tx,tw);
	}
	
	dfs1(1,1,0);
	dfs2(1,1);
	build(1,1,n);
	
	while(1){
		scanf("%s",c);
		if(c[0]=='S'){
			return 0;
		}
		scanf("%d%d",&tx,&ty);
		if(c[0]=='M'){
			if(tx>ty)swap(tx,ty);
			printf("%d\n",get_ans(tx,ty));
		}
		else if(c[1]=='h'){
			mo=1;
			ch_ans(xx[tx],yy[tx],ty);
		}
		else if(c[0]=='C'){
			if(tx>ty)swap(tx,ty);
			scanf("%d",&tw);
			mo=1;
			ch_ans(tx,ty,tw);
		}
		else if(c[0]=='A'){
			if(tx>ty)swap(tx,ty);
			scanf("%d",&tw);
			mo=0;
			ch_ans(tx,ty,tw);

		}
	}
	return 0;
}
2022/10/26 20:43
加载中...