CFD求调,WAon3
  • 板块学术版
  • 楼主快乐的大童
  • 当前回复2
  • 已保存回复2
  • 发布时间2023/4/9 00:09
  • 上次更新2023/10/23 18:59:47
查看原帖
CFD求调,WAon3
448884
快乐的大童楼主2023/4/9 00:09

RT,思路大概是每次操作二就暴力更换重儿子、本身、父亲的子树大小、子树和、重儿子和父亲等信息。不知道为什么 WA 了。

睡觉去了,明天再看。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<map>
#include<unordered_map>
#include<vector>
#include<queue>
#include<bitset>
#include<set>
#include<ctime>
#include<random>
#define x1 xx1
#define y1 yy1
#define IOS ios::sync_with_stdio(false)
#define ITIE cin.tie(0);
#define OTIE cout.tie(0);
#define PY puts("Yes")
#define PN puts("No")
#define PW puts("-1")
#define popcount __builtin_popcount
#define pii pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define rep(a,b,c) for(int a=b;a<=c;a++)
#define per(a,b,c) for(int a=b;a>=c;a--)
#define reprange(a,b,c,d) for(int a=b;a<=c;a=d)
#define perrange(a,b,c,d) for(int a=b;a>=c;a=d)
#define graph(i,j,k,l) for(int i=k[j];i;i=l[i].nxt)
#define lowbit(x) (x&-x)
#define lson(x) (x<<1)
#define rson(x) (x<<1|1)
#define int long long
//#define int __int128
using namespace std;
inline int R(){
	int x=0,f=1;int ch=getchar();
	while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
	while(isdigit(ch)){x=x*10+ch-48;ch=getchar();}return x*f;
}
inline void write(int x){
	if(x<0){x=-x;putchar('-');}
	int y=0;char z[70];
	while(x||!y){z[y++]=x%10+48;x/=10;}
	while(y--)putchar(z[y]);
}
inline void writesp(int x){
	write(x);putchar(32);
}
inline void writeln(int x){
	write(x);putchar(10);
}
const int maxn=1e6+5,maxm=2e6+5;
int n,m,a[maxn];
struct edge{
	int to,nxt,w;
}e[maxm];
int head[maxn],edges;
void add(int x,int y,int z=0){
	e[++edges]=(edge){y,head[x],z};
	head[x]=edges;
}
int fa[maxn],siz[maxn],son[maxn],sum[maxn];
typedef struct{
	bool operator()(const int &x,const int &y){
		if(siz[x]!=siz[y]) return siz[x]>siz[y];
		else return x<y;
	}
}cmp;
set<int,cmp>v[maxn];
void dfs(int x,int y){
	sum[x]=a[x],siz[x]=1,fa[x]=y;
	graph(i,x,head,e){
		int u=e[i].to;
		if(u==y) continue;
		dfs(u,x); 
		sum[x]+=sum[u],siz[x]+=siz[u];
		v[x].insert(u);
	}
	if(!v[x].empty()) son[x]=*v[x].begin();
	else son[x]=0;
}
void solve_the_problem(){
	n=R(),m=R();
	rep(i,1,n)a[i]=R();
	rep(i,1,n-1){
		int x=R(),y=R();
		add(x,y),add(y,x);
	}
	dfs(1,0);
	rep(i,1,m){
		int op=R(),x=R();
		if(op==1){
			writeln(sum[x]);
		}else{
			if(!son[x]) continue;
			int sonx=son[x],fafa=fa[x]; 
			int sumx=sum[x],sizx=siz[x];
			sum[x]-=sum[sonx],sum[sonx]=sumx;
			siz[x]-=siz[sonx],siz[sonx]=sizx;
			fa[x]=sonx,fa[sonx]=fafa;
			v[x].erase(sonx);
			v[sonx].insert(x);
			v[fafa].erase(x);
			v[fafa].insert(sonx);
			son[sonx]=*v[sonx].begin();
			if(!v[x].empty()) son[x]=*v[x].begin();
			else son[x]=0;
			son[fafa]=*v[fafa].begin();
		}
	}
}
signed main(){int _=1;while(_--)solve_the_problem();}

2023/4/9 00:09
加载中...