結果

提出番号	1808
提出者	yamad
言語	C++
提出日時	2018-08-04 14:00:12
問題名	(72)K-th DigitSum
結果	WA
点数	0%

---

テストケース

ソースコード

#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 dp[1001][1001][2];

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