「BSOJ2900」 Angry Birds - 线性Dp+搜索

题目描述

绿猪们的势力越来越强大了，小鸟们正在筹划如何对付这群强大的东西。

可是绿猪们派来一位绿猪巫师洗清了所有前线鸟儿的记忆。现在，他们只记得他们各自的领导，但是这是一个自由散漫的团队，必须要找到他们的鸟队长。（鸟队长是能管理所有的鸟儿作战，鼓舞士气，勇于拼搏在前线的英雄鸟）。

你是在后方暗地操作的司令，现有M（1 <= M <= 500）件装备，每只不同的鸟儿可以分配零件至M件装备，分配到不同的装备数量会有不同的战斗力。

绿猪们也不是吃素的，他们规划了一个方案，站成一字长蛇阵在草地上摆开阵势，静候着小鸟们。此时，你也只好得选出一条长链来应战绿猪。（温馨提示：小鸟的编号从1开始）

现在，你的任务有三：

1.  输出鸟队长的编号
2.  输出将M件装备分配后整个团队的战斗力的最大值
3.  以鸟队长为首的最大战斗力的战斗长链的鸟儿的序列

分析

对于第一问，简单的记录即可；第二问，简单的资源分配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;
}