Maalaus

from dataclasses import dataclass, field

@dataclass
class GridLine:
    main_color: int = 0
    removed_positions: set[int] = field(default_factory=set)

@dataclass(frozen=True)
class Operation:
    is_row: bool
    index: int
    color: int

def parse_op(op):
    type_char, index, color = op
    return Operation(type_char == "R", int(index) - 1, int(color))

height, width, n_colors, n_ops = (int(x) for x in input().split())
ops = [parse_op(input().split()) for _ in range(n_ops)]

rows = [GridLine() for _ in range(height)]
painted_rows: list[GridLine] = []
columns = [GridLine() for _ in range(width)]
painted_columns: list[GridLine] = []

for op in ops:
    dir1 = rows if op.is_row else columns
    dir2 = columns if op.is_row else rows
    painted_dir1 = painted_rows if op.is_row else painted_columns
    painted_dir2 = painted_columns if op.is_row else painted_rows
    
    main_line = dir1[op.index]

    if main_line.main_color == 0:
        painted_dir1.append(main_line)
        for col in painted_dir2:
            col.removed_positions.add(op.index)
    else:
        for removed_pos in main_line.removed_positions:
            dir2[removed_pos].removed_positions.add(op.index)
        main_line.removed_positions.clear()
        
    main_line.main_color = op.color
        
color_counts = [0 for _ in range(n_colors)]

for row in rows:
    if row.main_color != 0:
        color_counts[row.main_color - 1] += width - len(row.removed_positions)
for col in columns:
    if col.main_color != 0:
        color_counts[col.main_color - 1] += height - len(col.removed_positions)

print(" ".join(str(c) for c in color_counts))

Test 1 (public)

Group: 1, 3, 4

Verdict: ACCEPTED

Test 2 (public)

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 1, 3, 4

Verdict: ACCEPTED

Test 4

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 3, 4

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 1, 3, 4

Verdict: ACCEPTED

Test 11

Group: 1, 3, 4

Verdict: ACCEPTED

Test 12

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 4

Verdict: TIME LIMIT EXCEEDED