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().encode()
    leimaukset = read_numbers()
    leimaukset_tulossa = sorted((-kohta, prioriteetti) for prioriteetti, kohta in enumerate(leimaukset))

    leimaukset_voimassa = []
    taulu = bytearray(b'.' * tuloksen_koko)
    for tulos_i in range(1, tuloksen_koko + 1):
        # Siirrä leimaukset, jotka alkavat nyt, voimassaoleviin
        while leimaukset_tulossa and leimaukset_tulossa[-1][0] == -tulos_i:
            kohta, prioriteetti = leimaukset_tulossa.pop()
            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]]
    print(taulu.decode())

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: ACCEPTED