結果

問題 No.315 世界のなんとか3.5
ユーザー koyumeishi
提出日時 2015-09-20 00:57:22
言語 C++11
(gcc 13.3.0)
結果
TLE  
実行時間 -
コード長 1,784 bytes
コンパイル時間 686 ms
コンパイル使用メモリ 80,100 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-07-19 08:15:19
合計ジャッジ時間 10,547 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 15 TLE * 2 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

//TLE
//桁DP
#include <iostream>
#include <vector>
#include <cstdio>
#include <sstream>
#include <map>
#include <string>
#include <algorithm>
#include <queue>
#include <cmath>
#include <set>
using namespace std;
#define MOD 1000000007
int P;
long long func(string& s){
    
    //dp[mod 3*P][lower | has3]
    long long my_mod = 3*P;
    vector<vector<long long>> dp (my_mod, vector<long long>(4, 0));
    dp[0][0] = 1;
    for(int i=0; i<s.size(); i++){
        vector<vector<long long>> dp_(my_mod, vector<long long>(4, 0));
        for(int mod3P=0; mod3P<my_mod; mod3P++)
        for(int lower=0; lower<2; lower++)
        for(int has3=0; has3<2; has3++)
        {
            int state = (lower<<1) | (has3);
            for(int d=0; d<10; d++){
                if(lower==0 && d>s[i]-'0') continue;
                int next_state = (lower || d<s[i]-'0')<<1;
                next_state |= (has3 || d==3);
                int next_mod = (mod3P*10 + d) % my_mod;
            
                long long& dp_now  = dp[mod3P][state];
                long long& dp_next = dp_[next_mod][next_state];
                dp_next += dp_now;
                if(dp_next>=MOD) dp_next -= MOD;
            }
        }
        swap(dp, dp_);
    }
    long long ret = 0;
    for(int mod3P=0; mod3P<my_mod; mod3P++)
    for(int lower=0; lower<2; lower++)
    for(int has3=0; has3<2; has3++)
    {
        int state = (lower<<1) | (has3);
        if( mod3P % P == 0 ) continue;
        if( has3==0 && mod3P%3!=0 ) continue;
        ret += dp[mod3P][state];
        if(ret>=MOD) ret -= MOD;
    }
    return ret;
}
int main(){
    string a,b;
    cin >> a >> b;
    cin >> P;
    long long ans = 0;
    ans += func(b);
    ans -= func(a);
    ans = (ans+MOD)%MOD;
    int mod3 = 0;
    int has3 = 0;
    int modP = 0;
    for(int i=0; i<a.size(); i++){
        if(a[i] == '3') has3 = 1;
        mod3 = (mod3*10 + a[i]-'0') % 3;
        modP = (modP*10 + a[i]-'0') % P;
    }
    if( (mod3==0 || has3==1) && modP!=0 ) ans += 1;
    ans = (ans+MOD)%MOD;
    cout << ans << endl;
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0