HihoCoder 1271 艦隊遊戲（二進制枚舉、貪心）

題意:

n行∗m列≤16個空位,每個空位有個權值ai,b物品權值bi,c物品權值ci
每個空位只能放置一個物品,且物品貢獻爲ai∗bj或者ai∗cj
現要求滿足a物品貢獻≥s的情況下,最大的c物品貢獻，不滿足輸出“Not Exist”
否則輸出最大的c物品貢獻,且如果每行有至少有一個c物品輸出“Yes”，否則“No”

分析:

雖然物品很多，但其實有用的是最大的那幾個
先給a排個降序，16這個數，顯然二進制枚舉，枚舉後就可以貪心了，a大的乘b,c大的
枚舉b物品的放置情況，由於a已經降序，顯然b也從大到小選
剩下的肯定按照a從大到小放c，自然是能放多少是多少，也從大到小
注意物品個數的邊界情況，問題解決
時間復雜度O(nm∗2nm)

代碼:

//
//  Created by TaoSama on 2016-03-06
//  Copyright (c) 2016 TaoSama. All rights reserved.
//
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>

using namespace std;
#define pr(x) cout << #x << " = " << x << "  "
#define prln(x) cout << #x << " = " << x << endl
const int N = 1e5 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7;

int n, m, t, s;
int a[16], r[16];

bool cmp(int x, int y) {
    return a[x] > a[y];
}

int main() {
#ifdef LOCAL
    freopen("C:\\Users\\TaoSama\\Desktop\\in.txt", "r", stdin);
//  freopen("C:\\Users\\TaoSama\\Desktop\\out.txt","w",stdout);
#endif
    ios_base::sync_with_stdio(0);

    int q; scanf("%d", &q);
    while(q--) {
        scanf("%d%d%d%d", &n, &m, &t, &s);
        int nm = n * m;
        for(int i = 0; i < nm; ++i) scanf("%d", a + i), r[i] = i;
        sort(r, r + nm, cmp);

        vector<int> air, ship;
        for(int i = 1; i <= t; ++i) {
            int x; scanf("%d", &x);
            if(x) air.push_back(x);
        }
        for(int i = 1; i <= t; ++i) {
            int x; scanf("%d", &x);
            if(x) ship.push_back(x);
        }
        sort(air.begin(), air.end(), greater<int>());
        sort(ship.begin(), ship.end(), greater<int>());

        int maxAtk = -INF, yes = 0;
        for(int mask = 0; mask < 1 << nm; ++mask) {
            int f, g, each, d, atk;
            f = g = each = d = atk = 0;
            for(int i = 0; i < nm; ++i) {
                if(mask >> i & 1) {
                    if(f < air.size()) d += a[r[i]] * air[f++];
                } else {
                    if(g < ship.size()) {
                        atk += a[r[i]] * ship[g++];
                        each |= 1 << (r[i] / m);
                    }
                }
            }
            each = each == (1 << n) - 1;
            if(d >= s) {
                if(atk > maxAtk) {
                    maxAtk = atk;
                    yes = each;
                } else if(atk == maxAtk) yes |= each;
            }
        }

        if(maxAtk != -INF) {
            printf("%d\n", maxAtk);
            puts(yes ? "Yes" : "No");
        } else puts("Not Exist");
    }
    return 0;
}