简单树形DP

    树形DP,即是在一颗树上进行DP,一般是有叶子节点状态推出根节点状态。结合几个简单例子分析。

例1. POJ2342/POJ3342

【题意】

公司有n个人,每个人有价值vi,有一天举办年会,每个人都可以参加,但有严格的等级制度,参加活动时,不能同时出现a和a的上司,问如何才能使总和最大。

【分析】

每个人只有去和不去两种状态,设DP[i][0]和DP[i][1]分别表示第i个人不参加和参加年会,获得的总的最大价值。

则状态转移方程为:

DP[i][1] += DP[j][0],

DP[i][0] += max{DP[j][0],DP[j][1]};其中j为i的孩子节点。

这样,从根节点r进行dfs,最后结果为max{DP[r][0],DP[r][1]}。

POJ3342与上面类似,多了一个问题问最后最大的结果是不是唯一的,判断唯一性:当某个节点i取不取一样,并且它的某个孩子节点取不取一样,那么结果才不唯一。


例2. HDU2196

【题意】

一棵树,询问一颗树上每个点离它最远的点是哪个?

【分析】

一棵树上离它最远的点必定在树的直径上(否则可以找到离它更远的点),到底是两端点的哪个点呢?

求树的直径:从任意一点BFS,得到最远点k,则k点必定是树的直径其中一端点,再从k点BFS,得到另一端点j,在求解过程中用两个数组记录其它点到这两点的距离,最后比较较大者为答案。


例3. POJ1463

【题意】


一颗树,n个节点,现在要在节点处设置尽量少的哨兵,使得每条边都被观察到。

【分析】

一个节点要么设立哨兵,要么不设立,若节点i不设立哨兵,则节点i的所有的孩子节点必须设立哨兵;否则,看若节点i的孩子设立哨兵的值是否

小于不设哨兵的值,取二者中的较小者。

DP[i][0] = sum{DP[j][1]},

DP[i][1] = sum{min(DP[j][1],DP[j][0])} + 1, j是i的孩子节点。



例4. POJ4045

【题意】

n个城市节点构成的一棵树,节点i和节点j的电量损耗为I*I*R*i到j的路径所含边数,现在要修建一个供电站,使得总的损耗量最小。求最小

损耗和可以修建供电站的节点。

【分析】

一种简单的想法就是枚举所有节点,然后bfs求和,复杂度为O(n^2)。

首先,选定任意一点作为根节点dfs,设两个数组num[i]和dp[i],分别表示以节点i为根的子树的节点个数 和 根到其子节点的距离(边数)之和。

num[i] = sum{num[j]}+1,dp[i] = sum{num[j]+dp[j]}.

从root开始第二次dfs,依次考虑每个点作为根节点,考虑将x(当前根节点)的孩子y作为根节点:

dp[y] = dp[y] + (dp[x]-dp[y]-num[y])+(num[x]-num[y]) = dp[x]-2*num[y] + num[x],记录最小值即可。

附POJ2342和POJ4045代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int Max = 6005;

struct Node
{
    int fa;
    int value;
    vector<int> child;
}node[Max];
int dp[Max][2];
bool vis[Max];

int max(int a, int b)
{
    return a>b?a:b;
}

void dfs(int now)
{
    vis[now] = true;
    int s = node[now].child.size();
    for(int i = 0; i < s; i++)
    {
        int child = node[now].child[i];
        if(!vis[child])
        {
            dfs(child);
            dp[now][1] += dp[child][0];
            dp[now][0] += max(dp[child][0],dp[child][1]);
        }
    }
}

int main()
{
    int i;
    int a,b;
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(node,0,sizeof(Node));
        memset(vis,0,sizeof(vis));
        for(i = 1; i <= n; i++)
        {
            scanf("%d",&dp[i][1]);
            dp[i][0] = 0;
        }
        while(scanf("%d %d",&a,&b)!=EOF)
        {
            if(a==0&&b==0)
                break;
            node[a].fa = b;
            node[b].child.push_back(a);
        }
        int root = 1;
        for(i = 1; i <= n; i++)
        {
            if(!node[i].fa)
            {
                root = i;
                break;
            }
        }
        dfs(root);
        printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0;
}
 #include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int Max = 51000;
typedef __int64 int64;
int n;
int I,R;

struct Edge
{
    int to;
    int next;
}e[Max*2];
int head[Max],numm;
bool vis[Max];
int64 dp[Max],num[Max];

void addEdge(int from, int to)
{
    e[numm].to = to;
    e[numm].next = head[from];
    head[from] = numm++;
}

void dfs1(int now)
{
    dp[now] = 0;
    num[now] = 1;
    vis[now] = true;
    for(int i = head[now]; i != -1; i = e[i].next)
    {
        int next = e[i].to;
        if(!vis[next])
        {
            dfs1(next);
            num[now] += num[next];
            dp[now] += dp[next]+num[next];
        }
    }
}

void dfs2(int now)
{
    vis[now] = true;
    for(int i = head[now]; i != -1; i = e[i].next)
    {
        int next = e[i].to;
        if(!vis[next])
        {
            dp[next] = dp[now] - 2*num[next] + n;
            dfs2(next);
        }
    }
}

int main()
{
    int i,j;
    int t;
    int x,y;
    scanf("%d",&t);
    while(t--)
    {
        numm = 0;
        memset(head,-1,sizeof(head));
        memset(vis,0,sizeof(vis));
        scanf("%d %d %d",&n,&I,&R);
        for(i = 1; i < n; i++)
        {
            scanf("%d %d",&x,&y);
            addEdge(x,y);
            addEdge(y,x);
        }
        dfs1(1);
        memset(vis,0,sizeof(vis));
        dfs2(1);
        int64 result = ((int64)1)<<61;
        vector<int> ans;
        ans.clear();
        for(i = 1; i <= n; i++)
        {
            if(dp[i]<result)
            {
                ans.clear();
                ans.push_back(i);
                result = dp[i];
            }
            else if(dp[i]==result)
                ans.push_back(i);
        }
        printf("%I64d\n",result*I*I*R);
        int r = ans.size();
        for(i = 0; i < r; i++)
        {
            printf("%d",ans[i]);
            printf("%c",i==r-1?'\n':' ');
        }
        printf("\n");
    }
    return 0;
}
 

你可能感兴趣的:(简单树形DP)