Maalaus

#include <bits/stdc++.h>
using namespace std;

int main() {
   
    int n, m, k, q;
    cin >> n >> m >> k >> q;
   
    map<int, pair<int, int>> row_colors;
    map<int, pair<int, int>> column_colors;
   
    for (int i=0; i < q; i++){
        char type;
        int ind;
        int color;
        cin >> type >> ind >> color;
        if(type == 'R'){
            row_colors[ind] = {color, i};
        }
        if(type == 'C'){
            column_colors[ind] = {color, i};
        }
    }


    set<int> row_indexes;
    for(auto row_info : row_colors){
        row_indexes.insert(row_info.second.second);
    }
    
    set<int> col_indexes;
    for(auto col_info : column_colors){
        col_indexes.insert(col_info.second.second);
    }

    vector<long long> result;
    for(int color=1; color <= k; color++){
        long long colored_squares = 0;
        // Check each row for this color
        for(auto row_info : row_colors){
            // This row is right color
            if(row_info.second.first == color){
                auto pos = distance(col_indexes.begin(), col_indexes.lower_bound(row_info.second.second));
                colored_squares += m-(col_indexes.size()-pos);
            }
        }
        
        // Check each column for this color
        for(auto column_info : column_colors){
            // This column is right color
            if(column_info.second.first == color){
                auto pos = distance(row_indexes.begin(), row_indexes.lower_bound(column_info.second.second));
                
                colored_squares += n-(row_indexes.size()-pos);
            }
            
        }

        result.push_back(colored_squares);
    }
    
    for(int i=0; i < k; i++){
        cout << result[i] << " ";
    }
    return 0;
}

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

Test 13

Group: 2, 4

Verdict: ACCEPTED

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