• [codeforces274b]Zero Tree(树形dp)


    题意:给出一棵树,每个点有权值,每次操作可以对一个联通子集中的点全部加1,或者全部减1,且每次操作必须包含点1,问最少通过多少次操作可以让整棵树每个点的权值变为0.

    解题关键:自底向上dp,记录up,down两个数组 代表u被加的次数和减的次数,以1为根,则

    $up[u] = max (up[v])$

    $down[u] = max (down[v])$

    而子树确定,该节点改变的次数也就确定了。从而推出该点的up和down的影响,至于为什么取max,因为左右子树可以互相影响,只要包含根节点即可。

     1 #include<cstdio>
     2 #include<cstring>
     3 #include<algorithm>
     4 #include<cstdlib>
     5 #include<cmath>
     6 #include<iostream>
     7 using namespace std;
     8 typedef long long ll;
     9 const int maxn=1e6+3;
    10 const int inf=0x3f3f3f3f;
    11 int head[maxn],tot,n,m,sum;
    12 ll cnt[maxn];
    13 ll up[maxn],down[maxn];
    14 struct edge{
    15     int to;
    16     int nxt;
    17     int w;
    18 }e[maxn<<2];
    19 void add_edge(int u,int v){
    20     e[tot].to=v;
    21     e[tot].nxt=head[u];
    22     head[u]=tot++;
    23 }
    24 
    25 void dfs1(int u,int fa){
    26     for(int i=head[u];i!=-1;i=e[i].nxt){
    27         int v=e[i].to;
    28         if(v==fa) continue;
    29         dfs1(v,u);
    30         up[u]=max(up[u],up[v]);
    31         down[u]=max(down[u],down[v]);
    32     }
    33     cnt[u]+=up[u]-down[u];
    34     if(cnt[u]>0) down[u]+=cnt[u];
    35     else up[u]-=cnt[u];
    36 }
    37 inline int read(){
    38     char k=0;char ls;ls=getchar();for(;ls<'0'||ls>'9';k=ls,ls=getchar());
    39     int x=0;for(;ls>='0'&&ls<='9';ls=getchar())x=(x<<3)+(x<<1)+ls-'0';
    40     if(k=='-')x=0-x;return x;
    41 }
    42 
    43 int main(){
    44     int k=0;
    45     while(scanf("%d",&n)!=EOF){
    46         memset(head,-1,sizeof head);
    47         tot=0;
    48         int a,b;
    49         for(int i=0;i<n-1;i++){
    50             a=read();
    51             b=read();
    52             add_edge(a,b);
    53             add_edge(b,a);
    54         }
    55         for(int i=1;i<=n;i++) scanf("%lld",&cnt[i]);
    56         dfs1(1,-1);
    57         printf("%lld
    ",up[1]+down[1]);
    58     }
    59     return 0;
    60 }
  • 相关阅读:
    学生管理系统(2:添加主界面窗口)
    Qfile
    QButtonGroup
    comboBox
    QLineEdit
    QMessageBox
    实现简单的计算器(控制器代码)
    实现简单的计算器(计算功能模块实现)
    网络编程基础【HTML编程】
    网络编程基础【正则表达式】
  • 原文地址:https://www.cnblogs.com/elpsycongroo/p/7455447.html
Copyright © 2020-2023  润新知