树形dp--luoguP1351 NOIP2014联合权值

水题···简单的树形dp

用f[i][0]记录i的儿子w值得和，f[i][1]记录i的儿子中最大的w值

注意有两种转移方式，一种是i和它的孙子，一种是i的儿子们互相联合

因为忘了考虑第二种wa两次···

具体看代码把

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define maxn 200005
#define LL long long
using namespace std;
int n,cnt,head[maxn];
LL mx,sum,f[maxn][2],w[maxn];
const int mod=10007;

inline int rd(){
	int x=0,f=1;char c=' ';
	while(c<'0' || c>'9') {if(c=='-')f=-1;c=getchar();}
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*f;
}

struct EDGE{
	int to,nxt;
}edge[maxn<<1];

inline void add(int x,int y){
	edge[++cnt].to=y; edge[cnt].nxt=head[x]; head[x]=cnt;
}

inline void dfs(int u,int fa){
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(v==fa) continue;
		dfs(v,u);
		mx=max(mx,max(f[v][1]*w[u],f[u][1]*w[v]));
		(sum+=w[u]*f[v][0]+w[v]*f[u][0])%=mod;
		(f[u][0]+=w[v])%=mod; f[u][1]=max(f[u][1],w[v]);
	} return;
}

int main(){
	n=rd();
	for(int i=1;i<n;i++){
		int x=rd(),y=rd();
		add(x,y); add(y,x);
	}
	for(int i=1;i<=n;i++) w[i]=rd();
	dfs(1,0);
	printf("%lld %lld\n",mx,(sum*2)%mod);
	return 0;
}