Submission #33828

#	Time	Username	Problem	Language	Result	Execution time	Memory
33828		mohammad_kilani	Simurgh (IOI17_simurgh)	C++14	0 / 100	0 ms	2544 KiB

simurgh

#include "simurgh.h"
#include <bits/stdc++.h>
using namespace std;
const int N = 501;
int  vis[N] , vi = 0;
vector< pair<int,int> > g[N], T[N];
vector< int > tree , anss ;
bool ok[N * N] = {0};
bool asked[N * N] = {0} ;


void DFS(int node){
	vis[node] = vi;
	for(int i=0;i<g[node].size();i++){
		if(vis[g[node][i].first] != vi){
			tree.push_back(g[node][i].second);
			T[node].push_back(g[node][i]);
			T[g[node][i].first].push_back(make_pair(node,g[node][i].second));
			DFS(g[node][i].first);
		}
	}
}

bool DFS2(int node,int target){
	if(node == target) return true;
	vis[node] = vi;
	for(int i=0;i<T[node].size();i++){
		int newnode = T[node][i].first;
		if(vis[newnode] != vi && DFS2(newnode,target)) return true;
	}
	return false;
}

std::vector<int> find_roads(int n, std::vector<int> u, std::vector<int> v) {
	for(int i=0;i<u.size();i++){
		g[u[i]].push_back(make_pair(v[i],i));
		g[v[i]].push_back(make_pair(u[i],i));
	}
	vi++;
	DFS(0);
	assert(tree.size() == n-1);
	sort(tree.begin(),tree.end());
	int res = count_common_roads(tree);
	for(int i=0;i<n;i++){
		vector< pair<int,int> > v;
		int mx = res;
		for(int j=0;j<g[i].size();j++){
			if(asked[g[i][j].second]) continue;
			//asked[g[i][j].second] = true;
			int newnode = g[i][j].first;
			int edge = -1;
			for(int k=0;k<T[i].size();k++){
				vi++;
				if(DFS2(T[i][k].first,newnode)){
					edge = T[i][k].second;
					break;
				}
			}
			vector<int> r;
			for(int i=0;i<tree.size();i++){
				if(tree[i] == edge) continue;
				r.push_back(tree[i]);
			}
			r.push_back(g[i][j].second);
			sort(r.begin(),r.end());
			int cur = count_common_roads(r);
			mx = max(mx,cur);
			v.push_back(make_pair(g[i][j].second,cur));
		}
		for(int i=0;i<v.size();i++){
			if(v[i].second == mx){
				ok[v[i].first] = true;
			}
		}
		
	}
	for(int i=0;i<u.size();i++) if(ok[i]) anss.push_back(i);
	return anss;
}

Compilation message (stderr)

simurgh.cpp: In function 'void DFS(int)':
simurgh.cpp:14:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<g[node].size();i++){
               ^
simurgh.cpp: In function 'bool DFS2(int, int)':
simurgh.cpp:27:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<T[node].size();i++){
               ^
simurgh.cpp: In function 'std::vector<int> find_roads(int, std::vector<int>, std::vector<int>)':
simurgh.cpp:35:15: 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++/5/cassert:43:0,
                 from /usr/include/x86_64-linux-gnu/c++/5/bits/stdc++.h:33,
                 from simurgh.cpp:2:
simurgh.cpp:41:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  assert(tree.size() == n-1);
                     ^
simurgh.cpp:47:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int j=0;j<g[i].size();j++){
                ^
simurgh.cpp:52:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(int k=0;k<T[i].size();k++){
                 ^
simurgh.cpp:60:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(int i=0;i<tree.size();i++){
                 ^
simurgh.cpp:70:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int i=0;i<v.size();i++){
                ^
simurgh.cpp:77:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<u.size();i++) if(ok[i]) anss.push_back(i);
               ^

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