树形动态规划 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[][];
 vector<int>e[];
 int v[];
 int dp[][];
 int tot=;
 int vis[];
 void dfs(int now)
 {
     int len = e[now].size();
     for(int i=; i<len; i++)
     {
         int to = e[now][i];
         if(!vis[to])
         {
             vis[to]=;
             dfs(to);
         }
     }
     for(int i=; i<len; i++)
     {
         int to = e[now][i];
         dp[now][] =dp[now][]+max(dp[to][],dp[to][]);  //0不放,1放
         dp[now][] =dp[now][]+dp[to][];
     }
     dp[now][]+=v[now];
 }
 
 int main()
 {
     char str[];
     char ttr[];
     int n;
     scanf("%d",&n);
     for(int i=; i<n; i++)
     {
         int value,f1=,f2=,from,to;
         scanf("%s%d%s",str,&value,ttr);
         for(int j=; j<tot; j++)
         {
             if(!strcmp(str,s[j]))
             {
                 from = j+;
                 f1=;
             }
             else if(!strcmp(ttr,s[j]))
             {
                 to = j+;
                 f2=;
             }
         }
         if(f1==)
         {
             strcpy(s[tot],str);
             tot++;
             from = tot;
         }
         if(f2==)
         {
             strcpy(s[tot],ttr);
             tot++;
             to = tot;
         }
         e[from].push_back(to);
         e[to].push_back(from);
         v[from] = value;
 
     }
     vis[]=;
     dfs();
     printf("%d\n",max(dp[][],dp[][]));
     return ;
 }