提出 #57850582


ソースコード 拡げる

import sys
from collections import deque


def debug(*args):
    print(*args, file=sys.stderr)


n = int(input())
a = list(map(int, input().split()))

appear = [deque([]) for _ in range(n+1)]
for i, ai in enumerate(a):
    appear[ai].append(i)

# i = 0 に対する和を求める
ans = 0
element_count = 0
for j in range(n):
    if j <= appear[a[j]][0]:
        element_count += 1
    ans += element_count

g_pre = ans

for i in range(1, n):
    appear[a[i-1]].popleft()
    if len(appear[a[i-1]]) == 0:
        g = g_pre - n + i - 1
    else:
        first_idx = appear[a[i-1]][0]
        g = g_pre - first_idx + i - 1
    ans += g
    g_pre = g

print(ans)

提出情報

提出日時
問題 E - I Hate Sigma Problems
ユーザ kmmtkm
言語 Python (PyPy 3.10-v7.3.12)
得点 475
コード長 696 Byte
結果 AC
実行時間 269 ms
メモリ 219016 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 475 / 475
結果
AC × 2
AC × 27
セット名 テストケース
Sample 00_sample_01.txt, 00_sample_02.txt
All 00_sample_01.txt, 00_sample_02.txt, 01_test_01.txt, 01_test_02.txt, 01_test_03.txt, 01_test_04.txt, 01_test_05.txt, 01_test_06.txt, 01_test_07.txt, 01_test_08.txt, 01_test_09.txt, 01_test_10.txt, 01_test_11.txt, 01_test_12.txt, 01_test_13.txt, 01_test_14.txt, 01_test_15.txt, 01_test_16.txt, 01_test_17.txt, 01_test_18.txt, 01_test_19.txt, 01_test_20.txt, 01_test_21.txt, 01_test_22.txt, 01_test_23.txt, 01_test_24.txt, 01_test_25.txt
ケース名 結果 実行時間 メモリ
00_sample_01.txt AC 68 ms 76316 KiB
00_sample_02.txt AC 68 ms 76132 KiB
01_test_01.txt AC 259 ms 217992 KiB
01_test_02.txt AC 252 ms 218388 KiB
01_test_03.txt AC 255 ms 218172 KiB
01_test_04.txt AC 247 ms 218492 KiB
01_test_05.txt AC 250 ms 218120 KiB
01_test_06.txt AC 254 ms 218124 KiB
01_test_07.txt AC 251 ms 218344 KiB
01_test_08.txt AC 262 ms 218532 KiB
01_test_09.txt AC 269 ms 218692 KiB
01_test_10.txt AC 251 ms 218660 KiB
01_test_11.txt AC 221 ms 218416 KiB
01_test_12.txt AC 224 ms 218472 KiB
01_test_13.txt AC 219 ms 218344 KiB
01_test_14.txt AC 265 ms 218292 KiB
01_test_15.txt AC 233 ms 218456 KiB
01_test_16.txt AC 205 ms 218740 KiB
01_test_17.txt AC 206 ms 218664 KiB
01_test_18.txt AC 111 ms 111516 KiB
01_test_19.txt AC 90 ms 88920 KiB
01_test_20.txt AC 217 ms 193168 KiB
01_test_21.txt AC 184 ms 217836 KiB
01_test_22.txt AC 68 ms 76688 KiB
01_test_23.txt AC 70 ms 76940 KiB
01_test_24.txt AC 69 ms 76588 KiB
01_test_25.txt AC 201 ms 219016 KiB