結果

問題 No.189 SUPER HAPPY DAY
ユーザー ctyl_0
提出日時 2015-12-29 17:39:52
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 92 ms / 5,000 ms
コード長 1,953 bytes
コンパイル時間 895 ms
コンパイル使用メモリ 93,084 KB
実行使用メモリ 43,520 KB
最終ジャッジ日時 2024-09-19 08:25:50
合計ジャッジ時間 2,118 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>
#include <numeric>
#include <functional>
#include <cmath>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <sstream>
#include <string>
#define repd(i,a,b) for (int i=(a);i<(b);i++)
#define rep(i,n) repd(i,0,n)
#define var auto
#define mod 1000000007
#define mod2 1000000009
#define inf 2147483647
#define mp make_pair
#define pb push_back
typedef long long ll;
using namespace std;

template <typename T>
inline void output(T a, int p) {
    if(p){
        cout << fixed << setprecision(p)  << a << "\n";
    }
    else{
        cout << a << "\n";
    }
}
// end of template

vector<vector<vector<ll>>> sum_of_digits(string s){
    int sz = (int)s.size();
    reverse(s.begin(), s.end());
    vector<vector<vector<ll>>> dp(sz + 1, vector<vector<ll>>(sz * 9 + 1, vector<ll>(2, 0)));
    dp[0][0][0] = 1;
    rep(i, sz){
        int n = s[i] - '0';
        rep(j, sz * 9 + 1){
            rep(k, 10){
                if(j + k > sz * 9) continue;
                if (k < n) {
                    dp[i + 1][j + k][0] += dp[i][j][0] + dp[i][j][1];
                }
                else if (k == n){
                    dp[i + 1][j + k][0] += dp[i][j][0];
                    dp[i + 1][j + k][1] += dp[i][j][1];
                }
                else{
                    dp[i + 1][j + k][1] += dp[i][j][0] + dp[i][j][1];
                }
                dp[i + 1][j + k][0] %= mod2;
                dp[i + 1][j + k][1] %= mod2;
            }
        }
    }
    return dp;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(0);
    // source code
    string m, d;
    cin >> m >> d;
    
    var dpm = sum_of_digits(m);
    var dpd = sum_of_digits(d);
    
    ll ans = 0;
    repd(i, 1, min(m.size(), d.size()) * 9){
        ans += dpm[m.size()][i][0] * dpd[d.size()][i][0];
        ans %= mod2;
    }
    
    output(ans, 0);
    
    return 0;
}
0