結果

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

ソースコード

diff # 

#include <algorithm>
#include <cmath>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <string>
#include <tuple>
#include <vector>

#define mkp make_pair
#define mkt make_tuple
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define all(v) v.begin(), v.end()
using namespace std;
typedef long long ll;
const ll MOD = 1e9 + 7;
// const ll MOD = 998244353;
template <class T>
void chmin(T &a, const T &b) {
    if (a > b) a = b;
}
template <class T>
void chmax(T &a, const T &b) {
    if (a < b) a = b;
}

void add(ll &a, ll b) {
    a = (a + b) % MOD;
}

void mul(ll &a, ll b) {
    a %= MOD;
    b %= MOD;
    a = a * b % MOD;
}

const int L = 10;

void solve() {
    int N, K;
    cin >> N >> K;
    vector<int> C(L, 0);
    for (int i = 1; i < L; i++) cin >> C[i];

    map<pair<vector<int>, int>, ll> mdp;
    mdp[mkp(C, 0)] = 1;
    for (int i = 0; i < N; i++) {
        map<pair<vector<int>, int>, ll> n_mdp;
        for (auto [key, val] : mdp) {
            auto [ds, rem] = key;
            for (int k = 1; k < L; k++) {
                if (ds[k] == 0) continue;
                ds[k]--;
                int n_rem = (rem * 10 + k) % K;
                add(n_mdp[mkp(ds, n_rem)], val);
                ds[k]++;
            }
        }
        swap(mdp, n_mdp);
    }

    ll ans = mdp[mkp(vector(L, 0), 0)];
    cout << ans << endl;
}

int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    solve();

    return 0;
}
0