Submission #1040039

#	Time	Username	Problem	Language	Result	Execution time	Memory
1040039		LaMatematica14	Split the Attractions (IOI19_split)	C++17	7 / 100	32 ms	9820 KiB

split

#include <bits/stdc++.h>
using namespace std;
vector<int> sz, pr;
vector<vector<int>> adj;

vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
    array<pair<int, int>, 3> t;
    t[0] = {a, 1}; t[1] = {b, 2}; t[2] = {c, 3};
    sort(t.begin(), t.end());
    sz.resize(n); pr.resize(n); adj.resize(n);

    vector<int> g(n, 0);
    for (int i = 0; i < p.size(); i++) {
        adj[p[i]].push_back(q[i]); adj[q[i]].push_back(p[i]);
    }
    int beg = 0;
    for (int i = 0; i < n; i++) {
        if (adj[i].size() == 1) beg = i;
    }
    g[beg] = t[0].second; t[0].first--;
    for (int i = 0; i < n; i++) {
        for (int x : adj[beg]) {
            if (g[x] > 0) continue;
            if (t[0].first > 0) {
                g[x] = t[0].second;
                t[0].first--;
            }
            else if (t[1].first > 0) {
                g[x] = t[1].second;
                t[1].first--;
            }
            else {
                g[x] = t[2].second;
                t[2].first--;
            }
            beg = x;
            break;
        }
    }
    return g;
}

Compilation message (stderr)

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:13:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   13 |     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