糖果_状态压缩DP

P1460 - [蓝桥杯2019初赛]糖果 - New Online Judgehttp://oj.ecustacm.cn/problem.php?id=1460

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
int dp[1 << 20]; // dp[i]为得到口味为i时需要的最少数量
int sg[128] = { 0 }; // sg[i]存第i包糖的口味
int main() {
    // system("chcp 65001");
    // freopen("C:/Users/zhaochen/Desktop/input.txt", "r", stdin);
    cin.tie(0);
    cout.tie(0);
    int n, m, k;
    cin >> n >> m >> k;
    memset(dp, INF, sizeof(dp));
    int total = (1 << m) - 1; // 二进制是m个1，表示所有m种口味
    for (int i = 0; i < n; i++) {
        int sugar = 0; // 这一包糖的口味
        for (int j = 0; j < k; j++) {
            int t;
            cin >> t;
            sugar |= 1 << (t - 1);
        }
        dp[sugar] = 1; // 得到口味为sugar时需要的最少数量为1
        sg[i] = sugar;
    }
    for (int i = 1; i <= total; i++) { // 遍历所有口味组合
        if (dp[i] != INF) { // 如果口味i是可以得到的，在其中加入一包糖果，更新新的组合状态
            for (int j = 0; j < n; j++) {
                int sugar = sg[j];
                // 在组合i中加入第j包糖果的口味组合
                if (dp[i] + 1 < dp[i | sugar]) {
                    dp[i | sugar] = dp[i] + 1;
                }
            }
        }
    }
    if (dp[total] == INF) {
        cout << -1;
    } else {
        cout << dp[total];
    }
    return 0;
}