1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115
| #include <bits/stdc++.h>
#pragma optimization_level 3 #pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,O3") #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx") #pragma GCC optimize("Ofast") #pragma GCC target("avx,avx2,fma") #pragma GCC optimization ("unroll-loops")
using namespace std;
struct PairHash {inline std::size_t operator()(const std::pair<int, int> &v) const { return v.first * 31 + v.second; }};
#define Code ios_base::sync_with_stdio(false); #define By ios::sync_with_stdio(0); #define Sumfi cout.tie(NULL);
using ll = long long; using ld = long double;
const ld PI = 3.14159265358979323846; const ll INF = 1e18; const ld EPS = 1e-9; const ll MAX_N = 505050; const ll mod = 998244353;
typedef pair<ll, ll> pll; typedef vector<pll> vpll; typedef array<ll,3> all3; typedef array<ll,5> all5; typedef vector<all3> vall3; typedef vector<all5> vall5; typedef vector<ld> vld; typedef vector<ll> vll; typedef vector<vll> vvll; typedef vector<int> vi; typedef vector<string> vs; typedef vector<vs> vvs; typedef unordered_set<ll> usll; typedef unordered_set<pll, PairHash> uspll; typedef unordered_map<ll, ll> umll; typedef unordered_map<pll, ll, PairHash> umpll;
#define rep(i,m,n) for(ll i=m;i<n;i++) #define rrep(i,m,n) for(ll i=n;i>=m;i--) #define all(a) begin(a), end(a) #define rall(a) rbegin(a), rend(a) #define ZERO(a) memset(a,0,sizeof(a)) #define MINUS(a) memset(a,0xff,sizeof(a)) #define sz(x) ll((x).size()) #define pyes cout<<"Yes\n"; #define pno cout<<"No\n"; #define pneg1 cout<<"-1\n"; #define CASE(x) cout<<"Case #"<<x<<": ";
template <typename T> void print(T &&t) { cout << t << "\n"; } template<typename T> void printv(vector<T>v){ll n=v.size();rep(i,0,n)cout<<v[i]<<" ";cout<<"\n";}
ll __gcd(ll x, ll y) { return !y ? x : __gcd(y, x % y); } ll __lcm(ll x, ll y) { return x * y / __gcd(x,y); } ll modpow(ll n, ll x, ll MOD = mod) { if(!x) return 1; ll res = modpow(n,x>>1,MOD); res = (res * res) % MOD; if(x&1) res = (res * n) % MOD; return res; }
ll dp[101][4][2];
ll helper(string& s, ll k, ll p, ll free) { if(!k) return 1; if(p >= sz(s)) return 0; ll& res = dp[p][k][free]; if(res != -1) return res; res = 0;
if(free) { res += helper(s,k,p+1,true); res += helper(s,k-1,p+1,true) * 9; } else { if(s[p] == '0') res += helper(s,k,p+1,false); else { res += helper(s,k,p+1,true); res += helper(s,k - 1,p+1,true) * (s[p] - '1'); res += helper(s,k-1,p+1,false); } }
return res; }
ll solve(string s, ll k) { MINUS(dp); return helper(s, k, 0, false); }
int main() { Code By Sumfi
cout.precision(12);
ll tc = 1; rep(i,1,tc+1) { string s; ll k; cin>>s>>k; print(solve(s,k)); } return 0; }
|