結果

問題 No.1634 Sorting Integers (Multiple of K) Hard
ユーザー chocoruskchocorusk
提出日時 2021-07-30 21:40:57
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,285 ms / 3,000 ms
コード長 1,918 bytes
コンパイル時間 3,645 ms
コンパイル使用メモリ 196,084 KB
実行使用メモリ 88,832 KB
最終ジャッジ日時 2024-09-15 23:28:27
合計ジャッジ時間 30,971 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 63 ms
13,952 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 190 ms
5,376 KB
testcase_06 AC 172 ms
5,376 KB
testcase_07 AC 199 ms
5,376 KB
testcase_08 AC 513 ms
55,424 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 882 ms
88,704 KB
testcase_12 AC 2,285 ms
79,744 KB
testcase_13 AC 863 ms
88,704 KB
testcase_14 AC 2,052 ms
88,704 KB
testcase_15 AC 2,169 ms
88,448 KB
testcase_16 AC 856 ms
88,704 KB
testcase_17 AC 282 ms
21,760 KB
testcase_18 AC 214 ms
26,752 KB
testcase_19 AC 621 ms
43,136 KB
testcase_20 AC 280 ms
40,192 KB
testcase_21 AC 27 ms
8,320 KB
testcase_22 AC 1,594 ms
27,904 KB
testcase_23 AC 2,025 ms
47,872 KB
testcase_24 AC 1,985 ms
67,328 KB
testcase_25 AC 2,016 ms
73,472 KB
testcase_26 AC 1,865 ms
82,048 KB
testcase_27 AC 1,289 ms
85,376 KB
testcase_28 AC 1,390 ms
88,704 KB
testcase_29 AC 756 ms
88,704 KB
testcase_30 AC 802 ms
88,832 KB
testcase_31 AC 825 ms
88,704 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;
int n, k;
int c[10];
int n1,n2;
map<pair<int, ll>, int> mp;
ll ans;
void dfs(ll v, ll r, int t){
    if(t==n1){
        mp[make_pair(r, v)]++;
        return;
    }
    ll r1=(r*10+1)%k;
    for(int i=1; i<=9; i++){
        ll d=((v>>((i-1)<<2))&15);
        if(c[i]>d){
            v+=(1ll<<((i-1)<<2));
            dfs(v, r1, t+1);
            v-=(1ll<<((i-1)<<2));
        }
        r1++;
        if(r1>=k) r1-=k;
    }
}
ll q;
ll c0;
void dfs2(ll v, ll r, int t){
    if(t==n2){
        ll x=(k-q*r%k);
        if(x>=k) x-=k;
        ll w=c0-v;
        auto itr=mp.find(make_pair(x, w));
        if(itr!=mp.end()){
            ans+=itr->second;
        }
        return;
    }
    ll r1=(r*10+1)%k;
    for(int i=1; i<=9; i++){
        ll d=((v>>((i-1)<<2))&15);
        if(c[i]>d){
            v+=(1ll<<((i-1)<<2));
            dfs2(v, r1, t+1);
            v-=(1ll<<((i-1)<<2));
        }
        r1++;
        if(r1>=k) r1-=k;
    }
}
int main()
{
	cin>>n>>k;
	for(int i=1; i<=9; i++){
		cin>>c[i];
        c0^=((ll)c[i]<<((i-1)*4));
	}
    n1=min(7, n);
    dfs(0, 0, 0);
    if(n1==n){
        for(auto p:mp) if(p.first.first==0) ans+=p.second;
        cout<<ans<<endl;
        return 0;
    }
    n2=n-n1;
    q=1;
    for(int i=0; i<n1; i++) q=q*10%k;
    dfs2(0, 0, 0);
    cout<<ans<<endl;
    return 0;
}
0