UVA 1220 Party at Hali-Bula

最新推荐文章于 2021-05-11 19:08:46 发布

Shimmer_

最新推荐文章于 2021-05-11 19:08:46 发布

阅读量703

点赞数

CC 4.0 BY-SA版权

分类专栏：动态规划

本文链接：https://blog.youkuaiyun.com/Shimmer_/article/details/38408505

动态规划专栏收录该内容

60 篇文章

订阅专栏

本文介绍了一种将无根树转换为有根树的方法，并通过动态规划解决特定问题。利用map和vector实现树的构建过程，最终通过递归函数完成遍历并计算最优解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个题还是很简单的、最主要的还是通过各种方法，把一个无根树变成有根树。然后进行搜索。

这个建树的过程很重要。用的是 map 标记和 vector 的下标。

然后再来实现就简单了。

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <string>
#include <map>
#include <vector>
#include <set>
#include <queue>
#include <stack>
using namespace std;
#define ll long long
#define maxn 200+10
#define INF 1<<30
map <string , int> q;
vector <string> fa;
vector <int> son[maxn];
int f[maxn][3] = {0};
int init(){
    q.clear();
    fa.clear();
    for(int i = 0; i < maxn; i++)
        son[i].clear();
    memset(f,0,sizeof(f));
}
int get_id(string  &s){
    if(q.count(s))
        return q[s];
    fa.push_back(s);
    return q[s] = fa.size() - 1;
}
int dp(int i,int j){
    if(son[i].empty()){
        f[i][j] = 1;
        if(j == 0)
            return 0;
        return 1;
    }
    int v = 0;
    if(j == 0){
        int len = son[i].size();
        int ok = 1;
        for(int k = 0; k < len; k++){
            int no_cho = dp(son[i][k] , 0),cho = dp(son[i][k] , 1);
            v += max(no_cho , cho);
            if(no_cho > cho ){
                if(f[son[i][k]][0] != 1) ok = 0;
            }
            else if(no_cho < cho)
                if(f[son[i][k]][1] != 1) ok=0;
            else ok=0;
        }
        f[i][j]=(ok ? 1 : 0);
    }
    else{
        v = 1;
        int len = son[i].size(), ok = 1;
        for(int k = 0; k < len; k++){
            v += dp(son[i][k] , 0);
            if(f[son[i][k]][0] != 1) ok=0;
        }
        f[i][j]=(ok ? 1 : 0);
    }
    return v;
}
int main (){
    int n;
    while(scanf("%d",&n) && n){
        init();
        string s;
        cin >> s;
        get_id(s); // 分配地址
        for(int i = 1; i < n; i++){
            string s1,s2;
            cin >> s1 >> s2;
            son[get_id(s2)].push_back(get_id(s1));
        }
        int no_cho = dp(0,0), cho = dp(0,1);
        int flag = 1;//唯一
        int max_;
        if(no_cho == cho){
            flag = 0;
            max_ =cho;
        }
        else if(no_cho > cho){
            if(f[0][0] != 1)
                flag = 0;
            max_ = no_cho;
        }
        else {
            if(f[0][1] != 1)
                flag = 0;
            max_ = cho;
        }
        cout << max_ << (flag ? " Yes" : " No") << endl;
    }
    return 0;
}