Submission #29544002

Source Code Expand

import queue

N = int(input())
A = [0] * N

s = queue.LifoQueue()
num = 0
for a in map(int, input().split()):
    if s.empty():
        s.put((a, 1))
        num += 1
    else:
        before = s.get()

        if before[0] == a:
            if before[1]+1 == a:
                num -= a-1
            else:
                s.put((a, before[1]+1))
                num += 1
        else :
           s.put(before)
           s.put((a, 1))
           num += 1
    print(num)

Submission Info

Submission Time
Task D - Strange Balls
User is0292hx
Language Python (3.8.2)
Score 400
Code Size 496 Byte
Status AC
Exec Time 830 ms
Memory 28040 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 400 / 400
Status
AC × 2
AC × 22
Set Name Test Cases
Sample example_00.txt, example_01.txt
All example_00.txt, example_01.txt, test_00.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt, test_19.txt
Case Name Status Exec Time Memory
example_00.txt AC 29 ms 9576 KiB
example_01.txt AC 20 ms 9432 KiB
test_00.txt AC 830 ms 27744 KiB
test_01.txt AC 817 ms 27944 KiB
test_02.txt AC 25 ms 9300 KiB
test_03.txt AC 794 ms 16688 KiB
test_04.txt AC 792 ms 16608 KiB
test_05.txt AC 795 ms 16960 KiB
test_06.txt AC 782 ms 16848 KiB
test_07.txt AC 803 ms 16976 KiB
test_08.txt AC 802 ms 27536 KiB
test_09.txt AC 793 ms 27732 KiB
test_10.txt AC 813 ms 27536 KiB
test_11.txt AC 809 ms 27536 KiB
test_12.txt AC 800 ms 27668 KiB
test_13.txt AC 814 ms 27540 KiB
test_14.txt AC 801 ms 27840 KiB
test_15.txt AC 810 ms 27652 KiB
test_16.txt AC 795 ms 27532 KiB
test_17.txt AC 810 ms 27536 KiB
test_18.txt AC 811 ms 26524 KiB
test_19.txt AC 804 ms 28040 KiB