码迷,mamicode.com
首页 > 其他好文 > 详细

UVa Placing Lampposts 树型DP

时间:2015-04-07 21:37:44      阅读:160      评论:0      收藏:0      [点我收藏+]

标签:acm   算法   uva   动态规划   dp   

        大致思路和大白书上的相同,不过感觉书上的决策部分讲解的并不是非常清楚,因此我在这里讲解一下我的决策思路。

        首先,d(i,j)表示根节点为i的子树,当它的父节点为j(j=0或1)时的x的最小值(x的含义书上有讲解),要将该子树根节点和父节点相连的边的情况计算在内。接下来遍历森林中的每一棵树,对于每一棵树的根节点进行特别的处理,然后就对该树进行深度优先搜索dfs(i)。

        对于d[i][0]的情况,因为当前子树根节点i的父节点为0,所以该子树根节点的状态必为1,则d[i][0]=sum{d[k][1]|k为i的子节点}+M+1。然而,对于d[i][1]则需要分两种情况进行讨论:当i为1时,d1=sum{d[k][1]|k为i的子节点}+M,当i为0时,d0=sum{d[k][0]|k为i的子节点}+1,则d[i][1]=min(d0,d1)。最后每棵树的根节点都按照自身为1和自身为0,分别加上对于的sum{d[k][j]|k为对应根节点的子节点}即可。详细解答见代码。


#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAX 1000+5
#define M   2000
using namespace std;

int T,n,m,x;
int G[MAX][MAX],vis[MAX],d[MAX][2];
void dfs(int);

int main()
{
    cin>>T;
    while(T--){
        cin>>n>>m;
        memset(G,0,sizeof(G));
        x=0;

        int a,b;
        for(int i=0;i<m;++i){
            cin>>a>>b;
            G[a][b]=G[b][a]=1;
        }

        memset(vis,0,sizeof(vis));
        memset(d,0,sizeof(d));

        for(int i=0;i<n;++i) if(!vis[i]){
            vis[i]=1;
            int d0=0,d1=0;

            for(int j=0;j<n;++j) if(G[i][j]){
                dfs(j);
                d0+=d[j][0];
                d1+=d[j][1];
            }
            d1+=M;
            x+=min(d0,d1);
           
        }
        cout<<x/M<<" "<<m-x%M<<" "<<x%M<<endl;
    }

    return 0;
}

void dfs(int root)
{
    int leaf=1,d0=0,d1=0;
    vis[root]=1;

    for(int i=0;i<n;++i) if(!vis[i]&&G[root][i]){
        leaf=0;
        dfs(i);
        d[root][0]+=d[i][1];
        d0+=d[i][0];
        d1+=d[i][1];
    }
    d[root][0]+=M+1;
    d[root][1]=min(d1+M,d0+1);

    if(leaf){//叶子节点特殊处理
        d[root][0]=M+1;
        d[root][1]=1;
    }

    return;
}



UVa Placing Lampposts 树型DP

标签:acm   算法   uva   动态规划   dp   

原文地址:http://blog.csdn.net/u011915301/article/details/44925981

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!