結果

問題 No.1631 Sorting Integers (Multiple of K) Easy
ユーザー monnumonnu
提出日時 2021-07-30 21:43:30
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 831 ms / 3,000 ms
コード長 1,458 bytes
コンパイル時間 3,885 ms
コンパイル使用メモリ 231,860 KB
実行使用メモリ 131,456 KB
最終ジャッジ日時 2024-09-15 23:32:20
合計ジャッジ時間 13,869 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll=long long;
using Graph=vector<vector<pair<int,int>>>;
#define MAX 100000
#define MOD 1000000007
#define INF 1000000000


int main(){
  int N;
  int K;
  cin>>N>>K;
  vector<int> c(9);
  for(int i=0;i<9;i++){
    cin>>c[i];
  }
  vector<int> res(N,1);
  for(int i=1;i<N;i++){
    res[i]=res[i-1]*10%K;
  }
  vector<vector<ll>> dp(1<<N,vector<ll>(K,0));
  dp[0][0]=1;
  int sum=0;
  for(int i=0;i<9;i++){
    if(c[i]==0){
      continue;
    }
    vector<int> bits;;
    for(int j=0;j<(1<<N);j++){
      int cnt=0;
      int bit=0;
      for(int k=0;k<N;k++){
        if(((j>>k)&1)==1){
          bit+=(1<<k);
          cnt++;
        }
      }
      if(cnt==c[i]){
        bits.push_back(bit);
      }
    }
    int n=bits.size();
    for(int j=0;j<n;j++){
      int x=0;
      for(int k=0;k<N;k++){
        if(((bits[j]>>k)&1)==1){
          x+=res[k];
        }
      }
      //cout<<x<<endl;
      x*=i+1;
      x%=K;
      //cout<<x<<endl;
      for(int k=0;k<(1<<N);k++){
        if((bits[j]&k)!=0){
          continue;
        }
        int cnt=0;
        for(int l=0;l<N;l++){
          if(((k>>l)&1)==1){
            cnt++;
          }
        }
        if(cnt!=sum){
          continue;
        }
        for(int l=0;l<K;l++){
          dp[bits[j]+k][(l+x)%K]+=dp[k][l];
        }
      }
    }
    sum+=c[i];
  }
  cout<<dp[(1<<N)-1][0]<<endl;
}
0