Language: C++

#include <algorithm>
#include <iostream>
#include <vector>
#include <map>
#include <cstdio>
#include <string>
#include <cmath>
#include <queue>
#include <tuple>
#include <set>
#include <assert.h>
#include <sstream>
#include <string>
//#include <bits/stdc++.h>
#define maxs(x,y) x = max(x,y)
#define mins(x,y) x = min(x,y)
#define rep(i,n) for(int (i)=0;(i)<(n);(i)++)
#define repr(i, n) for (int i = (n) - 1; i >= 0; i--)
#define FOR(i,i0,n) for(int (i)=(i0);(i)<(n);(i)++)
#define FORR(i,i0,n) for(int (i)=(n)-1; (i)>=(i0);(i)--)
#define SORT(x) sort(x.begin(),x.end())
#define SORTR(x) sort(x.begin(),x.end(),greater <>())
#define rn return
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define mp make_pair
#define mt make_tuple
using namespace std;
using ll = long long;
typedef std::pair<ll,ll> P;
//#include <boost/multiprecision/cpp_int.hpp>
//using bint = boost::multiprecision::cpp_int;




// auto mod int
// https://youtu.be/L8grWxBlIZ4?t=9858
// https://youtu.be/ERZuLAxZffQ?t=4807 : optimize
// https://youtu.be/8uowVvQ_-Mo?t=1329 : division
const int mod = 1000000007;
struct mint {
    ll x; // typedef long long ll;
    mint(ll x=0):x((x%mod+mod)%mod){}
    
    mint& operator+=(const mint a) {
        if ((x += a.x) >= mod) x -= mod;
        return *this;
    }
    
    mint& operator-=(const mint a) {
        if ((x += mod-a.x) >= mod) x -= mod;
        return *this;
    }
    mint& operator*=(const mint a) {
        (x *= a.x) %= mod;
        return *this;
    }
    mint operator+(const mint a) const {
        mint res(*this);
        return res+=a;
    }
    mint operator-(const mint a) const {
        mint res(*this);
        return res-=a;
    }
    mint operator*(const mint a) const {
        mint res(*this);
        return res*=a;
    }
    mint pow(ll t) const {
        if (!t) return 1;
        mint a = pow(t>>1);
        a *= a;
        if (t&1) a *= *this;
        return a;
    }
    
    // for prime mod
    mint inv() const {
        return pow(mod-2);
    }
    mint& operator/=(const mint a) {
        return (*this) *= a.inv();
    }
    mint operator/(const mint a) const {
        mint res(*this);
        return res/=a;
    }
};
// combination mod prime
struct combination {
    vector<mint> fact, ifact;
    combination(int n):fact(n+1),ifact(n+1) {
        assert(n < mod);
        fact[0] = 1;
        for (int i = 1; i <= n; ++i) fact[i] = fact[i-1]*i;
        ifact[n] = fact[n].inv();
        for (int i = n; i >= 1; --i) ifact[i-1] = ifact[i]*i;
    }
    mint operator()(int n, int k) {
        if (k < 0 || k > n) return 0;
        return fact[n]*ifact[k]*ifact[n-k];
    }
};




mint ones(string n){
    mint ans;
    ans = 0;
    int leng = (int) n.size();
    int maxint = n[0] - '0';
    rep(i,leng-1){
        ans += 9;
    }
    ans += maxint;
    rn ans;
}



mint twos(string n){
    mint ans = 0;
    int leng = (int) n.size();
    int maxint = n[0] - '0';
    
    rep(i,leng-2){
        mint add = 1;
        add *= 81;
        add *= (i+1);
        ans += add;
    }
    
    mint add = 1;
    add *= 9;
    add *= (leng-1)*(maxint-1);
    ans += add;
    string s = n;
    if (s.size() >= 2) { s = s.erase(0,1); ans += ones(s)};
    
    return ans;
}



mint threes(string n){
    mint ans = 0;
    
    int leng = (int) n.size();
    int maxint = n[0] - '0';
    
    rep(i,leng-2){
        mint add = 1;
        add*=9*9*9;
        add*= ((i+1)*(i))/2;
        ans += add;
    }
    mint add = 1;
    add *= 9*9;
    add *= ((leng-1)*(leng-2)*(maxint-1))/2;
    ans += add;
    

    string s = n;
    
    int i = 0;
    s = s.erase(0,1);
    while (s[i] == '0' && s.size()){
        s = s.erase(0,1);
    }
    if (s.size()) ans += twos(s);
    return ans;
}


int main(){
    string n;
    cin >> n;
    int k;
    cin >> k;
    mint ans = 0;
    if (k==1){
        ans = ones(n);
    }
    
    if (k==2){
        ans = twos(n);
    }
    
    if (k==3){
        ans = threes(n);
    }
    
    
    cout << ans.x;
}


Compiler output:
a.cc: In function 'mint twos(std::string)':
a.cc:143:58: error: expected ';' before '}' token
  143 |     if (s.size() >= 2) { s = s.erase(0,1); ans += ones(s)};
      |                                                          ^
      |                                                          ;