Submission #1080469

#	Time	Username	Problem	Language	Result	Execution time	Memory
1080469		speedcode	Split the Attractions (IOI19_split)	C++17	7 / 100	49 ms	10680 KiB

split

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

std::vector<int> find_split(int n, int a, int b, int c, std::vector<int> p, std::vector<int> q){
    vector<vector<int>> graphe(n);
    for(int i = 0; i < p.size(); i++){
        graphe[p[i]].push_back(q[i]);
        graphe[q[i]].push_back(p[i]);
    }
    vector<int> result(n);
    fill(result.begin(), result.end(), 3);
    int begin  = 0;
    for(int i = 0; i < n; i++) {
        if(graphe[i].size() == 1) begin = i;
    }
    while(a--) {
        result[begin] = 1;
        for(auto k : graphe[begin]) {
            if(result[k] == 3) {
                begin = k;
            }
        }
    }
    while(b--) {
        result[begin] = 2;
        for(auto k : graphe[begin]) {
            if(result[k] == 3) {
                begin = k;
            }
        }
    }

    return result;
}

Compilation message (stderr)

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:7:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    7 |     for(int i = 0; i < p.size(); i++){
      |                    ~~^~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 0 / 11
• Subtask 3: 0 / 22
• Subtask 4: 0 / 24
• Subtask 5: 0 / 36