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

#	User	Rating
1	tourist	4009
2	jiangly	3773
3	Radewoosh	3646
4	ecnerwala	3624
5	jqdai0815	3620
5	Benq	3620
7	orzdevinwang	3612
8	Geothermal	3569
8	cnnfls_csy	3569
10	gyh20	3447

#	User	Contrib.
1	cry	161
2	awoo	160
2	maomao90	160
4	atcoder_official	156
4	-is-this-fft-	156
6	nor	155
7	adamant	153
8	maroonrk	152
8	Um_nik	152
10	djm03178	146

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