Submission #447546

#	Time	Username	Problem	Language	Result	Execution time	Memory
447546		dxz05	Split the Attractions (IOI19_split)	C++14	18 / 100	122 ms	19728 KiB

split

#include "split.h"
#include <bits/stdc++.h>

using namespace std;

const int MAXN = 3e5 + 3e2;

vector<int> perm = {0, 1, 2};

void sort(int &a, int &b, int &c){
    vector<int> v = {a, b, c};
    do {
        a = v[perm[0]], b = v[perm[1]], c = v[perm[2]];
        if (a <= b && b <= c) break;
    } while (next_permutation(perm.begin(), perm.end()));
}

vector<int> orig(vector<int> v){
    vector<int> res = v;
    for (int i = 0; i < v.size(); i++){
        res[i] = perm[v[i] - 1] + 1;
    }
    return res;
}

vector<int> g[MAXN];

bool used[MAXN];
vector<int> ord;
void dfs(int v){
    used[v] = true;
    ord.push_back(v);

    for (int u : g[v]){
        if (!used[u]) dfs(u);
    }
}

vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
	vector<int> res(n, 0);

	sort(a, b, c);

    int m = p.size();

    for (int i = 0; i < m; i++){
        g[p[i]].push_back(q[i]);
        g[q[i]].push_back(p[i]);
    }

    int ind = 0;
    for (int i = 0; i < n; i++){
        if (g[i].size() < g[ind].size()) ind = i;
    }

    dfs(ind);

    for (int i = 0; i < n; i++){
        if (i < a) res[ord[i]] = 1; else
        if (i < a + b) res[ord[i]] = 2; else
            res[ord[i]] = 3;
    }

	return orig(res);
}

/*
7 6
1 2 4
0 1
1 2
2 3
3 4
4 5
5 6
 */

Compilation message (stderr)

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

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