HDU 2412 树状DP

最新推荐文章于 2018-12-20 10:33:57 发布

lentty

最新推荐文章于 2018-12-20 10:33:57 发布

阅读量607

点赞数

分类专栏：大学ACM记忆

本文链接：https://blog.csdn.net/lentty1452/article/details/8716382

版权

大学ACM记忆专栏收录该内容

85 篇文章 0 订阅

订阅专栏

求最多人数直接树状DP就好了跟HDU 1520类似

这题关键是求解是否唯一。。

看了某个PPT后了解到：

新加一个状态dup[i][j]，表示相应的dp[i][j]是否是唯一方案。
对于叶子结点, dup[k][0] = dup[k][1] = 1.
对于非叶子结点,
对于i的任一儿子j，若(dp[j][0] > dp[j][1] 且 dup[j][0] == 0) 或 (dp[j][0] < dp[j][1] 且 dup[j][1] == 0) 或 (dp[j][0] == dp[j][1])，则dup[i][0] = 0
对于i的任一儿子j有dup[j][0] = 0, 则dup[i][1] = 0

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<stdlib.h>
#include<algorithm>
#include<vector>
#include<map>
#include<iostream>
#include<string>
using namespace std;
#define maxn 220
int dp[maxn][2];
int dup[maxn][2];
vector<int> vec[maxn];
map<string,int> mp;
void init()
{
    for(int i=0;i<maxn;i++)
    vec[i].clear();
    mp.clear();
}
void dfs(int rt)
{
    int sz=vec[rt].size();
    for(int i=0;i<sz;i++)
    {
        int v=vec[rt][i];
        dfs(v);
        dp[rt][0]+=max(dp[v][0],dp[v][1]);
        dp[rt][1]+=dp[v][0];
        if(dp[v][0]>dp[v][1]&&dup[v][0]==0)
        dup[rt][0]=0;
        else if(dp[v][0]<dp[v][1]&&dup[v][1]==0)
        dup[rt][0]=0;
        else if(dp[v][0]==dp[v][1])
        dup[rt][0]=0;
        if(dup[v][0]==0)
        dup[rt][1]=0;
    }
}
int main()
{
    freopen("in.txt","r",stdin);
    int n;
    while(scanf("%d",&n)==1&&n)
    {
        init();
        string u,v;
        cin>>u;
        int cnt=1;
        if(mp[u]==0)
        mp[u]=cnt++;
        int i;
        for(i=0;i<n-1;i++)
        {
            cin>>u>>v;
            if(mp[u]==0)
            mp[u]=cnt++;
            if(mp[v]==0)
            mp[v]=cnt++;
            vec[mp[v]].push_back(mp[u]);
        }
        for(i=1;i<cnt;i++)
        {
            dp[i][0]=0;
            dp[i][1]=1;
            dup[i][0]=1;
            dup[i][1]=1;
        }
        dfs(1);
        if(dp[1][0]>dp[1][1])
        {
            printf("%d ",dp[1][0]);
            if(dup[1][0]==0)
            printf("No\n");
            else
            printf("Yes\n");
        }
        else if(dp[1][0]<dp[1][1])
        {
            printf("%d ",dp[1][1]);
            if(dup[1][1]==0)
            printf("No\n");
            else
            printf("Yes\n");
        }
        else if(dp[1][0]==dp[1][1])
        printf("%d No\n",dp[1][1]);
    }
    return 0;
}