結果
問題 |
No.2374 ASKT Subsequences
|
ユーザー |
|
提出日時 | 2023-07-07 22:51:52 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,587 bytes |
コンパイル時間 | 161 ms |
コンパイル使用メモリ | 82,700 KB |
実行使用メモリ | 219,820 KB |
最終ジャッジ日時 | 2024-07-21 19:07:40 |
合計ジャッジ時間 | 11,136 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 WA * 2 |
other | AC * 1 WA * 22 OLE * 2 -- * 3 |
ソースコード
#!/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)