CODEFORCES 274B Zero Tree

题目:给你一棵树,每个节点都有权值,你可以实行下面操作

选一条包含1节点的子树,对所有点的权值+1或-1

问:实行多少次操作才能使得整棵树每个点权值都为0

分析:动态dp,dfs

样例:

Input
3
1 2
1 3
1 -1 1
Output
3
 
   
#include 
#include 
#include 
#include 
using namespace std;
#define MAXN 100050
#define LL long long
vector mp[MAXN];
LL dp[MAXN][2];
LL w[MAXN];
void dfs(int now,int pre)
{
    LL maxi=0,maxd=0;
    for(unsigned int i = 0;i0)
        dp[now][0]+=w[now];
    else
        dp[now][1]+= (-w[now]);
}
int main()
{
   int n;
   cin >> n;

       LL x,y;
       for(int i = 1 ; i < n; i++)
       {
           cin >> x >> y;
           mp[x].push_back(y);
           mp[y].push_back(x);
       }
       for(int i = 1;i<=n;i++)
       {
           cin >> w[i];

       }
       dfs(1,0);
       cout << dp[1][0]+dp[1][1]<



你可能感兴趣的:(求解策略:动态规划)