結果

問題 No.3257 +|+
ユーザー LyricalMaestro
提出日時 2025-09-07 10:14:48
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 749 bytes
コンパイル時間 409 ms
コンパイル使用メモリ 82,232 KB
実行使用メモリ 248,260 KB
最終ジャッジ日時 2025-09-07 10:15:01
合計ジャッジ時間 11,827 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7 TLE * 1 -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/3257


    

def main():
    N = int(input())
    A = list(map(int, input().split()))

    max_a = max(A)

    dp = [{} for _ in range(2 * max_a + 1)]
    answer = 0
    for i in range(N):
        a = A[i]
        i0 = i + 1

        # answer計算
        b = -a + i0
        m = 1
        while max_a >= b:
            if m in dp[b + max_a]:
                answer += dp[b + max_a][m]
            b += i0
            m += 1

        # dp更新
        b = a - i0
        m = 1
        while -max_a <= b:
            if m not in dp[b + max_a]:
                dp[b + max_a][m] = 0
            dp[b + max_a][m] += 1
            b -= i0
            m += 1

    print(answer)


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