Maalaus

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        int k = scanner.nextInt();
        int q = scanner.nextInt();
        scanner.nextLine();
        Map<Integer, int[]> rows = new HashMap<>();
        Map<Integer, int[]> columns = new HashMap<>();
        for (int time = 1; time <= q; time++){
            String[] op = scanner.nextLine().split(" ");
            if (op[0].equals("R")){
                int color = Integer.parseInt(op[2]);
                int row = Integer.parseInt(op[1])-1;
                rows.put(row, new int[]{time, color});
            }
            if (op[0].equals("C")){
                int color = Integer.parseInt(op[2]);
                int column = Integer.parseInt(op[1])-1;
                columns.put(column, new int[]{time, color});
            }
        }
        List<Integer> rowtimes = new ArrayList<>();
        for (int[] value : rows.values()) {
            rowtimes.add(value[0]); 
        }
        List<Integer> columntimes = new ArrayList<>();
        for (int[] value : columns.values()) {
            columntimes.add(value[0]);
        }
        int unpaintedrows = n-rows.size();
        int unpaintedcolumns = m-columns.size();
        long[] colorcounts = new long[k + 1];
        
        for (int[] value : rows.values()) {
            int rowtime = value[0];
            int rowcolor = value[1];
            int columnless = 0;
            for (int ct : columntimes){
                if (ct < rowtime){
                    columnless++;
                }
            }
            long totalcells = columnless + unpaintedcolumns;
            if  (1 <= rowcolor && rowcolor <= k) {
                colorcounts[rowcolor] += totalcells;
            }
        }
        for (int[] value : columns.values()) {
            int columntime = value[0];
            int columncolor = value[1];
            int rowless = 0;
            for (int rt : rowtimes){
                if (rt < columntime){
                    rowless++;
                }
            }
            long totalcells = rowless + unpaintedrows;
            if  (1 <= columncolor && columncolor <= k) {
                colorcounts[columncolor] += totalcells;
            }
        }
        List<String> result = new ArrayList<>();
        for (int i = 1; i <= k; i++){
            result.add(String.valueOf(colorcounts[i]));
        }
        System.out.println(String.join(" ", result));
    }
}

Test 1 (public)

Group: 1, 3, 4

Verdict: ACCEPTED

Test 2 (public)

Group: 3, 4

Verdict: ACCEPTED

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

Test 7

Group: 3, 4

Verdict: ACCEPTED

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