結果

問題 No.1268 Fruit Rush 2
コンテスト
ユーザー tamato
提出日時 2020-10-23 23:05:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 471 ms / 2,000 ms
コード長 553 bytes
コンパイル時間 196 ms
コンパイル使用メモリ 82,520 KB
実行使用メモリ 231,720 KB
最終ジャッジ日時 2024-07-21 12:24:21
合計ジャッジ時間 7,649 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    N = int(input())
    A = list(map(int, input().split()))
    A_set = set(A)
    dic = {}
    for a in A:
        dic[a] = 1
        if a-1 not in dic:
            dic[a-1] = 0
        if a+1 not in dic:
            dic[a+1] = 0
    B = list(dic.keys())
    B.sort()
    ans = 0
    for i in range(1, len(B)):
        if B[i] - 1 in A_set:
            dic[B[i]] += dic[B[i] - 2]
        ans += dic[B[i]]
    print(ans)


if __name__ == '__main__':
    main()
0