結果

問題 No.1185 完全な3の倍数
ユーザー もりを
提出日時 2020-08-22 13:23:40
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 561 ms / 2,000 ms
コード長 1,152 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 44,032 KB
最終ジャッジ日時 2024-11-24 09:18:28
合計ジャッジ時間 25,480 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

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