Round 967 (Div 2) Problem D [Test case needed]

#	User	Rating
1	tourist	3993
2	jiangly	3743
3	orzdevinwang	3707
4	Radewoosh	3627
5	jqdai0815	3620
6	Benq	3564
7	Kevin114514	3443
8	ksun48	3434
9	Rewinding	3397
10	Um_nik	3396

#	User	Contrib.
1	cry	167
2	Um_nik	163
3	maomao90	162
3	atcoder_official	162
5	adamant	159
6	-is-this-fft-	158
7	awoo	155
8	TheScrasse	154
9	Dominater069	153
10	djm03178	152

Hi,

I am stuck at 2001D - Longest Max Min Subsequence. My logic is similar to other accepted solutions. Also, my solution works for all the small test cases I can find from other people's accepted solutions. Some test cases are big and truncated on CF, so I could not test my code with those cases.

Can someone please point out the bug/ provide a test case for which the code does not work?

Thanks in advance.

My submission: 280104081

my solution

import heapq


def solve(nums):
    uq = set()
    cum = [0] * len(nums)
    for i, v in enumerate(nums[::-1]):
        uq.add(v)
        cum[i] = len(uq)
    cum = cum[::-1]

    min_heap, max_heap = [], []
    for i, v in enumerate(nums):
        heapq.heappush(min_heap, (-cum[i], v, i))
        heapq.heappush(max_heap, (-cum[i], -v, i))

    last_idx = -1
    ans, used = [], set()
    for i in range(len(uq)):
        if i%2: # pick small
            while min_heap and (min_heap[0][2] < last_idx or min_heap[0][1] in used):
                heapq.heappop(min_heap)
            now = heapq.heappop(min_heap)
            ans.append(now[1])
            used.add(now[1])
            last_idx = now[2]
            continue

        while max_heap and (max_heap[0][2] < last_idx or -max_heap[0][1] in used):
            heapq.heappop(max_heap)
        now = heapq.heappop(max_heap)
        ans.append(-now[1])
        used.add(-now[1])
        last_idx = now[2]

    return ans

if __name__ == '__main__':
    for _ in range(int(input())):
        n = int(input())
        num = list(map(int, input().split()))

        ans_ = solve(num)
        print(len(ans_))
        print(*ans_)

DhurBaal's blog