#142417 (C++14) No.315 世界のなんとか3.5

提出ソース

結果

問題	No.315 世界のなんとか3.5
ユーザー	tossy
提出日時	2017-01-01 22:21:40
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	905 ms / 2,000 ms
コード長	2,034 bytes
コンパイル時間	953 ms
コンパイル使用メモリ	94,104 KB
実行使用メモリ	5,376 KB
最終ジャッジ日時	2024-05-09 15:10:26
合計ジャッジ時間	13,489 ms
ジャッジサーバーID （参考情報）	judge2 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	6 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	6 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	1 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	228 ms 5,376 KB
testcase_13	AC	222 ms 5,376 KB
testcase_14	AC	454 ms 5,376 KB
testcase_15	AC	394 ms 5,376 KB
testcase_16	AC	436 ms 5,376 KB
testcase_17	AC	490 ms 5,376 KB
testcase_18	AC	210 ms 5,376 KB
testcase_19	AC	266 ms 5,376 KB
testcase_20	AC	224 ms 5,376 KB
testcase_21	AC	227 ms 5,376 KB
testcase_22	AC	415 ms 5,376 KB
testcase_23	AC	392 ms 5,376 KB
testcase_24	AC	411 ms 5,376 KB
testcase_25	AC	394 ms 5,376 KB
testcase_26	AC	407 ms 5,376 KB
testcase_27	AC	484 ms 5,376 KB
testcase_28	AC	406 ms 5,376 KB
testcase_29	AC	754 ms 5,376 KB
testcase_30	AC	751 ms 5,376 KB
testcase_31	AC	847 ms 5,376 KB
testcase_32	AC	797 ms 5,376 KB
testcase_33	AC	489 ms 5,376 KB
testcase_34	AC	850 ms 5,376 KB
testcase_35	AC	905 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <cassert>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <map>
#include <bitset>
#include <functional>
#include <numeric>
using namespace std;

#define repl(i,a,b) for(int i=(int)(a);i<(int)(b);i++)
#define rep(i,n) repl(i,0,n)
#define mp(a,b) make_pair((a),(b))
#define pb(a) push_back((a))
#define all(x) (x).begin(),(x).end()
#define dbg(x) cout<<#x"="<<((x))<<endl
#define fi first
#define se second

#define INF 2147483600
#define MOD 1000000007
#define long long long


long dp[2][2][2][3][800];
// [-][EQ?][3を使った][mod3][mod p]

// 数値，8の倍数を加算するか，8の倍数でないものを加算するか，sとイコールを含めるか
long solve(string &s, int p, bool eq=true){
  int n = s.size();

  auto prev = dp[0];
  auto crnt = dp[1];
  fill(prev[0][0][0], prev[2][0][0], 0);
  prev[1][0][0][0] = 1;

  rep(i,max(n-5,0)){
    fill(crnt[0][0][0], crnt[2][0][0], 0);
    rep(j,2) rep(k,3){
      rep(m,10) (crnt[0][j|(m==3)][(k*10+m)%3][0] += prev[0][j][k][0]) %= MOD;
      int mx = s[i]-'0';
      rep(m,mx+1) (crnt[m==mx][j|(m==3)][(k*10+m)%3][0] += prev[1][j][k][0]) %= MOD;
    }
    swap(prev, crnt);
  }

  repl(i,max(n-5,0),n){
    fill(crnt[0][0][0], crnt[2][0][0], 0);
    rep(j,2) rep(k,3) rep(l,p){
      rep(m,10) (crnt[0][j|(m==3)][(k*10+m)%3][(l*10+m)%p] += prev[0][j][k][l]) %= MOD;
      int mx = s[i]-'0';
      rep(m,mx+1) (crnt[m==mx][j|(m==3)][(k*10+m)%3][(l*10+m)%p] += prev[1][j][k][l]) %= MOD;
    }
    swap(prev, crnt);
  }

  long ret = 0;
  rep(i,2) if(i==0 || eq){
    rep(j,3)repl(k,1,p) ret = (ret+prev[i][1][j][k])%MOD; // 3を使った
    repl(k,1,p) ret = (ret+prev[i][0][0][k])%MOD; // 3を使わずmod3=0
  }
  return ret;
}

int main(){
  std::cin.tie(0); std::ios::sync_with_stdio(false);
  string a,b;
  int p;
  cin>>a>>b>>p;

  cout << (solve(b,p,true) - solve(a,p,false) + MOD)%MOD << endl;

  return 0;
}

yukicoder

結果

テストケース

ソースコード