Submission #62113376

Source Code Expand

n = int(input())
a = list(map(int, input().split()))
g = [[] for i in range(n + 1)]
for i in range(n):
    g[a[i] - 1].append(i)

def f(x):
    res = 0
    for i in range(1, len(x)):
        c = x[i] - x[i - 1]
        res += c * (c - 1) // 2
    return res

ans = 0
for i in range(1, n + 1):
    ans += f([-1] + g[i] + [n])
    ans -= f([-1] + sorted(g[i - 1] + g[i]) + [n])
print(ans)

Submission Info

Submission Time
Task F - Double Sum 3
User sounansya
Language Python (PyPy 3.10-v7.3.12)
Score 525
Code Size 405 Byte
Status AC
Exec Time 312 ms
Memory 138400 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 525 / 525
Status
AC × 3
AC × 31
Set Name Test Cases
Sample 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt
All 00_sample_00.txt, 00_sample_01.txt, 00_sample_02.txt, 01_handmade_00.txt, 01_handmade_01.txt, 01_handmade_02.txt, 02_permutation_00.txt, 02_permutation_01.txt, 02_permutation_02.txt, 02_permutation_03.txt, 02_permutation_04.txt, 02_permutation_05.txt, 02_permutation_06.txt, 02_permutation_07.txt, 02_permutation_08.txt, 02_permutation_09.txt, 02_permutation_10.txt, 02_permutation_11.txt, 03_random_00.txt, 03_random_01.txt, 03_random_02.txt, 03_random_03.txt, 03_random_04.txt, 03_random_05.txt, 03_random_06.txt, 03_random_07.txt, 03_random_08.txt, 03_random_09.txt, 03_random_10.txt, 03_random_11.txt, 03_random_12.txt
Case Name Status Exec Time Memory
00_sample_00.txt AC 62 ms 76540 KiB
00_sample_01.txt AC 59 ms 76356 KiB
00_sample_02.txt AC 60 ms 76416 KiB
01_handmade_00.txt AC 60 ms 76524 KiB
01_handmade_01.txt AC 200 ms 138080 KiB
01_handmade_02.txt AC 239 ms 138124 KiB
02_permutation_00.txt AC 137 ms 106136 KiB
02_permutation_01.txt AC 161 ms 113232 KiB
02_permutation_02.txt AC 307 ms 133288 KiB
02_permutation_03.txt AC 298 ms 130384 KiB
02_permutation_04.txt AC 305 ms 133988 KiB
02_permutation_05.txt AC 312 ms 137960 KiB
02_permutation_06.txt AC 308 ms 138400 KiB
02_permutation_07.txt AC 308 ms 137964 KiB
02_permutation_08.txt AC 301 ms 138228 KiB
02_permutation_09.txt AC 312 ms 137988 KiB
02_permutation_10.txt AC 251 ms 138000 KiB
02_permutation_11.txt AC 243 ms 138032 KiB
03_random_00.txt AC 119 ms 92944 KiB
03_random_01.txt AC 120 ms 95096 KiB
03_random_02.txt AC 189 ms 117964 KiB
03_random_03.txt AC 122 ms 94936 KiB
03_random_04.txt AC 153 ms 105364 KiB
03_random_05.txt AC 283 ms 137732 KiB
03_random_06.txt AC 292 ms 137768 KiB
03_random_07.txt AC 295 ms 138388 KiB
03_random_08.txt AC 282 ms 138244 KiB
03_random_09.txt AC 282 ms 137844 KiB
03_random_10.txt AC 95 ms 93920 KiB
03_random_11.txt AC 127 ms 106008 KiB
03_random_12.txt AC 127 ms 103956 KiB