Maalaus

#include <iostream>
#include <string>
#include <vector>
#include <map>

using namespace std;

class Line
{
public:
    int main_color, n, dir;
    string test;
    vector<int> off_colors_list = {};
    map<int, int> off_colors = {}; // value is the place and value is which color
    void add_off_color(int pos, int color) {
        if (color == main_color) 
        {
            return;
        }
        
        if (!off_colors.count(pos))
        {
            off_colors_list.push_back(color);
        }
        off_colors[pos] = color;
        //printf("Added off color (%d) to line (%d)(%d) to pos (%d). \n", color, n, dir, pos);
        
    }
    void paint_line(int new_color){
        //printf("Painted line (%d)(%d).\n", n, dir);
        off_colors.clear();
        off_colors_list.clear();
        //printf("Size of off_colors_list: %d\n", off_colors_list.size());
        main_color = new_color;
    }
    vector<int> get_color_list()
    {
        return off_colors_list;
    }
};



int main()
{   
    const int ROW = -1;
    const int COLUMN = 1;
    // Get starting values 
    int n, m, k, q;
    cin >> n;
    cin >> m;
    cin >> k;
    cin >> q;

    vector<vector<int>> operations;
    // get operations
    for (int i = 0; i < q; i++)
    {
        int direction, n, color;
        string dir;
        cin >> dir;
        if (dir == "R")
        {
            direction = ROW;
        }
        else
        {
            direction = COLUMN;
        }
        cin >> n;
        n--;
        cin >> color;
        color--;
        operations.push_back({direction, n, color});
    }
    // color list
    vector<long> colors(k, 0);

    map<int, map<int, Line>> lines;
    lines[ROW] = {};
    lines[COLUMN] = {};

    vector<Line> rows;
    vector<Line> columns;
    vector<int> active_rows;
    vector<int> active_columns;



    for (auto && operation : operations)
    {
        
        if (lines[operation[0]].count(operation[1]))
        {
            //cout << "This should be printed exactly once.\n";
            lines[operation[0]].at(operation[1]).paint_line(operation[2]);
        }
        else
        {
            Line new_line;
            new_line.n = operation[1];
            new_line.dir = operation[0];
            new_line.paint_line(operation[2]);
            Line& new_line_ref = new_line;
            lines[operation[0]][operation[1]] = new_line_ref;
            if (operation[0] == ROW)
            {
                rows.push_back(lines[operation[0]][operation[1]]);
                active_rows.push_back(operation[1]);
            }
            else
            {
                columns.push_back(lines[operation[0]][operation[1]]);
                active_columns.push_back(operation[1]);
            }
            lines[operation[0]][operation[1]].test = "Does this work?";
        }
        if (operation[0] == ROW)
        {
            for (auto &&line : active_columns)
            {
                lines[COLUMN][line].add_off_color(operation[1], operation[2]);
            }
        }
        else
        {
            for (auto &&line : active_rows)
            {
                lines[ROW][line].add_off_color(operation[1], operation[2]);
            }
        }

    }
    // calculate number of colors
    for (auto &&active : active_rows)
    {
        //printf("row (%d) off colors: %d \n", lines[ROW][active].n, lines[ROW][active].get_color_list().size());
        colors[lines[ROW][active].main_color] += (m - lines[ROW][active].get_color_list().size());
        //for (auto &&off_color : lines[ROW][active].get_color_list())
        //{
        //    printf("offcolor: %d\n", off_color);
        //    colors[off_color] += 1;
        //}
    }
    for (auto &&active : active_columns)
    {
        colors[lines[COLUMN][active].main_color] += (n - lines[COLUMN][active].get_color_list().size());
        //printf("column (%d) off colors: %d\n", lines[COLUMN][active].n, lines[COLUMN][active].get_color_list().size());
        //for (auto &&off_color : lines[COLUMN][active].get_color_list())
        //{
        //    printf("offcolor: %d\n", off_color);
        //    colors[off_color] += 1;
        //}
    }

    for (auto &&color : colors)
    {
        cout << color << " ";
    }
    
    
    
    

    return 0;
}

Test 1 (public)

Group: 1, 3, 4

Verdict: ACCEPTED

Test 2 (public)

Group: 3, 4

Verdict: WRONG ANSWER

Test 3

Group: 1, 3, 4

Verdict: WRONG ANSWER

Test 4

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 5

Group: 2, 4

Verdict: TIME LIMIT EXCEEDED

Test 6

Group: 3, 4

Verdict: WRONG ANSWER

Test 7

Group: 3, 4

Verdict: WRONG ANSWER

Test 8

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 4

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 1, 3, 4

Verdict: WRONG ANSWER

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