Pakkaus

#! python3
# CSES 2017: Tehtävä 1

from itertools import groupby

def parser(string):
    """Split string between each and every integer"""
    return [''.join(g) for _, g in groupby(string, str)] #str.isalpha

def string_formation(int_str_list):
    """Form a string by multiplying each string by the preceeding number, then put the strings together"""
    finished_string = ""
    try: # Takes care of test cases that don't begin with a number
        int(int_str_list[0])
    except ValueError:
        finished_string += int_str_list[0]
    for i in int_str_list:
        if i.isdigit():
            finished_string += str(
                int_str_list[int_str_list.index(i)+1]*int(i)
            )
    return finished_string

def main():
    #while True:
        string_to_be_modified = input()
        parsed = parser(string_to_be_modified)
        print(string_formation(parsed))


if __name__ == "__main__":
    main()

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: WRONG ANSWER

Test 4

Group: 1

Verdict: WRONG ANSWER

Test 5

Group: 1

Verdict: WRONG ANSWER

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 2

Verdict: WRONG ANSWER

Test 9

Group: 2

Verdict: WRONG ANSWER

Test 10

Group: 2

Verdict: WRONG ANSWER

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: WRONG ANSWER

Test 14

Group: 3

Verdict: WRONG ANSWER

Test 15

Group: 3

Verdict: WRONG ANSWER