题目描述
绿猪们的势力越来越强大了,小鸟们正在筹划如何对付这群强大的东西。
可是绿猪们派来一位绿猪巫师洗清了所有前线鸟儿的记忆。现在,他们只记得他们各自的领导,但是这是一个自由散漫的团队,必须要找到他们的鸟队长。(鸟队长是能管理所有的鸟儿作战,鼓舞士气,勇于拼搏在前线的英雄鸟)。
你是在后方暗地操作的司令,现有M(1 <= M <= 500)件装备,每只不同的鸟儿可以分配零件至M件装备,分配到不同的装备数量会有不同的战斗力。
绿猪们也不是吃素的,他们规划了一个方案,站成一字长蛇阵在草地上摆开阵势,静候着小鸟们。此时,你也只好得选出一条长链来应战绿猪。(温馨提示:小鸟的编号从1开始)
现在,你的任务有三:
- 输出鸟队长的编号
- 输出将M件装备分配后整个团队的战斗力的最大值
- 以鸟队长为首的最大战斗力的战斗长链的鸟儿的序列
分析
对于第一问,简单的记录即可;第二问,简单的资源分配Dp+记录路径即可;第三问,简单的搜索即可。最后注意范围即可。
代码
#include <cstdio>
#include <queue>
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;
int f[105][505];
int a[105][505],g[105][105];
bool s[105][505][505];
int n,m,ans,cnt;
int val[105];
vector<int> anss;
void dfs(int x,vector<int> s,int sum) {//暴力Dfs
if (sum>ans&&g[x][0]==0) {
ans=sum;
anss=s;
}
for (int i=1;i<=g[x][0];i++) {
int y=g[x][i];
s.push_back(y);
dfs(y,s,sum+val[y]);
s.pop_back();
}
}
int main() {
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) {
int fa;
scanf("%d",&fa);
if (fa==0) {
cnt++;
ans=i;
} else g[fa][++g[fa][0]]=i;
for (int j=1;j<=m;j++)
scanf("%d",&a[i][j]);
}
for (int i=1;i<=n;i++)
sort(g[i]+1,g[i]+g[i][0]+1);
if (cnt!=1) {
puts("not found");
return 0;
}
printf("%d\n",ans);
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
for (int k=0;k<=j;k++) {
if (f[i-1][j-k]+a[i][k]>f[i][j]) {
f[i][j]=f[i-1][j-k]+a[i][k];
s[i][j][k]=1;//路径记录
}
}
printf("%d\n",f[n][m]);
int t1=n,t2=m;
while (t1&&t2) {//寻找路径
for (int j=t2;j>=1;j--) {
if (s[t1][t2][j]) {
val[t1]=j;
t2-=j;
break;
}
}
t1--;
}
for (int i=1;i<=n;i++)
val[i]=a[i][val[i]];
vector<int> s;
s.push_back(ans);
ans=0;
dfs(s[0],s,0);
printf("%d",anss[0]);
for (int i=1;i<anss.size();i++) {
printf("->%d",anss[i]);
}
return 0;
}