結果

問題 No.941 商とあまり
ユーザー QCFium
提出日時 2019-09-16 11:23:09
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 397 ms / 2,000 ms
コード長 1,879 bytes
コンパイル時間 1,976 ms
コンパイル使用メモリ 183,660 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-06 12:55:46
合計ジャッジ時間 10,736 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 6
other AC * 104
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
int ri() {
    int n;
    scanf("%d", &n);
    return n;
}
std::string solve(int n, int x, const std::vector<int> &a_) {
    auto a = a_;
    if (n == 1) {
        std::string s(x, '0');
        if (a[0] <= x) s[a[0] - 1] = '1';
        return s;
    }
    std::sort(a.begin(), a.end());
    int64_t prod = 1;
    for (int i = 0; i < n; i++) {
        prod *= a[i] + 1;
        if (prod > x + 1) return std::string(x, '0');
    }
    if (std::count(a.begin(), a.end(), 1)) {
        std::string res(x, '0');
        for (int i = prod - 1; i <= x; i++) res[i - 1] = '1';
        return res;
    }
    std::vector<int> size(n + 1); // x / 3^i
    size[n] = x + 1;
    for (int i = n - 1; i >= 0; i--) size[i] = size[i + 1] / 3;
    std::vector<std::vector<bool> > dp(1 << n);
    for (int i = 0; i < 1 << n; i++) dp[i].resize(size[__builtin_popcount(i)]);
    for (int i = 0; i < n; i++) assert(a[i] < (int) dp[1 << i].size()), dp[1 << i][a[i]] = true;
    for (int i = 0; i < 1 << n; i++) {
        int cur_sz = dp[i].size();
        for (int j = 0; j < n; j++) {
            if (i >> j & 1) continue;
            std::vector<bool> tmp(dp[i | 1 << j].size());
            for (int k = 0; k < cur_sz; k++) {
                int64_t target = (int64_t) (k + 1) * (a[j] + 1) - 1;
                if (dp[i][k] && target < (int) tmp.size()) tmp[target] = true;
            }
            for (int k = 0; k + a[j] < (int) tmp.size(); k++) if (tmp[k]) tmp[k + a[j]] = true;
            for (int k = 0; k < (int) tmp.size(); k++) if (tmp[k]) dp[i | 1 << j][k] = true;
        }
    }
    std::string res;
    for (int i = 1; i <= x; i++) res.push_back(dp.back()[i] ? '1' : '0');
    return res;
}
int main() {
    int n = ri(), x = ri();
    assert(1 <= n && n <= 100);
    assert(1 <= x && x <= 500000);
    std::vector<int> a(n);
    for (int i = 0; i < n; i++) a[i] = ri(), assert(1 <= a[i] && a[i] <= x);
    clock_t r0 = clock();
    std::cout << solve(n, x, a) << std::endl;
    clock_t r1 = clock();
    std::cerr << (double)(r1 - r0) * 1000 / CLOCKS_PER_SEC << "ms" << std::endl;
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0