树形动态规划 fjutoj-2392 聚会的快乐

聚会的快乐

TimeLimit:1000MS  MemoryLimit:128MB

64-bit integer IO format:%lld

Problem Description

你要组织一个由你公司的人参加的聚会。你希望聚会非常愉快，尽可能多地找些有趣的热闹。但是劝你不要同时邀请某个人和他的上司，因为这可能带来争吵。给定N个人（姓名，他幽默的系数，以及他上司的名字），编程找到能使幽默系数和最大的若干个人。

Input

第一行一个整数N（N<100）。接下来有N行，每一行描述一个人的信息，信息之间用空格隔开。姓名是长度不超过20的字符串，幽默系数是在0到100之间的整数。

Output

所邀请的人最大的幽默系数和。

SampleInput

5
BART 1 HOMER
HOMER 2 MONTGOMERY
MONTGOMERY 1 NOBODY
LISA 3 HOMER
SMITHERS 4 MONTGOMERY

SampleOutput

8

思路 先建树，深搜遍历，再回溯时 进行状态转移

dp[now][0] =dp[now][0]+max(dp[to][0],dp[to][1]);     //0不放,1放  now为当前节点，to为子节点
dp[now][1] =dp[now][1]+dp[to][0];      //当前节点为1时，子节点一定不能有，当前节点为0时，子节点可有可无

最后输出dp[1][0],dp[1][1]中的较小值

下面附上代码

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
char s[105][30];
vector<int>e[105];
int v[105];
int dp[105][2];
int tot=0;
int vis[105];
void dfs(int now)
{
    int len = e[now].size();
    for(int i=0; i<len; i++)
    {
        int to = e[now][i];
        if(!vis[to])
        {
            vis[to]=1;
            dfs(to);
        }
    }
    for(int i=0; i<len; i++)
    {
        int to = e[now][i];
        dp[now][0] =dp[now][0]+max(dp[to][0],dp[to][1]);  //0不放,1放
        dp[now][1] =dp[now][1]+dp[to][0];
    }
    dp[now][1]+=v[now];
}

int main()
{
    char str[30];
    char ttr[30];
    int n;
    scanf("%d",&n);
    for(int i=0; i<n; i++)
    {
        int value,f1=0,f2=0,from,to;
        scanf("%s%d%s",str,&value,ttr);
        for(int j=0; j<tot; j++)
        {
            if(!strcmp(str,s[j]))
            {
                from = j+1;
                f1=1;
            }
            else if(!strcmp(ttr,s[j]))
            {
                to = j+1;
                f2=1;
            }
        }
        if(f1==0)
        {
            strcpy(s[tot],str);
            tot++;
            from = tot;
        }
        if(f2==0)
        {
            strcpy(s[tot],ttr);
            tot++;
            to = tot;
        }
        e[from].push_back(to);
        e[to].push_back(from);
        v[from] = value;

    }
    vis[1]=1;
    dfs(1);
    printf("%d\n",max(dp[1][0],dp[1][1]));
    return 0;
}