結果

問題 No.315 世界のなんとか3.5
ユーザー koyumeishi
提出日時 2015-12-08 23:29:47
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 2,341 bytes
コンパイル時間 836 ms
コンパイル使用メモリ 73,900 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-14 20:26:27
合計ジャッジ時間 2,487 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:92:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   92 |         scanf("%s %s %d", a_,b_, &p);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~

ソースコード

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

#include <vector>
#include <iostream>
#include <string>
#include <cstdio>
using namespace std;
constexpr long long MOD = 1000000007;
vector<vector<vector<pair<int,int>>>> memo(10, vector<vector<pair<int,int>>>(16));
void make_memo(){
    for(int k=0; k<10; k++){
        for(int a=0; a<2; ++a)
        for(int c=0; c<2; ++c)
        for(int b=0; b<(c==0?3:1); ++b){
            int s = (a<<3)|(b<<1)|c;
            vector<int> next(16, 0);
            for(int d=0; d<10; ++d){
                if(a==0 && d>k) break;
                int a_ = a | (d<k);
                int b_ = (b+d)%3;
                int c_ = c | (d==3);
                if(c_) b_ = 0;
                int s_ = (a_<<3)|(b_<<1)|c_;
                next[s_]++;
            }
            for(int i=0; i<16; i++){
                if(next[i]>0) memo[k][s].push_back({i, next[i]});
            }
        }
    }
}
long long calc(string& x, int p){
    vector<vector<long long>> dp(800, vector<long long>(16, 0));
    vector<vector<long long>> dp_(800, vector<long long>(16, 0));
    dp[0][0] = 1;
    int sz = x.size();
    for(int i=0; i<sz; ++i){
        int k = x[i] - '0';
        for(int j=0; j<((i<sz-4)?1:p); ++j){
            for(int a=0; a<2; ++a)
            for(int c=0; c<2; ++c)
            for(int b=0; b<(c==0?3:1); ++b){
                int s=(a<<3)|(b<<1)|c;
                long long val = dp[j][s];
                dp[j][s] = 0;
                if(i<sz-5){
                    for(auto& p: memo[k][s]){
                        dp_[0][p.first] += val * p.second;
                        if(dp_[0][p.first] >= MOD){
                            dp_[0][p.first] %= MOD;
                        }
                    }
                }else for(int d=0; d<10; ++d){
                    if(a==0 && d>k) break;
                    int a_ = a | (d<k);
                    int b_ = (b+d)%3;
                    int c_ = c | (d==3);
                    if(c_) b_ = 0;
                    int s_ = (a_<<3)|(b_<<1)|c_;
                    int j_ = (j*10+d)%p;
                    dp_[j_][s_] += val;
                    if(dp_[j_][s_]>=MOD) dp_[j_][s_]-=MOD;
                }
            }
        }
        swap(dp,dp_);
    }
    long long ret = 0;
    for(int j=1; j<p; ++j)
    for(int a=0; a<2; ++a)
    for(int c=0; c<2; ++c)
    for(int b=0; b<(c==0?3:1); ++b){
        if(b!=0 && c==0) continue;
        int s=(a<<3)|(b<<1)|c;
        ret += dp[j][s];
        if(ret>=MOD) ret-=MOD;
    }
    return ret;
}
int main(){
    char a_[200010];
    char b_[200010];
    int p;
    //cin >> a >> b >> p;
    scanf("%s %s %d", a_,b_, &p);
    string a(a_);
    string b(b_);
    
    make_memo();
    long long ans = calc(b, p);
    ans = (ans - calc(a,p) + MOD) % MOD;
    int x=0;
    int y=0;
    int z=0;
    for(auto& c:a){
        int k = c-'0';
        y |= k==3;
        if(y==0) x = (x+k)%3;
        z = (z*10+k)%p;
    }
    if((x==0 || y==1) && z!=0){
        ans++;
        ans %= MOD;
    }
    printf("%lld\n", ans);
    //cout << ans << endl;
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0