結果

問題 No.1634 Sorting Integers (Multiple of K) Hard
ユーザー fumofumofunifumofumofuni
提出日時 2022-01-12 01:18:35
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 2,969 ms / 3,000 ms
コード長 2,566 bytes
コンパイル時間 2,547 ms
コンパイル使用メモリ 221,512 KB
実行使用メモリ 202,664 KB
最終ジャッジ日時 2023-08-09 06:25:46
合計ジャッジ時間 20,117 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,384 KB
testcase_03 AC 3 ms
4,380 KB
testcase_04 AC 2 ms
4,384 KB
testcase_05 AC 389 ms
4,380 KB
testcase_06 AC 364 ms
4,628 KB
testcase_07 AC 387 ms
4,380 KB
testcase_08 AC 1,698 ms
108,604 KB
testcase_09 AC 1 ms
4,384 KB
testcase_10 AC 1 ms
4,384 KB
testcase_11 AC 2,944 ms
202,460 KB
testcase_12 AC 2,812 ms
165,380 KB
testcase_13 AC 2,969 ms
202,312 KB
testcase_14 AC 2,911 ms
202,216 KB
testcase_15 AC 2,919 ms
201,840 KB
testcase_16 AC 2,939 ms
202,664 KB
testcase_17 AC 866 ms
63,608 KB
testcase_18 AC 848 ms
63,512 KB
testcase_19 AC 1,451 ms
98,192 KB
testcase_20 AC 322 ms
29,984 KB
testcase_21 AC 47 ms
8,532 KB
testcase_22 AC 1,093 ms
45,028 KB
testcase_23 AC 2,177 ms
105,484 KB
testcase_24 AC 1,664 ms
81,448 KB
testcase_25 AC 2,588 ms
147,776 KB
testcase_26 AC 2,136 ms
114,564 KB
testcase_27 AC 2,883 ms
172,408 KB
testcase_28 AC 2,521 ms
150,900 KB
testcase_29 AC 2,960 ms
191,012 KB
testcase_30 AC 2,959 ms
193,664 KB
testcase_31 AC 2,942 ms
196,812 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,n) for(ll i=0;i<n;i++)
#define repl(i,l,r) for(ll i=(l);i<(r);i++)
#define per(i,n) for(ll i=(n)-1;i>=0;i--)
#define perl(i,r,l) for(ll i=r-1;i>=l;i--)
#define fi first
#define se second
#define pb push_back
#define ins insert
#define pqueue(x) priority_queue<x,vector<x>,greater<x>>
#define all(x) (x).begin(),(x).end()
#define CST(x) cout<<fixed<<setprecision(x)
#define vtpl(x,y,z) vector<tuple<x,y,z>>
#define rev(x) reverse(x);
using ll=long long;
using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using pl=pair<ll,ll>;
using vpl=vector<pl>;
using vvpl=vector<vpl>;
const ll MOD=1000000007;
const ll MOD9=998244353;
const int inf=1e9+10;
const ll INF=4e18;
const ll dy[9]={0,1,-1,0,1,1,-1,-1,0};
const ll dx[9]={1,0,0,-1,1,-1,1,-1,0};
template<class T> inline bool chmin(T& a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T& a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}


int main(){
    ll n,k;cin >> n >> k;
    vpl v(9);
    rep(i,9){
        ll a;cin >> a;v.push_back({a,i+1});
    }
    sort(all(v));rev(all(v));
    ll hx=0,hy=0;
    vpl vx,vy;
    rep(i,9){
        if(hx>=hy){
            hy+=v[i].first;vy.push_back(v[i]);
        }
        else{
            hx+=v[i].first;vx.push_back(v[i]);
        }
    }
    vector<unordered_map<ll,int>> dp(1<<n);
    //ll cnt=0;
    {
        vl now;
        for(auto p:vx){
            rep(_,p.first)now.emplace_back(p.second);
        }
        while(now.size()<n)now.emplace_back(0);
        sort(all(now));
        do{
            //cnt++;
            ll rem=0;
            int bit=0;
            rep(i,n){
                if(now[i])bit|=(1<<i);
                rem=rem*10+now[i];
            }
         	rem%=k;
            dp[bit][rem]++;
        }while(next_permutation(all(now)));
    }
    //cout << cnt << endl;
    //cnt=0;
    ll ans=0;
    vl now;
    for(auto p:vy){
        rep(_,p.first)now.emplace_back(p.second);
    }
    while(now.size()<n)now.emplace_back(0);
    sort(all(now));
    int mask=(1<<n)-1;
    do{
        ll rem=0;
        int bit=0;
        rep(i,n){
            if(now[i])bit|=(1<<i);
            rem=(rem*10+now[i]);
        }
      	rem%=k;
        //rep(i,n)cout << now[i] <<" ";cout << endl;
        rem=(-rem+k)%k;
        if(dp[mask^bit].count(rem))ans+=dp[mask^bit][rem];
        //cnt++;
    }while(next_permutation(all(now)));
    cout << ans << endl;
    //cout << cnt << endl;
}
0