結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
ユーザー mugen_1337mugen_1337
提出日時 2021-07-08 15:16:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,501 ms / 3,000 ms
コード長 837 bytes
コンパイル時間 2,306 ms
コンパイル使用メモリ 200,352 KB
最終ジャッジ日時 2025-01-22 18:12:10
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>

using namespace std;

vector<vector<long long>> dp(1<<14, vector<long long>(1001, 0));

signed main(){
	int N,K;
	cin >> N >> K;
	vector<int> cnt(10, 0), A;
	for(int i = 1; i < 10; i++){
	    cin >> cnt[i];
	    for(int j = 0; j < cnt[i]; j++) A.push_back(i);
	}
	
	auto calc_nxt = [&](int bit, int use){
		auto C = 0, fs = -1;
		for(int i = 0; i < N; i++){
			if(A[i] == use and ((bit >> i) & 1)) C++;
			if(A[i] == use and fs < 0) fs = i;
		}
		if(C >= cnt[use] or fs < 0) return -1;
		return (bit | (1 << (fs + C)));
	};
	
	dp[0][0] = 1;
	
	for(int cur = 0; cur < (1 << N); cur++){
		for(int i = 0; i < K; i++){
			for(int j = 1; j < 10; j++){
				int nxt = calc_nxt(cur, j);
				if(nxt >= 0){
					dp[nxt][(i * 10 + j) % K] += dp[cur][i];
				}
			}
		}
	}
	cout << dp[(1 << N) - 1][0] << endl;
	return 0;
}
0