“科大讯飞杯”第18届上海大学程序设计联赛春季赛暨高校网络 H、纸牌游戏 · 贪心

题解

---

---

代码

#include <bits/stdc++.h>
using namespace std;
int n, k;
string s;
int num[10], cnt[10];

// 3^9 = 2e4
bool dfs(int now, int k, int mod) {//当前数字 还要选k个 余数
    if (!now) {//特判0 题目要求不允许出现前导零
        if (k > num[0] || mod) return false;

        cnt[0] = k;//剩下的全放0
        if (k == 1) return true;
        for (int i = 1; i <= 9; ++i) {
            if (cnt[i])//判断是否会出现前导零
                return true;
        }
        cnt[0] = 0;
        return false;
    }

    if (!k && !mod) return true;

    int r = min(k, num[now]);//判断当前数字最多能放多少张
    int l = max(0, r - 2);//由于余数循环 012 放4张能满足就肯定不会只选其中2张 所以每个数字最多只有3种可能
    for (int i = r; i >= l; --i) {
        cnt[now] = i;
        if (dfs(now - 1, k - i, (mod + i * now) % 3)) return true;
        cnt[now] = 0;
    }
    return false;
}

int main() {
    ios::sync_with_stdio(0);

    int T;
    cin >> T;
    for (int cs = 1; cs <= T; ++cs) {

        memset(num, 0, sizeof(num));//统计s里每个数字出现的个数
        memset(cnt, 0, sizeof(cnt));//统计答案里每个数字出现的个数

        cin >> s >> k;
        n = s.length();
        for (int i = 0; i < n; ++i) {
            num[s[i] - '0']++;
        }

        if (dfs(9, k, 0)) {
            for (int i = 9; i >= 0; --i) {
                for (int j = 0; j < cnt[i]; ++j) {
                    cout << i;
                }
            }
            cout << endl;

        } else
            cout << -1 << endl;

    }
    return 0;
}