题目描述
你要组织一个由你公司的人参加的聚会。你希望聚会非常愉快,尽可能多地找些有趣的热闹。但是劝你不要同时邀请某个人和他的上司,因为这可能带来争吵。给定N个人(姓名,他幽默的系数,以及他上司的名字),编程找到能使幽默系数和最大的若干个人。
输入输出格式
输入格式:
第一行一个整数N(N<100)。接下来有N行,每一行描述一个人的信息,信息之间用空格隔开。姓名是长度不超过20的字符串,幽默系数是在0到100之间的整数。
输出格式:
所邀请的人最大的幽默系数和。
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <fstream>
#include <iostream>
using namespace std;
int n,fa;
vector <int> M[100+5];
int a[100+5];
string boss[100+5],name[100+5];
int f[100+5][2];//0不选,1选
void dfs(int x)
{
int i,y;
f[x][1]=a[x];
for (i=0;i<M[x].size();i++)
{
y=M[x][i];
dfs(y);
f[x][0]+=max(f[y][0],f[y][1]); //boss不参加
f[x][1]+=f[y][0]; //boss参加
}
}
int main()
{
int i,j;
string s1,s2;
ifstream cin("a.txt");
cin>>n;
for (i=1;i<=n;i++)
cin>>name[i]>>a[i]>>boss[i];
for (i=1;i<=n;i++)
{
if (boss[i]=="NOBODY")
{
fa=i;
continue;
}
for (j=1;j<=n;j++)
if (boss[i]==name[j] )
{
M[j].push_back(i);
}
}
dfs(fa);
printf("%d\n",max(f[fa][0],f[fa][1]));
return 0;
}
按正常思路写出以上程序,有一个点通过不了。
检查数据发现有重名的数据,即同一个员工有两个上司,必须以后面的为准,忽略前面的!
用map来做就不存在这个问题!
#include<bits/stdc++.h>
using namespace std;
vector<int> son[110];
map<string,int> id;
int n,st,f[110][2];
struct node{string s,fa;int w;}a[110];
void dp(int x)
{
f[x][0]=0,f[x][1]=a[x].w;
for(int i=0;i<son[x].size();i++)
{
int y=son[x][i];
dp(y);
f[x][1]+=f[y][0];
f[x][0]+=max(f[y][0],f[y][1]);
}
}
int main()
{
//ios::sync_with_stdio(false);
ifstream cin("a.txt");
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i].s>>a[i].w>>a[i].fa,id[a[i].s]=i;
for(int i=1;i<=n;i++) son[id[a[i].fa]].push_back(i);
for(int i=1;i<=n;i++) if(!id[a[i].fa]) {st=i;break;}
dp(st);
printf("%d\n",max(f[st][0],f[st][1]));
printf("\n\n");
return 0;
}