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

提出ソース

結果

問題	No.315 世界のなんとか3.5
ユーザー	tossy
提出日時	2017-01-01 22:25:32
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	961 ms / 2,000 ms
コード長	2,040 bytes
コンパイル時間	1,045 ms
コンパイル使用メモリ	94,224 KB
実行使用メモリ	5,376 KB
最終ジャッジ日時	2024-05-09 15:12:44
合計ジャッジ時間	14,929 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
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	7 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	3 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	278 ms 5,376 KB
testcase_13	AC	250 ms 5,376 KB
testcase_14	AC	491 ms 5,376 KB
testcase_15	AC	495 ms 5,376 KB
testcase_16	AC	389 ms 5,376 KB
testcase_17	AC	480 ms 5,376 KB
testcase_18	AC	287 ms 5,376 KB
testcase_19	AC	247 ms 5,376 KB
testcase_20	AC	275 ms 5,376 KB
testcase_21	AC	255 ms 5,376 KB
testcase_22	AC	455 ms 5,376 KB
testcase_23	AC	459 ms 5,376 KB
testcase_24	AC	392 ms 5,376 KB
testcase_25	AC	463 ms 5,376 KB
testcase_26	AC	477 ms 5,376 KB
testcase_27	AC	514 ms 5,376 KB
testcase_28	AC	451 ms 5,376 KB
testcase_29	AC	879 ms 5,376 KB
testcase_30	AC	928 ms 5,376 KB
testcase_31	AC	925 ms 5,376 KB
testcase_32	AC	961 ms 5,376 KB
testcase_33	AC	509 ms 5,376 KB
testcase_34	AC	941 ms 5,376 KB
testcase_35	AC	914 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][800][2][2][3];
// [-][mod p][EQ?][3を使った][mod3]

// 数値，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[800][0][0], 0);
  prev[0][1][0][0] = 1;

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

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

  long ret = 0;
  rep(i,2) if(i==0 || eq){
    rep(j,3)repl(k,1,p) ret = (ret+prev[k][i][1][j])%MOD; // 3を使った
    repl(k,1,p) ret = (ret+prev[k][i][0][0])%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

結果

テストケース

ソースコード