Submission #123037

#	Time	Username	Problem	Language	Result	Execution time	Memory
123037		SirCeness	Simurgh (IOI17_simurgh)	C++14	0 / 100	435 ms	11664 KiB

simurgh

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

#define pb push_back
#define mp make_pair
#define inside sl<=l&&r<=sr
#define outside r<sl||sr<l
using namespace std;
typedef long long ll;

int n, m;
bitset<500> vis;
vector<int> tree;
list<int> adj[505];
map<pair<int, int>, int> mpp;

int get(int node1, int node2){
	//cout << "get(" << node1 << ", " << node2 << ")" << endl;
	if (mpp.count(mp(node2, node1)) == 0){
		assert(mpp.count(mp(node1, node2)) > 0);
		return mpp[mp(node1, node2)];
	} else return mpp[mp(node2, node1)];
}

void findtree(int node, int ata){
	//cout << "findtree(" << node << ", " << ata << ")" <<endl;
	if (vis[node] == 1) return;
	vis[node] = 1;
	if (ata != -1){
		tree.pb(get(node, ata));
	}
	for (list<int>::iterator it = adj[node].begin(); it != adj[node].end(); ++it){
		findtree(*it, node);
	}
}
vector<int> find_roads(int N, vector<int> u, vector<int> v) {
	n = N;
	set<int> ans;
	for (int i = 0; i < u.size(); i++){
		adj[u[i]].pb(v[i]);
		adj[v[i]].pb(u[i]);
		mpp[mp(u[i], v[i])] = i;
	}
	for (int i = 0; i < n-1; i++){
		vis.reset();
		vis[i] = 1;
		tree.clear();
		findtree(i+1, -1);
		assert(tree.size() == n-2);
		//cout << "tree bitti" << endl;
		vector<pair<int, int> > vals;
		int maxx = 0;
		tree.pb(0);
		for (list<int>::iterator it = adj[i].begin(); it != adj[i].end(); ++it){
			tree[n-2] = get(i, *it);
			int val = count_common_roads(tree);
			maxx = max(maxx, val);
			vals.pb(mp(val, *it));
		}
		assert(maxx > 0);
		
		for (int j = 0; j < vals.size(); j++){
			if (vals[j].first == maxx) ans.insert(get(i, vals[j].second));
		}
	}
	vector<int> ansv;
	for (set<int>::iterator it = ans.begin(); it != ans.end(); ++it){
		ansv.pb(*it);
	}
	assert(ansv.size() == n-1);
	return ansv;
}

Compilation message (stderr)

simurgh.cpp: In function 'std::vector<int> find_roads(int, std::vector<int>, std::vector<int>)':
simurgh.cpp:39:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < u.size(); i++){
                  ~~^~~~~~~~~~
In file included from /usr/include/c++/7/cassert:44:0,
                 from /usr/include/x86_64-linux-gnu/c++/7/bits/stdc++.h:33,
                 from simurgh.cpp:1:
simurgh.cpp:49:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   assert(tree.size() == n-2);
          ~~~~~~~~~~~~^~~~~~
simurgh.cpp:62:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for (int j = 0; j < vals.size(); j++){
                   ~~^~~~~~~~~~~~~
In file included from /usr/include/c++/7/cassert:44:0,
                 from /usr/include/x86_64-linux-gnu/c++/7/bits/stdc++.h:33,
                 from simurgh.cpp:1:
simurgh.cpp:70:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  assert(ansv.size() == n-1);
         ~~~~~~~~~~~~^~~~~~

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