結果

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

---

テストケース

ソースコード

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll> P;

#define fi first
#define se second
#define repl(i,a,b) for(ll i=(ll)(a);i<(ll)(b);i++)
#define rep(i,n) repl(i,0,n)
#define all(x) (x).begin(),(x).end()
#define dbg(x) cout<<#x"="<<x<<endl
#define mmax(x,y) (x>y?x:y)
#define mmin(x,y) (x<y?x:y)
#define maxch(x,y) x=mmax(x,y)
#define minch(x,y) x=mmin(x,y)
#define uni(x) x.erase(unique(all(x)),x.end())
#define exist(x,y) (find(all(x),y)!=x.end())
#define bcnt __builtin_popcount

#define INF 1e16
#define mod 1000000007

ll N,K;
ll dp2[1001][1001];

int main(){
  cin>>N>>K;
  dp2[0][0]=1;
  rep(i,1000)rep(j,N+1){
    rep(d,10){
      if(j+d<=N)dp2[i+1][j+d]+=dp2[i][j];
    }
  }

  ll d=1;
  while(1){
    ll cnt=0;
    repl(k,1,10){
      if(N-k>=0)cnt+=dp2[d-1][N-k];
    }
    if(cnt>=K)break;
    K-=cnt;
    d++;
  }
  string res;
  ll rest=N;
  rep(i,d){
    repl(k,i==0?1:0,10){
      ll nrest=rest-k;
      if(dp2[d-1-i][nrest]<K){
        K-=dp2[d-1-i][nrest];
      }else{
        res+=to_string(k);
        rest-=k;
        break;
      }
    }
  }
  cout<<res<<endl;
  return 0;
}