UVA1220[Party at Hali-Bula] 树上动态规划

最新推荐文章于 2020-01-17 17:00:23 发布

Venishel

最新推荐文章于 2020-01-17 17:00:23 发布

阅读量256

点赞数

分类专栏：其它——题解 UVA刷题文章标签：动态规划 uva UVa1220 dp 树上动态规划

本文链接：https://blog.csdn.net/qq_37321281/article/details/77877088

版权

UVA刷题同时被 2 个专栏收录

35 篇文章 2 订阅

订阅专栏

其它——题解

28 篇文章 0 订阅

订阅专栏

该博客探讨了UVA1220问题，一个关于公司员工聚会的挑战，其中参与者不能与他们的直接上司同场。博主提出了一种解决方案，使用树上动态规划来确定最多可以有多少员工参加聚会，同时判断方案是否唯一。通过dp[u][0]和dp[u][1]分别表示节点u不选和选的情况，以及f[u][0]和f[u][1]表示方案的独特性。

摘要由CSDN通过智能技术生成

题目链接

题意：一个公司员工要举行聚会，要求任意一个人不能和他的直接上司同时到场，一个员工只有一个支系上司，现在求最多有多少人到场，并且方案是否唯一。

solution：用dp[u][0], dp[u][1] 分别代表 u结点不选或选， f[u][0], f[u][1], 代表对应的方案是否唯一。

dp[u][0]+=max(dp[v][0],dp[v][1]) v是u的字结点
dp[u][1]+=dp[v][0]
f[u][1]=（f[v][0]==1)?1:0
f[u][0]与dp[u][0]类似

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
const int N = 205;
vector<int> G[N];
map<string,int> mp;
int dp[N][3], f[N][3], tot;

void d(int u){
    if ( !G[u].size() ){
        dp[u][0]=0;
        dp[u][1]=1;
        return ;    
    }
    int size=G[u].size();
    for ( int i=0; i<size; i++ ){
        int v=G[u][i];
        d(v);
        dp[u][1]+=dp[v][0];
        if( f[v][0] ) f[u][1]=1;
        if( dp[v][1]>dp[v][0] ){
            dp[u][0]+=dp[v][1];
            if( f[v][1] ) f[u][0]=1;
        }else {
            dp[u][0]+=dp[v][0];
            if( dp[v][1]==dp[v][0] || f[v][0] ) f[u][0]=1;
        }   
    }
    dp[u][1]++;
} 
int id(string str){
    if ( mp.count(str) ) return mp[str];
    mp[str]=++tot;
    return tot;
}
int main(){
    int n;
    while( scanf("%d", &n )==1 && n ){
        memset(dp,0,sizeof(dp));
        memset(f,0,sizeof(f));
        tot=0;
        string s1, s2;
        cin>>s1;
        int root=id(s1);
        for ( int i=1; i< n; i++ ){
            cin>>s1>>s2;
            G[id(s2)].push_back(id(s1));
        }
        d(1);
        if(dp[1][1]==dp[1][0])  
            printf("%d No\n",dp[1][1]);  
        else if(dp[1][1]>dp[1][0])  
            printf("%d %s\n",dp[1][1],f[1][1]?"No":"Yes");  
        else  
            printf("%d %s\n",dp[1][0],f[1][0]?"No":"Yes");  
        for(int i = 1;i<=n;i++)  
            G[i].clear();  
        mp.clear();  
    }
}