【bzoj4247】挂饰 背包dp

题目描述

JOI君有N个装在手机上的挂饰，编号为1...N。 JOI君可以将其中的一些装在手机上。

JOI君的挂饰有一些与众不同——其中的一些挂饰附有可以挂其他挂件的挂钩。每个挂件要么直接挂在手机上，要么挂在其他挂件的挂钩上。直接挂在手机上的挂件最多有1个。

此外，每个挂件有一个安装时会获得的喜悦值，用一个整数来表示。如果JOI君很讨厌某个挂饰，那么这个挂饰的喜悦值就是一个负数。

JOI君想要最大化所有挂饰的喜悦值之和。注意不必要将所有的挂钩都挂上挂饰，而且一个都不挂也是可以的。

输入

第一行一个整数N，代表挂饰的个数。

接下来N行，第i行(1<=i<=N)有两个空格分隔的整数Ai和Bi，表示挂饰i有Ai个挂钩，安装后会获得Bi的喜悦值。 

输出

输出一行一个整数，表示手机上连接的挂饰总和的最大值

样例输入

5
0 4
2 -2
1 -1
0 1
0 3

样例输出

5

---

题解

背包dp

根据题意很容易想到dp状态：f[i][j]表示从前i个物品中选择某些物品，使得剩下的挂钩数量为j的最大喜悦值。

但是这样会TLE。

思考：一个物品，最多只会消耗1个挂钩。因此如果已经有了大于等于超过n个挂钩，说明全部物品都可以挂上，记录过多的状态也就没有了意义。

所以我们把j的上界设为n即可，dp时取j+ai和n的最小值作为状态即可。

注意要先按照挂钩数量从大到小排序（其实不排序也行，就是会比较麻烦）

代码中把状态压到了一维，需要注意一下更新顺序啥的。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 2010
using namespace std;
struct data
{
	int a , b;
}w[N];
int f[N];
bool cmp(data x , data y)
{
	return x.a > y.a;
}
int main()
{
	int n , i , j , ans = 0;
	scanf("%d" , &n);
	for(i = 1 ; i <= n ; i ++ ) scanf("%d%d" , &w[i].a , &w[i].b) , w[i].a -- ;
	sort(w + 1 , w + n + 1 , cmp);
	memset(f , 0xc0 , sizeof(f)) , f[1] = 0;
	for(i = 1 ; i <= n ; i ++ )
	{
		if(~w[i].a) for(j = n ; j ; j -- ) f[min(j + w[i].a , n)] = max(f[min(j + w[i].a , n)] , f[j] + w[i].b);
		else for(j = 1 ; j <= n ; j ++ ) f[j - 1] = max(f[j - 1] , f[j] + w[i].b);
		for(j = 0 ; j <= n ; j ++ ) ans = max(ans , f[j]);
	}
	printf("%d\n" , ans);
	return 0;
}