結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
コンテスト
ユーザー ぷら
提出日時 2021-07-30 21:46:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,666 bytes
コンパイル時間 2,736 ms
コンパイル使用メモリ 204,380 KB
最終ジャッジ日時 2025-01-23 11:29:57
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main() {
    int N,K;
    cin >> N >> K;
    vector<int>tmp(9),a,b;
    for(int i = 0; i < 9; i++) {
        cin >> tmp[i];
        bool flag = a.size()+tmp[i] < N/2;
        for(int j = 0; j < tmp[i]; j++) {
            if(flag) {
                a.push_back(i+1);
            }
            else {
                b.push_back(i+1);
            }
        }
    }
    vector<map<int,int>>mp(1<<N);
    for(int i = 0; i < (1 << N); i++) {
        if(__builtin_popcount(i) != a.size()) {
            continue;
        }
        do {
            int cnt = 0;
            long long tmp2 = 1;
            long long tmp3 = 0;
            for(int j = 0; j < N; j++) {
                if(1 & (i >> j)) {
                    tmp3 += tmp2*a[cnt]%K;
                    tmp3 %= K;
                    cnt++;
                }
                tmp2 *= 10;
                tmp2 %= K;
            }
            mp[i][tmp3]++;
        }while(next_permutation(a.begin(),a.end()));
    }
    long long ans = 0;
    for(int i = 0; i < (1 << N); i++) {
        if(__builtin_popcount(i) != b.size()) {
            continue;
        }
        do {
            int cnt = 0;
            long long tmp2 = 1;
            long long tmp3 = 0;
            for(int j = 0; j < N; j++) {
                if(1 & (i >> j)) {
                    tmp3 += tmp2*b[cnt]%K;
                    tmp3 %= K;
                    cnt++;
                }
                tmp2 *= 10;
                tmp2 %= K;
            }
            ans += mp[((1 << N)-1)^i][(K-tmp3)%K];
        }while(next_permutation(b.begin(),b.end()));
    }
    cout << ans << endl;
}
0