关于我用普通tree dp水过了一道基环树dp的那档事

【luogu】P3796

注明：这是非正式题解，是个人非最优的玄学做法，仅供参考~

关于标签，基环树，我不知道怎么打，于是想到：特判！

所以多用了一下并查集~~~

#include<bits/stdc++.h>
#define maxn 1000001
using namespace std;
int fa[maxn],dp[maxn][2][2],a[maxn];
int n,root,range,len,last[maxn];
struct edge{int x,y,next;}p[maxn];
int find_father(int x)
{
	if(fa[x]!=x)fa[x]=find_father(fa[x]);
	return fa[x];
}
bool dg(char ch)
{
	if(ch<'0'||ch>'9')return true;
	return false;
}
int read()
{
	int r=0,w=1;char ch;ch=getchar();
	while(dg(ch)){if(ch=='-')w=-w;ch=getchar();}
	while(!dg(ch)){r=r*10+ch-'0';ch=getchar();}
	return r*w;
}
void ins(int x,int y)
{
	len++;
	p[len].x=x;p[len].y=y;
	p[len].next=last[x];last[x]=len;
}
void tr_dp(int x,int father)
{
	int one=0,two=0,three=0,four=0;
	for(int k=last[x];k;k=p[k].next)
	{
		int y=p[k].y;
		if(y!=father)
		{
			tr_dp(y,x);
			if(x!=range&&x!=root)
			{
			 	one+=dp[y][0][0];//1,0
			 	two+=max(dp[y][1][0],dp[y][0][0]);//0,0
			 	three+=dp[y][0][1];//1,1
			 	four+=max(dp[y][1][1],dp[y][0][1]);//0,1
			}
			else if(x==range)
			{
				two+=max(dp[y][1][0],dp[y][0][0]);
			 	three+=dp[y][0][1];
			}
			else
			{
				one+=dp[y][0][0];
				two+=max(dp[y][1][0],dp[y][0][0]);//0,0
				four+=max(dp[y][1][1],dp[y][0][1]);//0,1
			}
		}
	}
	if(x!=range)dp[x][1][0]=one+a[x];
	dp[x][0][0]=two;
	if(x!=root)dp[x][1][1]=three+a[x];
	if(x!=range)dp[x][0][1]=four;
	
}
int main()
{
	n=read();
	for(int i=1;i<=n;i++)a[i]=read();
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=n;i++)
	{
		int x,y;
		x=read();y=read();
		x++;y++;
		int fx=find_father(fa[x]),fy=find_father(fa[y]);
		if(fx==fy)
		{
			root=x;
			range=y;
		}
		else
		{
			ins(x,y);
			ins(y,x);
			fa[fx]=fy;
		}
	}
	tr_dp(root,0);
	double k;
	scanf("%lf",&k);
	double ans=((double)max(dp[root][1][0],dp[root][0][1]));
	ans=max((double)dp[root][0][0],ans);
	printf("%.1lf",ans*k);
	return 0;
}