CSUST 2036 随刄而行 ( 树 dp )

链接： 随刄而行

题意：
给一棵有根树，每个节点有一个权值，定义树的完美程度为节点的权值和，可以花费 m 的代价砍掉一颗子树，求最大完美程度。
思路：
大概就是 dp[ i ] [ j ] （ j=0 || j=1）表示选和不选 i 节点，选 i 节点 又可以由 选不选子节点转移，而如果不选 i 节点 那么它的权值即为 -m（就是直接把它砍掉，和子节点无关了） ,然后就愉快的 wa 了 ，注意数据范围，m有可能是 负数 （这里就比较坑），所以我在砍 i 节点的子树时 ，可以一个节点一个节点的砍，那么就能得到最大的价值。

代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>
#include<cmath>
#include<string>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=3e6+7;
ll T,val[maxn],head[maxn],num,n,m,dp[maxn][3],siz[maxn];
struct node{
       int to,next;
}e[maxn];
void add(int u,int v){
     e[num].next=head[u];
     e[num].to=v;
     head[u]=num++;
}
void dfs1(int u,int pre){
      siz[u]=1;
      for(int i=head[u];i!=-1;i=e[i].next){
          int to=e[i].to;
          if(to==pre) continue;
          dfs1(to,u);
          siz[u]+=siz[to];
      }
}
void dfs2(int u,int pre){
      dp[u][1]=val[u],dp[u][0]=max(-m,-siz[u]*m);
      for(int i=head[u];i!=-1;i=e[i].next){
          int to=e[i].to;
          if(to==pre) continue;
          dfs2(to,u);
          dp[u][1]+=max(dp[to][0],dp[to][1]);
      }
}
int main (){
    ios_base::sync_with_stdio(0);cin.tie(0);
    cin>>T;
    while(T--){
        cin>>n>>m;
        memset(head,-1,sizeof(head));num=0;
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++) cin>>val[i];
        for(int i=0,u,v;i<n-1;i++){
            cin>>u>>v;
            add(u,v);
            add(v,u);
        }
        dfs1(1,-1);dfs2(1,-1);
        cout<<max(dp[1][1],dp[1][0])<<endl;
    }
}