2021南京站 铜牌题 H - Crystalfly

赛季就要开始了

今天下午队内训练了一场21年的南京，因为要打的就是南京所以挑的（应该）。

大概1点半做到5点不到，大概三个多小时，做了三道题，卡了一个多小时的H，而且罚时多，看来要拿铜还要多努力。

赛后补了这道H题，据说是铜牌题，会了就能拿铜。

题意:有一棵树,每一个节点都有一个值a[i]表示这个点有多少只晶蝶,t[i]表示假设你到达该点连接的点之后,经过多少秒该点的晶蝶都会跑光(范围1~3).问最多可以抓多少晶蝶

做法：树形dp

分类：1.不考虑3的情况；

           2.考虑3的情况；

难度主要在dp的形式和t的值为3的考虑上。如果只有1、2，没有3的话，我们就只用一个dp[u]，表示不选当前节点的值时能得到的最大值。
为什么要这样设？
因为：
如果我按平常的g[u]表示到当前节点（不管选不选）的最大值，发现转移需要子节点中拿一个上去，显然转移不了。
但如果不去选当前节点的话，g[v]就会不包含子节点的值，这样for一遍,拿一个max节点就可以进行转移。

之后考虑3这个节点，我们让sum数组表示子节点的g之和。

这也就意味着，g[i]=max（sum[i]+最大的a[i]）

然后，我们想的还是以u为参照，v里面多拿一个（或者说回来多拿一个），此时的转移方程式有点复杂：

v表示t不是3的点，v2表示t是3的点。

g[u]=max(g[u],sum[u]-g[v]+sum[v]+a[v]+a[v2]);

写这么麻烦，因为此时不能用g[v]表示v的转移了，因为选完v再去v2不能选v的所有结点，应该用sum，这也就是sum[u]-g[v]+sum[v]（仔细理解一下）

然后再加上两个点的值即可。（话有点抽象，结合意思画画图就可以懂了）

之后我们需要用set的有序功能，找到最大的a[v2]（这样才能遍历的时候获得最大值）。

代码如下：

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
//#define double long double
#define pii pair<int,int>
#define pb push_back
#define rep(i,l,r) for (int i=l;i<=r;i++)
#define nep(i,r,l) for (int i=r;i>=l;i--)
#define CIO std::ios::sync_with_stdio(false);
#define cio cin.tie(0);cout.tie(0);
#define pc putchar
using namespace std;
const int INF=0x3f3f3f3f;
const int LINF=0x3f3f3f3f3f3f3f3f;
const double eps=1e-6;
const int N=2e5+5;
const double pi=acos(-1.0);
int g[N]; 
int a[N],t[N],sum[N];
vector<int> ve[N];
void dfs(int u,int fa){
	g[u]=0; 
	int s=0;
	multiset<int> s3;
	for (int i=0;i<ve[u].size();i++){
		int v=ve[u][i];
		if (v==fa) continue;
		dfs(v,u);
		sum[u]+=g[v];
		s=max(s,a[v]);
		if (t[v]==3) s3.insert(a[v]);
	}
	g[u]=sum[u]+s;
	s3.insert(-INF);
	for (int i=0;i<ve[u].size();i++){
		int v=ve[u][i];
		if (v==fa) continue;
		if (t[v]==3) s3.erase(s3.find(a[v]));
		g[u]=max(g[u],sum[u]-g[v]+sum[v]+a[v]+*s3.rbegin());
		if (t[v]==3) s3.insert(a[v]);
	}
}
void work(){
	int n;cin>>n;
	rep(i,1,n) cin>>a[i];
	rep(i,1,n) cin>>t[i];
	rep(i,1,n-1){
		int u,v;cin>>u>>v;
		ve[u].pb(v);
		ve[v].pb(u);
	}
	dfs(1,0);
	cout<<g[1]+a[1]<<endl;
	rep(i,1,n){
		g[i]=0,sum[i]=0;
		ve[i].clear();
	}
}
signed main(){
	CIO;
	cio;
	//多组样例别忘了清空数组,stl
	int _;cin>>_;while(_--)
	{
		work();
	}
	return 0;
}