結果

問題 No.2374 ASKT Subsequences
コンテスト
ユーザー 21kazu13
提出日時 2023-07-07 22:53:19
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,593 bytes
コンパイル時間 603 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 141,696 KB
最終ジャッジ日時 2024-07-21 19:09:12
合計ジャッジ時間 6,252 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
import sys
from bisect import bisect_left, bisect_right, insort_left, insort_right  # type: ignore
from collections import Counter, defaultdict, deque  # type: ignore
from math import gcd, sqrt, ceil  # type: ignore
from heapq import heapify, heappop, heappush, heappushpop, heapreplace, merge  # type: ignore
from itertools import accumulate, combinations, permutations, product  # type: ignore
from string import ascii_lowercase, ascii_uppercase # type: ignore

def LI(): return list(map(int, sys.stdin.buffer.readline().split()))
def I(): return int(sys.stdin.buffer.readline())
def LS(): return sys.stdin.buffer.readline().rstrip().decode("utf-8").split()
def S(): return sys.stdin.buffer.readline().rstrip().decode("utf-8")
def IR(n): return [I() for _ in range(n)]
def LIR(n): return [LI() for _ in range(n)]
def SR(n): return [S() for _ in range(n)]
def LSR(n): return [LS() for _ in range(n)]
def SRL(n): return [list(S()) for _ in range(n)]

N = I()
A = LI()
num = [[] for _ in range(2020)]
cnt = [[0]*(N+1) for _ in range(2020)]
for i,a in enumerate(A):
    num[a].append(i)
    for j in num[a-1]:
        cnt[a-1][j+1]+=1
        cnt[a-1][i]-=1

for i in range(2020):
    cnt[i] = list(accumulate(cnt[i]))
for i in range(2019,0,-1):
    for j in range(N+1):
        cnt[i-1][j] += cnt[i][j]

ans = 0
for i in range(N):
    a1 = A[i]
    for k in num[a1+10]:
        if k-i<2: continue
        a3 = A[k]
        # if cnt[a3+1][k]>0:
        #     print(i,k,a3+1,cnt[a3+1][k])
        #     print(cnt[a3+1])
        ans += cnt[a3+1][k]-cnt[a3+1][i]
print(ans)
0