結果

問題 No.1185 完全な3の倍数
ユーザー もりをもりを
提出日時 2020-08-22 13:23:40
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 560 ms / 2,000 ms
コード長 1,152 bytes
コンパイル時間 109 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 44,032 KB
最終ジャッジ日時 2024-05-03 10:12:31
合計ジャッジ時間 22,882 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 448 ms
43,908 KB
testcase_01 AC 456 ms
43,748 KB
testcase_02 AC 458 ms
43,780 KB
testcase_03 AC 457 ms
43,904 KB
testcase_04 AC 453 ms
43,644 KB
testcase_05 AC 472 ms
43,756 KB
testcase_06 AC 463 ms
43,640 KB
testcase_07 AC 454 ms
43,640 KB
testcase_08 AC 452 ms
43,624 KB
testcase_09 AC 457 ms
44,004 KB
testcase_10 AC 460 ms
44,004 KB
testcase_11 AC 468 ms
43,620 KB
testcase_12 AC 457 ms
43,908 KB
testcase_13 AC 460 ms
44,008 KB
testcase_14 AC 458 ms
43,748 KB
testcase_15 AC 450 ms
43,884 KB
testcase_16 AC 465 ms
43,648 KB
testcase_17 AC 451 ms
44,024 KB
testcase_18 AC 467 ms
43,752 KB
testcase_19 AC 560 ms
43,904 KB
testcase_20 AC 477 ms
44,012 KB
testcase_21 AC 452 ms
43,772 KB
testcase_22 AC 470 ms
44,008 KB
testcase_23 AC 463 ms
43,620 KB
testcase_24 AC 459 ms
44,008 KB
testcase_25 AC 459 ms
43,780 KB
testcase_26 AC 466 ms
43,880 KB
testcase_27 AC 465 ms
43,744 KB
testcase_28 AC 463 ms
43,900 KB
testcase_29 AC 471 ms
43,752 KB
testcase_30 AC 474 ms
43,648 KB
testcase_31 AC 461 ms
44,032 KB
testcase_32 AC 471 ms
43,904 KB
testcase_33 AC 481 ms
43,780 KB
testcase_34 AC 459 ms
43,904 KB
testcase_35 AC 472 ms
43,904 KB
testcase_36 AC 469 ms
43,904 KB
testcase_37 AC 472 ms
43,652 KB
testcase_38 AC 475 ms
43,524 KB
testcase_39 AC 458 ms
44,032 KB
testcase_40 AC 469 ms
43,640 KB
testcase_41 AC 486 ms
43,756 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np

def cond1(x):
    return len(str(x)) >= 2 and x % 3 == 0

def cond2(x):
    x = str(x)
    D = len(x)
    for i in range(D):
        for j in range(i + 1, D):
            p = int(x[i]) + int(x[j])
            if p % 3 != 0:
                return False
    return True

def cond3(x):
    x = str(x)
    D = len(x)
    for i in x:
        if int(i) % 3 == 0:
            return False
    return True

def cond(x):
    return cond1(x) and cond2(x)

n = int(input()) + 1    # 未満

if n <= 1000:
    res = 0
    for i in range(n):
        res += cond(i)
    print(res)
    exit(0)

s = str(n)
d = len(s)

res = 0
for j in range(min(100, n)):
    if cond(j) and cond3(j):
        res += 1

dp = np.zeros((d + 1, 2), dtype='i')
dp[0, 0] = 1

for i in range(d):
    dig = int(s[i])
    # dp[.][0] : 小さいことが未確定
    for j in range(0, 10, 3):
        if j > dig:
            break
        if j == dig:
            dp[i + 1][0] += dp[i][0]
        else:
            dp[i + 1][1] += dp[i][0]
    # dp[.][1] : 小さいことが確定
    for j in range(0, 10, 3):
        dp[i + 1][1] += dp[i][1]

print(dp[d, 1] + res - 4)
0