結果

問題 No.1294 マウンテン数列
ユーザー Kiri8128Kiri8128
提出日時 2020-11-21 01:04:25
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 330 ms / 2,000 ms
コード長 708 bytes
コンパイル時間 291 ms
コンパイル使用メモリ 82,428 KB
実行使用メモリ 81,188 KB
最終ジャッジ日時 2024-07-23 14:13:59
合計ジャッジ時間 3,777 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 49 ms
63,252 KB
testcase_01 AC 48 ms
64,592 KB
testcase_02 AC 49 ms
63,840 KB
testcase_03 AC 37 ms
51,944 KB
testcase_04 AC 38 ms
52,612 KB
testcase_05 AC 97 ms
75,516 KB
testcase_06 AC 97 ms
75,728 KB
testcase_07 AC 37 ms
52,564 KB
testcase_08 AC 37 ms
52,820 KB
testcase_09 AC 37 ms
52,468 KB
testcase_10 AC 285 ms
78,580 KB
testcase_11 AC 286 ms
78,644 KB
testcase_12 AC 320 ms
79,004 KB
testcase_13 AC 330 ms
81,188 KB
testcase_14 AC 294 ms
77,808 KB
testcase_15 AC 255 ms
78,440 KB
testcase_16 AC 301 ms
77,912 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
A = sorted([int(a) for a in input().split()])

def calc(A):
    n = len(A)
    A = A[::-1]
    M = max(a - b for a, b in zip(A, A[1:]))
    ans = 0
    prev = 0
    for k in range(M, A[0] - A[-1] + 1):
        X = [0]
        S = [0]
        j = 0
        for i, a in enumerate(A):
            if i == 0:
                continue
            if i == 1:
                X.append(1)
                S.append(1)
                continue
            while A[j] > a + k:
                j += 1
            X.append((S[-1] - S[j]) % P)
            S.append((S[-1] + X[-1]) % P)
        ans = (ans + (S[-1] - prev) * k) % P
        prev = S[-1]
    return ans * 2 % P

P = 998244353
print(calc(A))
0