結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
ユーザー ぷら
提出日時 2021-07-30 22:28:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,361 bytes
コンパイル時間 1,989 ms
コンパイル使用メモリ 196,448 KB
最終ジャッジ日時 2025-01-23 12:09:25
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 27 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

int dp[1 << 14][1000];

int main() {
    int N,K;
    cin >> N >> K;
    vector<int>tmp(N);
    tmp[0] = 1%K;
    for(int i = 1; i < N; i++) {
        tmp[i] = tmp[i-1]*10%K;
    }
    dp[0][0] = 1;
    int cnt = 0;
    for(int i = 1; i <= 9; i++) {
        int a;
        cin >> a;
        cnt += a;
        if(a == 0) {
            continue;
        }
        for(int j = 0; j < (1 << N); j++) {
            if(__builtin_popcount(j) != cnt) {
                continue;
            }
            vector<int>res;
            for(int k = 0; k < N; k++) {
                if(1 & (j >> k)) {
                    res.push_back(k);
                }
            }
            for(int k = 0; k < (1 << res.size()); k++) {
                if(__builtin_popcount(k) != a) {
                    continue;
                }
                long long sum = 0;
                int old = j;
                for(int l = 0; l < res.size(); l++) {
                    if(1 & (k >> l)) {
                        old -= (1 << res[l]);
                        sum += tmp[res[l]]*i;
                    }
                }
                sum %= K;
                for(int l = 0; l < K; l++) {
                    dp[j][(l+sum)%K] += dp[old][l];
                }
            }
        }
    }
    cout << dp[(1 << N)-1][0] << endl;
}
0