Codeforces Gym 100676G Training Camp 状压dp

http://codeforces.com/gym/100676

题目大意是告诉你要修n门课，每门课有一个权值w[i], 在第k天修该课程讲获得k*w[i]的学习点数，给出了课程与先修课程的关系，要修该课程必须修完先修课程。问最多能学到多少点数。

非常简单的一道状压dp(一开始我还误导队友写成两维的去了 T^T);

dp[s] : s 的二进制存放的是已经选择的课程，在该状态下的能获得的最大的点数。

这时如果再学一门课程k，将转移到状态ss (s | (1 << k) ) ，能否转移需要判断合法性。这里把课程k的先修课程存在一个int里面，S & prev[k] == prev[k]则表示可以选项k课程。dp数组一开始需要初始化成无穷小。最后只要找dp[(1 << n ) - 1] 。

/*输入真恶心*/

#include <map>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int INF = 999999999;

int n,m;

map<string, int> Map;

int w[20];

int prve[20];

void read() {
    cin >> n >> m;getchar();

    char _a[101];
    string _s, _t;
    int _num;
    int _len;
    for (int i=0; i<n; i++) {
        gets(_a);

        _len = strlen(_a);

        int k = 0;
        while(_a[k] < '0' || _a[k] > '9') k++;

        _a[k-1] = '';
        _s = (string) _a;

        _num = 0;
        while (k < _len && _a[k] >= '0' && _a[k] <= '9') {
            _num = _num * 10 + _a[k] - '0';
            k++;
        }

        Map[_s] = i;
        w[i] = _num;
    }

    for (int i=0; i<m; i++) {
        gets(_a);

        int k = 0;
        while (_a[k] != '-') k++;

        _a[k-1] = '';

        _s = (string) _a;
        _t = (string) (_a + k + 4);

        prve[Map[_t]] |= (1 << Map[_s]);
    }
}


int dp[(1<<18)+5];

void DP () {

    for (int s=0; s<(1<<n); s++) {

        int _num = 0;

        for (int i=0; i<n; i++) {
            _num += ((s >> i) & 1);
        }

        for (int i=0; i<n; i++) {
            int ss = s | (1 << i);

            if (((s >> i & 1) != 0) || ((s & prve[i]) != prve[i])) continue;

            if (_num == 0 && prve[i] == 0) {
                dp[ss] = w[i];
                continue;
            }

            dp[ss] = max(dp[ss], dp[s] + (_num + 1) * w[i]);
        }
    }

    cout << dp[(1<<n)-1] << endl;
}

void init() {
    Map.clear();
    fill(dp, dp + (1 << 18), -INF);
    memset(prve, 0, sizeof(prve));
}

int main () {
    int t;
    cin >> t ;
    while (t--) {
        init();
        read();
        DP();
    }
    return 0;
}