HDU 1520 Anniversary party (简单树形DP入门)

来源:互联网 发布:我的世界java路径 编辑:程序博客网 时间:2024/05/27 00:43

树形dp,可能是涉及到图论和dfs,显得比较高大上,我现在只做了一题,也没有什么发言权。目前总结如下:

1.“树”,当作有向图(而且还无环的)来处理,建图(树)的时候更为简单,只要在它和它的孩子节点之间加一条指向孩子的有向边就可以了。

2.dp还是一样的dp,但是树形dp往往不是“线性、水平”的了,而是要从叶子节点推到根节点(或者反过来),关系盘结的更为复杂。

这题是一道最简单的树形dp;方程也很容易想到

当i来的时候,dp[i][1] += dp[j][0];//j为i的下属

当i不来的时候,dp[i][0] +=max(dp[j][1],dp[j][0]);//j为i的下属


这里我是用vector实现的邻接表,因为是有向图,每个节点只会被访问一次,甚至连vis数组都可以不用了。用结构体也是一样。

【代码】

/* ***********************************************Author        :angon2016年8月15日 23:52:07************************************************ */#include <stdio.h>#include <string.h>#include <iostream>#include <algorithm>#include <stack>#include <vector>#include <queue>#include <set>#include <map>#include <string>#include <math.h>#include <stdlib.h>#include <time.h>using namespace std;#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)#define lld %I64d#define REP(i,k,n) for(int i=k;i<n;i++)#define REPP(i,k,n) for(int i=k;i<=n;i++)#define scan(d) scanf("%d",&d)#define scanl(d) scanf("%I64d",&d)#define scann(n,m) scanf("%d%d",&n,&m)#define scannl(n,m) scanf("%I64d%I64d",&n,&m)#define mst(a,k)  memset(a,k,sizeof(a))#define LL long long#define N 6005#define mod 1000000007inline int read(){int s=0;char ch=getchar();for(; ch<'0'||ch>'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}int n,dp[N][2],father[N];vector<int>G[N];void dfs(int root){    for(int i=0;i<G[root].size();i++)    {        dfs(G[root][i]);  //递归root的每个孩子    }    for(int i=0;i<G[root].size();i++) //回溯时计算    {        dp[root][0] += max(dp[G[root][i]][0],dp[G[root][i]][1]);        dp[root][1] += dp[G[root][i]][0];    }}int main(){    //freopen("in.txt","r",stdin);    //freopen("out.txt","w",stdout);    int u,v;    while(~scan(n))    {        mst(father,-1); mst(dp,0);        REPP(i,1,n) { scan(dp[i][1]); G[i].clear();}        while(scann(u,v)&&u&&v)        {            father[u]=v;            G[v].push_back(u);        }        int root=1;        while(father[root]!=-1) root=father[root];        dfs(root);        printf("%d\n",max(dp[root][1],dp[root][0]));    }    return 0;}



0 0
原创粉丝点击