社区讨论

求助,悬赏三关

P3313[SDOI2014] 旅行参与者 1已保存回复 0

讨论操作

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

当前回复
0 条
当前快照
1 份
快照标识符
@lo29eixm
此快照首次捕获于
2023/10/23 10:09
2 年前
此快照最后确认于
2023/11/03 10:21
2 年前
查看原帖
样例过了,WA23pts
CPP
#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
struct city{
	int xy,w;
}dfn[N];
struct node{
	int fa,dep,size,maxson,top,id;
	city w;
	vector<int>e;
}tr[N];
int tot=0;
void dfs1(int x,int fa,int dep){
	tr[x].fa=fa,tr[x].dep=dep,tr[x].size=1;
	int maxx=-1;
	for(auto it:tr[x].e){
		if(it==fa) continue;
		dfs1(it,x,dep+1);
		tr[x].size+=tr[it].size;
		if(tr[it].size>maxx) maxx=tr[it].size,tr[x].maxson=it;
	}
}
void dfs2(int x,int top){
	tr[x].top=top,dfn[++tot]=tr[x].w,tr[x].id=tot;
	if(!tr[x].maxson) return;
	dfs2(tr[x].maxson,top);
	for(auto it:tr[x].e){
		if(it==tr[x].maxson||it==tr[x].fa) continue;
		dfs2(it,it);
	}
}
namespace XDS{
	struct node{
		int sum,maxx;
		int ls,rs;
	}tr[N<<6];
	int tot=0;
	inline void pushup(int x){
		tr[x].sum=tr[tr[x].ls].sum+tr[tr[x].rs].sum;
		tr[x].maxx=max(tr[tr[x].ls].maxx,tr[tr[x].rs].maxx);
	}
	void insert(int &x,int l,int r,int k){
		city p=dfn[k];
		if(!x) x=++tot;
		if(l==r){
			tr[x]={p.w,p.w,0,0};
			return;
		}
		int mid=(l+r)/2;
		if(k<=mid) insert(tr[x].ls,l,mid,k);
		else insert(tr[x].rs,mid+1,r,k);
		pushup(x);
	}
	void del(int x,int l,int r,int k){
		if(!x) return;
		if(l==r){
			tr[x]={0,0,0,0};
			return;
		}
		int mid=(l+r)/2;
		if(k<=mid) del(tr[x].ls,l,mid,k);
		else del(tr[x].rs,mid+1,r,k);
		pushup(x);
	}
	int query_sum(int x,int l,int r,int ql,int qr){
		if(!x) return 0;
		if(ql<=l&&r<=qr) return tr[x].sum;
		int mid=(l+r)/2,sum=0;
		if(ql<=mid) sum+=query_sum(tr[x].ls,l,mid,ql,qr);
		if(qr>mid) sum+=query_sum(tr[x].rs,mid+1,r,ql,qr);
		return sum;
	}
	int query_max(int x,int l,int r,int ql,int qr){
		if(!x) return 0;
		if(ql<=l&&r<=qr) return tr[x].maxx;
		int mid=(l+r)/2,maxx=-0x3f3f3f3f;
		if(ql<=mid) maxx=max(maxx,query_max(tr[x].ls,l,mid,ql,qr));
		if(qr>mid) maxx=max(maxx,query_max(tr[x].rs,mid+1,r,ql,qr));
		return maxx;
	}
	void change(int x,int l,int r,int k,int w){
		if(l==r){
			tr[x].sum=tr[x].maxx=w;
			return;
		}
		int mid=(l+r)/2;
		if(k<=mid) change(tr[x].ls,l,mid,k,w);
		else change(tr[x].rs,mid+1,r,k,w);
		pushup(x);
	}
}
int root[N];
int n,q;
int main(){
	cin>>n>>q;
	for(int i=1;i<=n;i++) cin>>tr[i].w.w>>tr[i].w.xy;
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		tr[x].e.push_back(y);
		tr[y].e.push_back(x);
	}
	dfs1(1,1,1);
	dfs2(1,1);
	for(int i=1;i<=n;i++){
		XDS::insert(root[dfn[i].xy],1,n,i);
	}
	while(q--){
		string opt;
		cin>>opt;
		int x,y;
		cin>>x>>y;
		if(opt=="CC"){
			XDS::del(root[tr[x].w.xy],1,n,tr[x].id);
			XDS::insert(root[y],1,n,tr[x].id);
			tr[x].w.xy=y;
		}
		if(opt=="CW"){
			XDS::change(root[tr[x].w.xy],1,n,tr[x].id,y);
		}
		if(opt=="QS"){
			int xy=tr[x].w.xy;
			int sum=0;
			while(tr[x].top!=tr[y].top){
				if(tr[tr[x].top].dep<tr[tr[y].top].dep) swap(x,y);
				int tx=tr[x].top;
				sum+=XDS::query_sum(root[xy],1,n,tr[tx].id,tr[x].id);
				x=tr[tx].fa;
			}
			if(tr[x].dep>tr[y].dep) swap(x,y);
			cout<<sum+XDS::query_sum(root[xy],1,n,tr[x].id,tr[y].id)<<endl;
		}
		if(opt=="QM"){
			int xy=tr[x].w.xy;
			int maxx=-0x3f3f3f3f;
			while(tr[x].top!=tr[y].top){
				if(tr[tr[x].top].dep<tr[tr[y].top].dep) swap(x,y);
				int tx=tr[x].top;
				maxx=max(maxx,XDS::query_max(root[xy],1,n,tr[tx].id,tr[x].id));
				x=tr[tx].fa;
			}
			if(tr[x].dep>tr[y].dep) swap(x,y);
			cout<<max(maxx,XDS::query_max(root[xy],1,n,tr[x].id,tr[y].id))<<endl;
		}
	}
	return 0;
}

回复

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

正在加载回复...