結果

問題 No.653 E869120 and Lucky Numbers
コンテスト
ユーザー zbxah
提出日時 2018-02-28 14:12:50
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,142 bytes
コンパイル時間 171 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-12-18 00:44:57
合計ジャッジ時間 2,181 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

p = input()  # 桁がデカすぎるので文字列のまま
count = len(p)  # 桁数


def main() -> bool:

    # まず 1 桁の数字はナシ
    if count <= 1:
        return False

    # 先頭文字が 1, 6, 7, 8 ではない (但し 2 桁の時は 1, 7, 8 ではない) 場合は違う
    test_set = {'1', '6', '7', '8'} if count > 2 else {'1', '7', '8'}
    if p[0] not in test_set:
        return False

    # 最終文字が 2, 3, 4 のいずれかでない場合は違う
    if p[-1] not in {'2', '3', '4'}:
        return False

    # 先頭文字が 1, 8 の場合は中間文字のいずれも 3, 4, 5 であること
    middle = p[1:-1]
    if p[0] in {'1', '8'}:
        three, four, five = middle.count('3'), middle.count('4'), middle.count('5')
        return three + four + five == count - 2

    # 先頭文字が 6, 7 の場合は 3, 4, 5, 6, 7, 8 で構成され 8 は 1 つ以下であること
    if p[0] in {'6', '7'}:
        counts = [middle.count(str(i)) for i in range(3, 9)]
        return counts[5] <= 1 and sum(counts) == count - 2
    else:
        raise AssertionError


print('Yes' if main() else 'No')
0