結果

問題 No.1634 Sorting Integers (Multiple of K) Hard
ユーザー fumofumofuni
提出日時 2022-01-12 01:06:02
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,562 bytes
コンパイル時間 2,840 ms
コンパイル使用メモリ 214,472 KB
最終ジャッジ日時 2025-01-27 10:31:24
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

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,ll>> 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;
            ll 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));
    ll mask=(1<<n)-1;
    do{
        ll rem=0;
        ll 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