题目链接:CodeForces 337C Captains Mode
题目大意:dota2游戏选择英雄,先给出n个英雄的力量值,然后给出m个操作,p/b num,p代表第num个玩家可以选择一个英雄,b 代表第num个玩家可以封掉一个英雄,即谁都不可以选,然后两队选完之后,比较说两队英雄总力量值的差,两人都按照最优方案去选择。
解题思路:dp+位运算+贪心。首先选择英雄的时候,肯定是选择能量值最大的。其次m<= 20,也就是说不管是p还b,最多也就操作到20个英雄,所以将英雄按照力量值从大到小排序,选20个进行处理即可。
然后最多20个英雄,就可以用一个二进制数来表示,1表示不可选,0表示可选。碰到玩家1选择就加上能力值最大的英雄,玩家2选就剪掉能力值最大的英雄。如果碰到封除英雄,如果玩家1封,就要选策略中尽量大的,如果玩家2选,就要选择尽量小的。
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = (1 << 20) + 5;
const int M = 105;
const int INF = 0x3f3f3f3f;
int n, m, v[N], h[M], order[M][2], rec[N];
bool cmp(const int& a, const int& b) {
return a > b;
}
void init() {
memset(v, 0, sizeof(v));
char ch;
scanf("%d", &n);
for (int i = 0; i < n; i++) scanf("%d", &h[i]);
sort(h, h + n, cmp);
scanf("%d%*c", &m);
for (int i = 0; i < m; i++) {
scanf("%c %d%*c", &ch, &order[i][0]);
order[i][1] = (ch == 'p' ? 1 : 0);
}
n = min(n, 20);
}
int dp(int s, int d) {
if (v[s]) return rec[s];
if (d >= m) return 0;
int& ans = rec[s]; v[s] = 1;
ans = 0;
while (d < m) {
if (order[d][1]) {
int id;
for (id = 0; id < n; id++) if ((s & (1 << id)) == 0) break;
s += (1 << id);
order[d][0] == 1 ? ans += h[id] : ans -= h[id];
} else {
int tmp = order[d][0] == 1 ? -INF : INF;
for (int i = 0; i < n; i++) if ((s & (1 << i)) == 0) {
int t = dp(s + (1 << i), d+1);
if (order[d][0] == 1) tmp = max(tmp, t);
else tmp = min(tmp, t);
}
ans += tmp;
break;
}
d++;
}
return ans;
}
int main () {
init();
printf("%d\n", dp(0, 0));
return 0;
}