Submission #62084652


Source Code Expand

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

K = []
for i in range(N):
  K.append([0,N])
  
B = []
for i in range(N):
  B.append([])
for i in range(N):
  B[A[i]-1].append(i)
  
for i in range(N):
  for j in range(len(B[i])):
    t = B[i][j]
    if j != 0:
      K[t][0] = B[i][j-1] + 1
    if i != 0:
      Min = -1
      Max = len(B[i-1])
      while Min+1 != Max:
        Mid = (Min+Max)//2
        if t > B[i-1][Mid]:
          Min = Mid
        else:
          Max = Mid
      if Min != -1:
        K[t][0] = max(K[t][0],B[i-1][Min]+1)
      if Max != len(B[i-1]):
        K[t][1] = B[i-1][Max]
        
ans = 0
for i in range(N):
  ans += (i+1-K[i][0]) * (K[i][1]-i)
  
print(ans)

Submission Info

Submission Time
Task F - Double Sum 3
User Quez9271
Language Python (PyPy 3.10-v7.3.12)
Score 525
Code Size 729 Byte
Status AC
Exec Time 297 ms
Memory 157140 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 56 ms 76448 KiB
00_sample_01.txt AC 56 ms 76508 KiB
00_sample_02.txt AC 57 ms 76356 KiB
01_handmade_00.txt AC 57 ms 76504 KiB
01_handmade_01.txt AC 165 ms 155652 KiB
01_handmade_02.txt AC 181 ms 156860 KiB
02_permutation_00.txt AC 114 ms 119932 KiB
02_permutation_01.txt AC 124 ms 118948 KiB
02_permutation_02.txt AC 265 ms 151612 KiB
02_permutation_03.txt AC 207 ms 150788 KiB
02_permutation_04.txt AC 218 ms 156108 KiB
02_permutation_05.txt AC 252 ms 157100 KiB
02_permutation_06.txt AC 268 ms 157044 KiB
02_permutation_07.txt AC 273 ms 156660 KiB
02_permutation_08.txt AC 255 ms 157140 KiB
02_permutation_09.txt AC 261 ms 156956 KiB
02_permutation_10.txt AC 174 ms 156728 KiB
02_permutation_11.txt AC 175 ms 156660 KiB
03_random_00.txt AC 114 ms 101264 KiB
03_random_01.txt AC 117 ms 106836 KiB
03_random_02.txt AC 173 ms 123740 KiB
03_random_03.txt AC 119 ms 106540 KiB
03_random_04.txt AC 157 ms 120376 KiB
03_random_05.txt AC 297 ms 151128 KiB
03_random_06.txt AC 279 ms 151364 KiB
03_random_07.txt AC 291 ms 151184 KiB
03_random_08.txt AC 286 ms 151152 KiB
03_random_09.txt AC 268 ms 151152 KiB
03_random_10.txt AC 98 ms 105304 KiB
03_random_11.txt AC 121 ms 119272 KiB
03_random_12.txt AC 132 ms 117640 KiB