Submission #647476

#	Time	Username	Problem	Language	Result	Execution time	Memory
647476		a_aguilo	Simurgh (IOI17_simurgh)	C++14	13 / 100	3088 ms	376 KiB

simurgh

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

using namespace std;

vector<int> padre;
int N;
vector<int> choices;

int raiz(int nodo){
    if(padre[nodo] == nodo) return nodo;
    return raiz(padre[nodo]);
}

bool getTree(int pos, vector<pair<int, int>>& edges){
    //cout << pos << endl;
    if((choices.size()+edges.size()-pos) < (N-1)) return false;
    if(pos == edges.size()){
        if(choices.size() == N-1){
            //cout << "submit" << endl;
            //print(choices);
            if(count_common_roads(choices) == N-1) {
                    //cout << "yes "<< endl;
                    return true;
            }
            return false;
        }
        else return false;
    }
    pair<int, int> actEdge = edges[pos];
    int a = actEdge.first; int b = actEdge.second;
    int rootA = raiz(a); int rootB = raiz(b);
    if (getTree(pos+1, edges)) return true;
    else if (rootA!=rootB) {
        padre[rootB] = rootA;
        choices.push_back(pos);
        bool ret = getTree(pos+1, edges);
        padre[rootB] = rootB;
        if(ret) return true;
        choices.pop_back();
    }
    return false;
}

vector<int> find_roads(int n, vector<int> u, vector<int> v) {
	N = n;
	vector<pair<int, int>> edges(u.size());
	choices = vector<int>();
	for(int i = 0; i < u.size(); ++i){
        edges[i] = {u[i], v[i]};
	}
	padre = vector<int>(n);
	for(int i = 0; i < n; ++i) padre[i] = i;
	getTree(0, edges);
	return choices;
}

Compilation message (stderr)

simurgh.cpp: In function 'bool getTree(int, std::vector<std::pair<int, int> >&)':
simurgh.cpp:17:42: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   17 |     if((choices.size()+edges.size()-pos) < (N-1)) return false;
      |        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~
simurgh.cpp:18:12: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   18 |     if(pos == edges.size()){
      |        ~~~~^~~~~~~~~~~~~~~
simurgh.cpp:19:27: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   19 |         if(choices.size() == N-1){
      |            ~~~~~~~~~~~~~~~^~~~~~
simurgh.cpp: In function 'std::vector<int> find_roads(int, std::vector<int>, std::vector<int>)':
simurgh.cpp:49:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |  for(int i = 0; i < u.size(); ++i){
      |                 ~~^~~~~~~~~~

• Subtask 1: 13 / 13
• Subtask 2: 0 / 17
• Subtask 3: 0 / 21
• Subtask 4: 0 / 19
• Subtask 5: 0 / 30