【wikioi】1025 选菜

题目链接

算法：01背包DP

此题主要是预处理恶心。我提交了2次。。。第一次数组开小了。。。（体积要=V*10）

注意：

钱做为体积，美味价值作为价值

注意，因为体积（钱）是小数点后1位，故数组下标无法表示体积（01背包），所以体积（钱）要扩大10倍作为01背包的体积
还有因为有重复的，所以要去重再01

代码：

#include <iostream>
#include <algorithm>
using namespace std;
//钱做为体积，美味价值作为价值
//注意，因为体积（钱）是小数点后1位，故数组下标无法表示体积（01背包）。所以要扩大10倍，输出答案再缩小10倍
//还有因为有重复的，所以要去重再01
const int MAXN = 105;
int t[MAXN], n, k, V;			//t代表种类
int v1[MAXN], w1[MAXN];			//v1代表价格（体积），w1代表美味价值（价值）（去重前）
int v[MAXN], w[MAXN], nn = 0;	//v代表价格（体积），w代表美味价值（价值）
int f[MAXN*10];
bool m[MAXN];					//m[i]表示第i种是否被购买
int ans = 0;

int main()
{
	int i, j;
	double temp;
	int tem;
	cin >> n >> k >> temp;
	V=(int)(temp*10);
	for(i = 1; i <= n; i++)	{cin >> temp; v1[i] = (int)(temp*10);}
	for(i = 1; i <= n; i++)	cin >> w1[i];
	for(i = 1; i <= n; i++)	cin >> t[i];
	for(i = 1; i <= k; i++)
	{
		cin >> tem;
		m[tem] = 1; //必买种类已经用过的标志
		for(j = 1; j <= n; j++) //先处理必须买的
			if(tem == t[i])
			{
				ans += w1[i];
				V -= v1[i]; //必须买的后减小体积
				break;
			}
	}
	//题目说数据中不会出现小松带的钱不够买必买菜的情况，所以不用判断买完必要的后钱不够的情况
	if(V == 0) {cout << ans << endl;return 0;}
	for(i = 1; i <= n; i++)
		if(!m[t[i]]) //这种种类没有访问过
			w[++nn]=w1[i], v[nn]=v1[i], m[t[i]]=1;
	//01背包
	for(i = 1; i <= nn; i++)
		for(j = V; j >= v[i]; j--)
			f[j] = max(f[j], f[j-v[i]]+w[i]);
	cout << ans+f[V] << endl;
	return 0;
}