Maalaus

input/code.cpp: In function 'void print_colors(ulonglong*, uint)':
input/code.cpp:21:23: warning: comparison of integer expressions of different signedness: 'int' and 'uint' {aka 'unsigned int'} [-Wsign-compare]
   21 |     for (int i = 0; i < color_count; i++) {
      |                     ~~^~~~~~~~~~~~~

#include <iostream>
#include <vector>
#include <unordered_set>



const int MAX_INPUT = 200000;

typedef unsigned long long ulonglong;

typedef unsigned long ulong;

typedef unsigned int uint;

typedef struct Operation {
    char rowtype;
    uint color;
} Operation;

void print_colors(ulonglong color_sums[], uint color_count) {
    for (int i = 0; i < color_count; i++) {
        std::cout << color_sums[i] << ' ';
    }
    std::cout << '\n';
}


int main() {
    ulong height, width;
    uint color_count, operation_count;

    Operation operations[MAX_INPUT];

    uint operation_index = 0;

    char rowtypes[MAX_INPUT];
    ulong indices[MAX_INPUT];
    uint colors[MAX_INPUT];

    std::unordered_set<ulong> seen_row_indices;
    std::unordered_set<ulong> seen_col_indices;

    uint modified_row_count = 0;
    uint modified_col_count = 0;

    std::cin >> height >> width >> color_count >> operation_count;

    for (uint i = 0; i < operation_count; i++) {
        char rowtype;
        ulong index;
        uint color;
        std::cin >> rowtype >> index >> color;

        rowtypes[i] = rowtype;
        indices[i] = index;
        colors[i] = color;
    }

    for (int i = operation_count - 1; i >= 0; i--) {
        char rowtype = rowtypes[i];
        ulong index = indices[i];
        uint color = colors[i] - 1; // convert 1-20000 to 0-19999

        bool was_inserted;
        if (rowtype == 'R') {
            was_inserted = seen_row_indices.insert(index).second;
            if (was_inserted) { modified_row_count++; }
        } else {
            was_inserted = seen_col_indices.insert(index).second;
            if (was_inserted) { modified_col_count++; }
        }

        if (was_inserted) {
            Operation operation;
            operation.rowtype = rowtype;
            operation.color = color;
            operations[operation_index++] = operation;
        }
    }
    
    ulong unaffected_rows = height - modified_row_count;
    ulong unaffected_cols = width - modified_col_count;

    ulonglong color_sums[MAX_INPUT];

    for (uint i = 0; i < modified_row_count + modified_col_count; i++) {
        Operation operation = operations[i];
        if (operation.rowtype == 'R') {
            color_sums[operation.color] += unaffected_cols;
        } else {
            color_sums[operation.color] += unaffected_rows;
        }
    }
    //print_colors(color_sums, color_count);

    ulong current_row_width = modified_col_count;
    ulong current_col_width = modified_row_count;

    for (uint i = 0; i < modified_row_count + modified_col_count - 1; i++) {
        Operation operation = operations[i];
        if (operation.rowtype == 'R') {
            current_col_width--;
            color_sums[operation.color] += current_row_width;
            //std::cout << "Added " << current_row_width << " to " << operation.color << std::endl;
        } else {
            current_row_width--;
            color_sums[operation.color] += current_col_width;
            //std::cout << "Added " << current_col_width << " to " << operation.color << std::endl;
        }
    }

    for (uint i = 0; i < color_count; i++) {
        std::cout << color_sums[i] << ' ';
    }
    std::cout << '\n';

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

Test 5

Group: 2, 4

Verdict: ACCEPTED

Test 6

Group: 3, 4

Verdict: ACCEPTED

Test 7

Group: 3, 4

Verdict: ACCEPTED

Test 8

Group: 4

Verdict: ACCEPTED

Test 9

Group: 4

Verdict: ACCEPTED

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

Test 15

Group: 4

Verdict: ACCEPTED

Test 16

Group: 4

Verdict: ACCEPTED

Test 17

Group: 4

Verdict: ACCEPTED