結果

提出番号 1888
提出者 haji
言語 C++
提出日時 2018-08-04 14:24:07
問題名 (72)K-th DigitSum
結果 AC
点数 100%

テストケース

テストケース 結果 得点 実行時間 メモリ使用量
1 AC 100% 4ms 11328KB
2 AC 100% 4ms 10096KB
3 AC 100% 3ms 8400KB
4 AC 100% 2ms 8416KB
5 AC 100% 5ms 12224KB
6 AC 100% 5ms 12896KB
7 AC 100% 5ms 12064KB
8 AC 100% 2ms 8416KB
9 AC 100% 6ms 13440KB
10 AC 100% 5ms 13056KB
11 AC 100% 2ms 8672KB
12 AC 100% 2ms 8352KB
13 AC 100% 3ms 8432KB
14 AC 100% 4ms 10256KB
15 AC 100% 5ms 12064KB
16 AC 100% 4ms 10688KB
17 AC 100% 5ms 12640KB
18 AC 100% 5ms 12496KB
19 AC 100% 3ms 8624KB
20 AC 100% 5ms 11520KB
21 AC 100% 5ms 11088KB
22 AC 100% 2ms 8432KB
23 AC 100% 4ms 9840KB
24 AC 100% 2ms 8256KB
25 AC 100% 2ms 7792KB
26 AC 100% 2ms 8608KB
27 AC 100% 5ms 11792KB
28 AC 100% 6ms 12896KB
29 AC 100% 2ms 7808KB
30 AC 100% 6ms 13552KB
31 AC 100% 6ms 14896KB
32 AC 100% 4ms 10032KB
33 AC 100% 4ms 9968KB
34 AC 100% 3ms 9248KB
35 AC 100% 4ms 9936KB
36 AC 100% 5ms 14112KB
37 AC 100% 5ms 13168KB
38 AC 100% 3ms 8352KB
39 AC 100% 6ms 13184KB
40 AC 100% 5ms 12224KB
41 AC 100% 2ms 8352KB
42 AC 100% 4ms 10384KB
43 AC 100% 3ms 8192KB
44 AC 100% 3ms 9056KB
45 AC 100% 5ms 12080KB
46 AC 100% 3ms 9248KB
47 AC 100% 2ms 7632KB
48 AC 100% 2ms 8256KB
49 AC 100% 4ms 10448KB
50 AC 100% 2ms 8416KB
51 AC 100% 2ms 7968KB

ソースコード

#include <bits/stdc++.h>
#define GET_MACRO(_1,_2,_3,_4,_5,_6,_7,_8,NAME,...) NAME
#define pr(...) cerr<< GET_MACRO(__VA_ARGS__,pr8,pr7,pr6,pr5,pr4,pr3,pr2,pr1)(__VA_ARGS__) <<endl
#define pr1(a) (#a)<<"="<<(a)<<" "
#define pr2(a,b) pr1(a)<<pr1(b)
#define pr3(a,b,c) pr1(a)<<pr2(b,c)
#define pr4(a,b,c,d) pr1(a)<<pr3(b,c,d)
#define pr5(a,b,c,d,e) pr1(a)<<pr4(b,c,d,e)
#define pr6(a,b,c,d,e,f) pr1(a)<<pr5(b,c,d,e,f)
#define pr7(a,b,c,d,e,f,g) pr1(a)<<pr6(b,c,d,e,f,g)
#define pr8(a,b,c,d,e,f,g,h) pr1(a)<<pr7(b,c,d,e,f,g,h)
using namespace std;
using Int = long long;
using _int = int;
using ll = long long;
using Double = long double;
const Int INF = (1LL<<55)+1e9; // ~ 3.6 * 1e16
const Int mod = (1e9)+7;
const Double EPS = 1e-8;
const Double PI = 6.0 * asin((Double)0.5);
using P = pair<Int,Int>;
using T = tuple<Int,Int,Int>;
template<class T> T Max(T &a,T b){return a=max(a,b);}
template<class T> T Min(T &a,T b){return a=min(a,b);}
ostream& operator<<(ostream& o,P p){return o<<"("<<p.first<<","<<p.second<<")";}
ostream& operator<<(ostream& o,T t){return o<<"("<<get<0>(t)<<","<<get<1>(t)<<","<<get<2>(t)<<")";}
istream& operator>>(istream& i,P &p){return i>>p.first>>p.second;}
template<class T> ostream& operator<<(ostream& o,vector<T> &a){Int i=0;for(auto t:a)o<<(i++?" ":"")<<t;return o;}
template<class T> istream& operator>>(istream& i,vector<T> &a){for(auto &t:a)i>>t;return i;}
template<class T> void prArr(T a,string s=" "){Int i=0;for(auto t:a)cout<<(i++?s:"")<<t;cout<<endl;}

const Int N = 1010;
Int mem[2][N+10][N+10]; //length, sum
Int used[2][N+10][N+10];
Int n, K;

Int dfs(Int flag, Int len,Int sum){
  if(sum < 0) return 0;
  if(len == 0) return sum == 0 && flag;
  if(used[flag][len][sum]++) return mem[flag][len][sum];

  Int res = 0;
  for(Int i=0;i<10;i++){
    if(flag == 0 && i == 0) continue; 
    Int nflag = flag | (i!=0);
    res += dfs(nflag,len-1, sum-i);
    Min(res, INF);
  }
  return mem[flag][len][sum] = res;
}

string ans;
void dfs2(Int flag,Int len,Int sum,Int K){
  if(len == 0) assert(flag == 1 && K == 1 && sum == 0);
  if(len == 0) return;
  Int num = 0;
  Int cnt = 0;
  for(Int i=0;i<10;i++){
    if(flag == 0 && i == 0) continue;
    num = i;
    Int nflag = flag | (num!=0);
    Int a = dfs(nflag, len - 1, sum - num );
    if(cnt +  a >= K) break;
    cnt += a;
  }
  ans += char('0' + num);
  Int nflag = flag | (num!=0);
  dfs2(nflag, len-1, sum - num, K - cnt);


}

signed main(){
  srand((unsigned)time(NULL));
  cin.tie(0);
  ios_base::sync_with_stdio(0);
  cout << fixed << setprecision(12);

  cin>>n>>K;

  Int len = 1;
  Int sum = 0;
  while(1){
    Int a = dfs(0, len, n);
    if(sum + a >= K) break;
    sum += a;
    len++;
  }
  K-= sum;
  dfs2(0, len, n, K);
  cout<<ans<<endl;
  
  return 0;
}