1. 程式人生 > >DP——樹形DP——HDOJ1620

DP——樹形DP——HDOJ1620

題目連結:http://acm.hdu.edu.cn/showproblem.php?pid=1520 

大犇解題報告:http://blog.csdn.net/woshi250hua/article/details/7641589#comments 

用陣列模擬實現鄰接關係:

//http://acm.hdu.edu.cn/showproblem.php?pid=1520
#include <iostream>
#include <cstdio>
#include <cstring>
//using namespace std;
#define max(a,b) (a)>(b) ? (a) : (b)
int tree[6600][20]; // 這個陣列的第二位應該開到6000,
        // 防止極限資料,但執行時間會急劇增加。
        //這hdoj的測試平臺上,開到20就可以AC了。
int vis[6699],dp[6600][2],num[6600];

void f(int u)
{
    if(vis[u])
        return;
    else {
        vis[u]=1;
//        for (int j=tree[u][0];j>=1;j--){
            if(!vis[tree[u][j]]){
				f(tree[u][j]);
				int w=tree[u][j];
				dp[u][1]+=dp[w][0];
				dp[u][0]+=max(dp[w][0],dp[w][1]);
			}
        }
    }
    dp[u][1]+=num[u];
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF){
        for (int j=1;j<=n;j++)
            scanf("%d", &num[j]);

        memset(vis, 0,sizeof(vis));
        memset(tree,0,sizeof(tree));
        memset(dp,0,sizeof(dp));

        int a,b;
        while(scanf("%d%d",&a,&b),a+b){
            tree[a][++tree[a][0]]=b;
            tree[b][++tree[b][0]]=a;
        }

        f(1);
//        std::cout << max(dp[1][1],dp[1][0])<<std::endl;
        printf("%d\n", max(dp[1][1],dp[1][0]));
    }
}


用二維vector實現鄰接關係:

//http://acm.hdu.edu.cn/showproblem.php?pid=1520
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define max(a,b) (a)>(b) ? (a) : (b)
//int tree[6600][20]; // 這個陣列的第二位應該開到6000,
//        // 防止極限資料,但執行時間會急劇增加。
//        //這hdoj的測試平臺上,開到20就可以AC了。
int vis[6699],dp[6600][2],num[6600];
vector<vector< int> > tree;

void f(int u)
{
    if(vis[u])
        return;
    else {
        vis[u]=1;
        //        for (int j=tree[u][0];j>=1;j--){
        for (int j=tree[u].size()-1;j>=0;j--){
            if(!vis[tree[u][j]]){
				f(tree[u][j]);
				int w=tree[u][j];
				dp[u][1]+=dp[w][0];
				dp[u][0]+=max(dp[w][0],dp[w][1]);
			}
        }
    }
    dp[u][1]+=num[u];
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF){
        for (int j=1;j<=n;j++)
            scanf("%d", &num[j]);

        memset(vis, 0,sizeof(vis));
//        memset(tree,0,sizeof(tree));
        tree.clear();
        tree.resize(n+1);
        memset(dp,0,sizeof(dp));

        int a,b;
        while(scanf("%d%d",&a,&b),a+b){
//            tree[a][++tree[a][0]]=b;
//            tree[b][++tree[b][0]]=a;
            tree[a].push_back(b);
            tree[b].push_back(a);
        }

        f(1);
//        std::cout << max(dp[1][1],dp[1][0])<<std::endl;
        printf("%d\n", max(dp[1][1],dp[1][0]));
    }
}