We use cookies to ensure you have the best browsing experience on our website. Please read our cookie policy for more information about how we use cookies.
  1. Prepare
  2. Python
  3. Regex and Parsing
  4. Matrix Script
  5. Discussions

Matrix Script

  • Tinai4
     + 0 comments

    Wow its increadible how much easier this can be done. Thank you for sharing people. I have written solution myself, it gives the same answers in the tests but for some reason it says its not correct in here. Maybe you can help to see why

    Edit: I found that i do if r"name"== r"main" and that was the problem!

    import re

def main():
    first_multiple_input = input().rstrip().split()
    n = int(first_multiple_input[0])
    matrix = []
    for _ in range(n):
        matrix_item = input()
        matrix.append(matrix_item)
    my_string = matrix_to_string(matrix)
    print(decode(my_string))

def matrix_to_string(matrix):
    transposed_matrix = []
    for new_row in zip(*matrix):
        for s in new_row:
            transposed_matrix.append(s)
    return "".join(transposed_matrix)

def decode(my_string):
    code, part_string = find_begining(my_string)
    my_string = my_string.replace(part_string, "")
    decoded = code
    while code:
        try:
            code, part_string = find_begining(my_string)
        except AttributeError:
            break
        my_string = my_string.replace(part_string, "")
        decoded = f"{decoded} {code}"
    return decoded + my_string

def find_begining(string):
    my_match = re.match(r"(^[\W]*([\w]+))", string)
    return my_match.group(2), my_match.group(1)

if __name__ == "__main__":
    main()