Leimasin

import heapq

# Lukee rivin syötettä ja palauttaa sen sisältämät luvut listana.
def read_numbers():
    return list(map(int, input().strip().split()))

def read_string():
    return input().strip()

def main():
    # lue luvut n m k
    nmk = read_numbers()
    tuloksen_koko = nmk[0]
    leiman_koko = nmk[1]
    leimasin = read_string()
    leimaukset = read_numbers()
    # Priority queue: (kohta, prioriteetti)
    leimaukset_tulossa = []
    for prioriteetti, kohta in enumerate(leimaukset):
        heapq.heappush(leimaukset_tulossa, (kohta, prioriteetti))

    leimaukset_voimassa = []
    taulu = ['.'] * tuloksen_koko
    for tulos_i in range(1, tuloksen_koko + 1):
        # Siirrä leimaukset, jotka alkavat nyt, voimassaoleviin
        while leimaukset_tulossa and leimaukset_tulossa[0][0] == tulos_i:
            kohta, prioriteetti = heapq.heappop(leimaukset_tulossa)
            heapq.heappush(leimaukset_voimassa, (-prioriteetti, kohta))
        # Poista vanhentuneet leimaukset
        while leimaukset_voimassa and leimaukset_voimassa[0][1] + leiman_koko <= tulos_i:
            heapq.heappop(leimaukset_voimassa)
        # Jos voimassaoleva leimaus, käytä sitä
        if leimaukset_voimassa:
            taulu[tulos_i - 1] = leimasin[tulos_i - leimaukset_voimassa[0][1]]
    tulos = ''.join(taulu)
    print(tulos)

if __name__ == "__main__":
    main()

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

Test 9

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED