Submission #74771944

---

Source Code Expand

#include <bits/stdc++.h>
using ull = unsigned long long int;
using ll = long long int;
using lll = __int128;
using namespace std;
const int mod = 1e9 + 7;
int d[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
inline string sread() {
    string s;
    char ch = getchar();
    // 跳过空白字符
    while (ch == ' ' || ch == '\n' || ch == '\r') ch = getchar();
    // 读取非空白字符
    while (ch != ' ' && ch != '\n' && ch != '\r' && ch != EOF) {
        s += ch;
        ch = getchar();
    }
    return s;
}
ll read(){
    ll k = 0,f = 1;
    char c = getchar();
    while(c < '0' || c > '9')
    {
        if(c == '-')f = -1;
        c = getchar();
    }
    while(c >= '0' && c <= '9')k = k * 10 + c - '0',c = getchar();
    return k * f; // 别忘记标记的负数要乘进去
}
// 调用时用 n = in();
void print(ll x){
    if(x < 0)putchar('-'),x = -x;
    if(x < 10)putchar(x + '0');
    else print(x / 10),putchar(x % 10 + '0');
}
// 直接调用 out(n) 就行了
ll ksm(ll base, ll e){
    ll res = 1;
    while(e){
        if(e & 1){
            res = res * base % mod;
        }
        base = base * base % mod;
        e >>= 1;
    }
    return res;
}//ksm板子
vector<vector<ll>> matrix_mul (vector<vector<ll>>& a, vector<vector<ll>>& b) {
    int l1 = a.size(), l2 = b[0].size();
    vector<vector<ll>>res(l1, vector<ll>(l2, 0));
    for(int i = 0; i < l1; i++){
        for(int k = 0; k < a[0].size(); k++){
            if(a[i][k]){
                for(int j = 0; j < l2; j++){
                    res[i][j] = (res[i][j] + a[i][k] * b[k][j]);
                }
            }
        }
    }
    return res;
}// 矩阵a * 矩阵b(矩阵乘法)
void Solve(){
    ll n = read();
    string num = to_string(n);
    int w = num.size();
    ll memo[20][2][2][180][11];
    memset(memo, -1, sizeof(memo));
    auto DP = [&](auto && self, int len, int is_limit, int is0, int sum, int pre) -> ll{
        if(len == w){
            return sum * (is0 + 1);
        }
        if(memo[len][is_limit][is0][sum][pre] != -1){
            return memo[len][is_limit][is0][sum][pre];
        }
        int cur = num[len] - '0';
        ll res = 0;
        if(is_limit){
            if(pre == 10){
                res = (res + self(self, len + 1, 0, 0, 0, 10)) % mod;
                for(int j = 1; j <= cur; j++){
                    res = (res + self(self, len + 1, j == cur, 0, 0, j)) % mod;
                }
            }
            else{
                for(int j = 0; j <= cur; j++){
                    res = (res + self(self, len + 1, j == cur, is0 | (j == 0), sum + abs(j - pre), j)) % mod;
                }
            }
        }
        else{
            if(pre == 10){
                res = (res + self(self, len + 1, 0, 0, 0, 10)) % mod;
                for(int j = 1; j < 10; j++){
                    res = (res + self(self, len + 1, 0, 0, 0, j)) % mod;
                }
            }
            else{
                for(int j = 0; j < 10; j++){
                    res = (res + self(self, len + 1, 0, is0 | (j == 0), sum + abs(pre - j), j)) % mod;
                }
            }
        }
        return memo[len][is_limit][is0][sum][pre] = res;
    };
    cout << DP(DP, 0, 1, 0, 0, 10) << '\n';
}
signed main(){
    int T = 1;
    while(T--){
        Solve();
    }
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function 'std::vector<std::vector<long long int> > matrix_mul(std::vector<std::vector<long long int> >&, std::vector<std::vector<long long int> >&)':
./Main.cpp:53:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |         for(int k = 0; k < a[0].size(); k++){
      |                        ~~^~~~~~~~~~~~~

Judge Result