結果
問題 |
No.315 世界のなんとか3.5
|
ユーザー |
![]() |
提出日時 | 2020-05-05 18:53:14 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,476 bytes |
コンパイル時間 | 140 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 15,804 KB |
最終ジャッジ日時 | 2024-06-27 00:31:32 |
合計ジャッジ時間 | 9,825 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 11 WA * 25 |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines MOD = 10**9 + 7 A, B, P = readline().split() P = int(P) A = [0, 0, 0, 0, 0] + [int(x) - ord('0') for x in A] B = [0, 0, 0, 0, 0] + [int(x) - ord('0') for x in B] def solve(A, P, include=True): total = 0 no_3 = 0 cnt = (0, 1, 2, 3, 3, 4, 5, 6, 7, 8) MOD3 = 3 * MOD for x in A[:-6]: total = (10 * total + x) % MOD3 no_3 = (9 * no_3 + cnt[x]) % MOD n = no_3 * 3 no_3 = [n, n, n] x = A[-6] for d in range(x): if d != 3: no_3[(total + d) % 3] += 1 total = (total * 10 + x) % MOD3 use_3 = (total - sum(no_3)) % MOD full_mod_3 = total % 3 full_use_3 = 3 in A[:-5] U = 0 ret = 0 for x in range(10**5): if x % P == 0: continue if '3' in str(x): ret += total else: ret += use_3 + no_3[(-x) % 3] for x in A[-5:]: U = 10 * U + x for x in range(U + include): if x % P == 0: continue if '3' in str(x) or full_use_3: ret += 1 else: ret += (full_mod_3 + x) % 3 == 0 return ret def solve_naive(A, P, include): N = A + include ret = 0 for x in range(N + 1): if x % P == 0: continue ret += '3' in str(x) or x % 3 == 0 return ret x = solve(B, P, True) - solve(A, P, False) print(x % MOD)