ZOJ 3769 Diablo III

题意：

13种装备，每种只能取一件。双手武器不能和单手装备同时装备。求满足防御大于等于m的情况下的最大攻击力。

思路：
dp，把大于等于M的状态都放到M上。做两次dp，第一次不加入双手武器，第二次不加入单手武器。还有戒指能带两个，所以单独处理，也不加入dp。最后枚举两个戒指组成一个新装备再更新当前的状态。当然你不能暴力更新。你可以把状态按攻击排序，新组合出装备按防御排序。这样，如果当前最大攻击的状态所拥有的防御+当前新装备最大的防御<m，那么最大攻击的状态就可以扔掉了。

code：

#include <algorithm>
#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <sstream>
#include <stdio.h>
#include <string>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <list>
#include <set>
#include <map>
using namespace std;
 
#define N  100010
#define ALL(x)     x.begin(),x.end()
#define CLR(x,a)   memset(x,a,sizeof(x))
typedef long long    ll;
typedef pair<int,int> PI;
const int INF  = 0x3fffffff;
const int MOD  = 1000000007;
/*-----------code------------*/

int n,M,ans;
string tbl[]={
"Head", "Shoulder", "Neck", "Torso", "Hand", "Wrist", "Waist", "Legs", "Feet", "Finger", "Shield", "Weapon", "Two-Handed"
};
vector<PI> A[13],B[13];
int dp[2][50010];

int hash(string c){
	for(int i=0;i<13;i++) if(c==tbl[i]) return i;
}

void solve(){
	CLR(dp,-1);
	int now=0,pre=1;
	dp[pre][0]=0;
	for(int i=0;i<13;i++){
		CLR(dp[now],-1);
		for(int j=0;j<B[i].size();j++){
			PI &o=B[i][j];
			for(int k=0;k<=M;k++) if(dp[pre][k]!=-1){
				dp[now][min(k+o.second,M)]=max(dp[now][min(k+o.second,M)],dp[pre][k]+o.first);
			}
		}
		for(int k=0;k<=M;k++) if(dp[pre][k]!=-1) dp[now][k]=max(dp[now][k],dp[pre][k]);
		swap(pre,now);
	}
	vector<PI> &v=A[9];
	vector<PI> a,b;
	for(int i=0;i<v.size();i++){
		for(int j=i+1;j<v.size();j++){
			PI o;
			o.second=v[i].first+v[j].first;
			o.first=v[i].second+v[j].second;
			a.push_back(o);
		}
		a.push_back(make_pair(v[i].second,v[i].first));
	}
	sort(ALL(a)); 	
	for(int i=0;i<=M;i++) if(dp[pre][i]!=-1){
		b.push_back(make_pair(dp[pre][i],i));
	}
	sort(ALL(b));
	ans=max(ans,dp[pre][M]);
	for(int i=a.size()-1;i>=0;i--){
		while(!b.empty() && a[i].first+b.rbegin()->second<M) b.pop_back();
		if(b.empty()) break;
		ans=max(ans,a[i].second+b.rbegin()->first);
	}
}

int main(){
	int re;
	scanf("%d",&re);
	while(re--){
		scanf("%d%d",&n,&M);
		for(int i=0;i<13;i++) A[i].clear();
		for(int i=0;i<n;i++){
			char eq[16];
			PI o;
			scanf("%s %d %d",eq,&o.first,&o.second);
			A[hash(eq)].push_back(o);
		}
		ans=-1;
		for(int i=0;i<13;i++) if(i!=12 && i!=9) 
			B[i]=A[i]; else B[i].clear();
		solve();
		for(int i=0;i<13;i++) if(i!=11 && i!=10 && i!=9)
			B[i]=A[i]; else B[i].clear();
		solve();
		printf("%d\n",ans);
	}
	return 0;
}