Submission #521025

#	Time	Username	Problem	Language	Result	Execution time	Memory
521025		zsombor	Split the Attractions (IOI19_split)	C++17	18 / 100	77 ms	10692 KiB

split

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

using namespace std;

int bcount;
vector <vector <int>> g(1e5);
vector <bool> bo(1e5,true);
vector<int> res;

void dfs(int x){
    if (bcount==-1 || !bo[x]) return;
    if (bcount==0){
        res[x]=1;
        bcount--;
        return;
    }
    res[x]=2;
    bcount--;
    bo[x]=false;
    for (int i : g[x]) dfs(i);
}


vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
	res.resize(n,3);
	int cur=0;
	for (int i=0;i<p.size();i++){
        g[p[i]].push_back(q[i]);
        g[q[i]].push_back(p[i]);
    }
	if (a == 1) {
		bcount=b;
		dfs(0);
	} else {
        for (int i=0;i<n;i++) if (g[i].size()==1) cur = i;
        for (int i = 0;i<a;i++){
            res[cur]=1;
            bo[cur]=false;
            if (bo[g[cur][0]]){
                cur=g[cur][0];
            }else{
                cur=g[cur][1];
            }
        }
        for (int i = 0;i<b;i++){
            res[cur]=2;
            bo[cur]=false;
            if (bo[g[cur][0]]){
                cur=g[cur][0];
            }else{
                cur=g[cur][1];
            }
        }
	}
	return res;
}

Compilation message (stderr)

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

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