zoukankan      html  css  js  c++  java
  • 【树形动规】HDU 5834 Magic boy Bi Luo with his excited tree

    题目链接:

      http://acm.hdu.edu.cn/showproblem.php?pid=5834

    题目大意:

      一棵N个点的有根树,每个节点有价值ci,每条树边有费用di,节点的值只能取一次,边权每次经过都要扣,问从每一个节点开始走最大能获得的价值。

    题目思路:

      【树形动态规划】

      首先用dfs求出从根1往下走的:节点u往下走最后回到节点u的最大值g[u],节点u往下走最后不回到u的最优值和次优值f[0][u],f[1][u]

      接着考虑一个节点u,除了以上的情况还有可能是往它的父亲方向走,这里就分两种,一种是走父亲那边再回来走自己的子树,还有一种是走自己的子树再回来走父亲那边

      (肯定最后都不会特意回到u,因为边权>0,回到自己不会更优)而这些状态都可以通过dfs里求得f和g推出。

      具体推法我已写在代码注释中,希望没有写错。。

      1 //
      2 //by coolxxx
      3 //#include<bits/stdc++.h>
      4 #include<iostream>
      5 #include<algorithm>
      6 #include<string>
      7 #include<iomanip>
      8 #include<map>
      9 #include<stack>
     10 #include<queue>
     11 #include<set>
     12 #include<bitset>
     13 #include<memory.h>
     14 #include<time.h>
     15 #include<stdio.h>
     16 #include<stdlib.h>
     17 #include<string.h>
     18 //#include<stdbool.h>
     19 #include<math.h>
     20 #define min(a,b) ((a)<(b)?(a):(b))
     21 #define max(a,b) ((a)>(b)?(a):(b))
     22 #define abs(a) ((a)>0?(a):(-(a)))
     23 #define lowbit(a) (a&(-a))
     24 #define sqr(a) ((a)*(a))
     25 #define swap(a,b) ((a)^=(b),(b)^=(a),(a)^=(b))
     26 #define mem(a,b) memset(a,b,sizeof(a))
     27 #define eps (1e-8)
     28 #define J 10
     29 #define mod 1000000007
     30 #define MAX 0x7f7f7f7f
     31 #define PI 3.14159265358979323
     32 #define N 100004
     33 using namespace std;
     34 typedef long long LL;
     35 int cas,cass;
     36 int n,m,lll,ans;
     37 int w[N],last[N],g[N];
     38 int f[2][N],from[3][N],h[3][N];
     39 struct xxx
     40 {
     41     int next,to,d;
     42 }a[N+N];
     43 bool mark[N];
     44 void add(int x,int y,int z)
     45 {
     46     a[++lll].d=z;
     47     a[lll].to=y;
     48     a[lll].next=last[x];
     49     last[x]=lll;
     50 }
     51 void dfs(int u,int fa)//从根开始往下走的解
     52 {
     53     int i,j,v;
     54     g[u]=w[u];
     55     for(i=last[u];i;i=a[i].next)
     56     {
     57         v=a[i].to;
     58         if(v==fa)continue;
     59         dfs(v,u);
     60         g[u]+=max(0,g[v]-a[i].d-a[i].d);//g[u]统计最后回到u的最优解
     61     }
     62     for(i=last[u];i;i=a[i].next)
     63     {
     64         v=a[i].to;
     65         if(v==fa || f[0][v]<=a[i].d)continue;
     66         j=g[u]-max(0,g[v]-a[i].d-a[i].d)+max(0,f[0][v]-a[i].d);
     67         //枚举从u哪一条走下去不回,如果g[u]计算时有走v则要扣掉,再加上选择走v不回的最优值
     68         if(f[0][u]<=j)//不回u的最优值
     69         {
     70             f[1][u]=f[0][u],from[1][u]=from[0][u];
     71             f[0][u]=j,from[0][u]=i;
     72         }
     73         else if(f[1][u]<j)//不回u的次优值
     74             f[1][u]=j,from[1][u]=i;
     75     }
     76     f[0][u]=max(f[0][u],g[u]);
     77     f[1][u]=max(f[1][u],g[u]);
     78 }
     79 void work(int u,int fa)//计算最后答案
     80 {
     81     int i,j,v;
     82     for(i=last[u];i;i=a[i].next)
     83     {
     84         v=a[i].to;
     85         if(v==fa)return;
     86         j=max(0,g[v]-a[i].d-a[i].d);//u走到v再走回来是否更优
     87         h[0][v]=f[0][v]+max(0,g[u]-j-a[i].d-a[i].d);//g[u]扣除掉走v子树的值,先从v向上走到u再从u走回来,然后走回v的最优值
     88         h[1][v]=f[1][v]+max(0,g[u]-j-a[i].d-a[i].d);//次优值
     89         from[2][v]=i;
     90         if(g[v]>=a[i].d+a[i].d)//这种情况下前面多扣了一次边权
     91         {
     92             if(from[0][u]!=i)h[2][v]=h[0][u]+a[i].d;//v往上走回头再往下走不回头
     93             else h[2][v]=h[1][u]+a[i].d;//当前是最优值,选另一条走次优值
     94         }
     95         else//前面少扣了一次边权
     96         {
     97             if(from[0][u]!=i)h[2][v]=h[0][u]+g[v]-a[i].d;//v往下走回头再往上走不回头
     98             else h[2][v]=h[1][u]+g[v]-a[i].d;
     99         }
    100         if(h[2][v]>h[1][v])swap(h[2][v],h[1][v]),swap(from[2][v],from[1][v]);
    101         if(h[1][v]>h[0][v])swap(h[1][v],h[0][v]),swap(from[1][v],from[0][v]);
    102         g[v]+=max(0,g[u]-j-a[i].d-a[i].d);//更新答案
    103         work(v,u);
    104     }
    105 }
    106 int main()
    107 {
    108     #ifndef ONLINE_JUDGE
    109 //    freopen("1.txt","r",stdin);
    110 //    freopen("2.txt","w",stdout);
    111     #endif
    112     int i,j,k;
    113     int x,y,z;
    114 //    for(scanf("%d",&cass);cass;cass--)
    115     for(scanf("%d",&cas),cass=1;cass<=cas;cass++)
    116 //    while(~scanf("%s",s+1))
    117 //    while(~scanf("%d",&n))
    118     {
    119         mem(f,0);mem(from,0);mem(last,0);lll=0;
    120         printf("Case #%d:
    ",cass);
    121         scanf("%d",&n);
    122         for(i=1;i<=n;i++)
    123             scanf("%d",&w[i]);
    124         for(i=1;i<n;i++)
    125         {
    126             scanf("%d%d%d",&x,&y,&z);
    127             add(x,y,z);
    128             add(y,x,z);
    129         }
    130         dfs(1,0);
    131         h[0][1]=f[0][1],h[1][1]=f[1][1];
    132         work(1,0);
    133         for(i=1;i<=n;i++)
    134             printf("%d
    ",h[0][i]);
    135     }
    136     return 0;
    137 }
    138 /*
    139 //
    140 
    141 //
    142 */
    View Code
  • 相关阅读:
    sharepoint环境的安装及form表单验证等遇到的问题及总结。
    话说在这里开辟个情感专栏怎么样呢?
    Android学习4—短信发送器的实现
    Android学习2项目文件列表简单分析
    Android学习3—电话拨号器
    腾讯内推一面记录
    日程日历示例
    屏幕设配介绍
    深入理解移动开发的模板复用机制
    沉浸屏幕状态栏
  • 原文地址:https://www.cnblogs.com/Coolxxx/p/5815465.html
Copyright © 2011-2022 走看看