結果
問題 |
No.1185 完全な3の倍数
|
ユーザー |
![]() |
提出日時 | 2020-08-22 14:05:44 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 32 ms / 2,000 ms |
コード長 | 1,146 bytes |
コンパイル時間 | 70 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 11,008 KB |
最終ジャッジ日時 | 2024-11-24 09:27:08 |
合計ジャッジ時間 | 2,291 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 |
ソースコード
import math def bf(n): d = int(math.log10(n)) + 1 digits = [int(c) for c in str(n)] cnt = 0 for i in range(12, n+1, 3): if i < 100: cnt += 1 elif all([c in '0369' for c in str(i)]): cnt += 1 return cnt def solve(n): d = int(math.log10(n)) + 1 digits = [int(c) for c in str(n)] leading_zero = [0, 0, 30] + [3*4**i for i in range(2, 10)] if n < 100: cnt = 0 for i in range(12, n+1, 3): cnt += 1 return (cnt) else: dp = [[0]*2 for i in range(d)] GREATEST = 0 LESS = 1 dp[0][GREATEST] = 1 if digits[0] % 3 == 0 else 0 dp[0][LESS] = max(digits[0]-1, 0)//3 for i in range(1, d): dp[i][GREATEST] = dp[i-1][GREATEST] * \ (1 if digits[i] % 3 == 0 else 0) if digits[i] > 0: dp[i][LESS] = dp[i-1][GREATEST] * (max(digits[i]-1, 0)//3 + 1) dp[i][LESS] += dp[i-1][LESS] * 4 return (sum(dp[d-1]) + sum(leading_zero[:d])) def main(): n = int(input()) print(solve(n)) if __name__ == "__main__": main()