제출 #410040

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
410040		dreezy	Simurgh (IOI17_simurgh)	C++17	0 / 100	15 ms	292 KiB

simurgh

#include <bits/stdc++.h>
#include "simurgh.h"
using namespace std;

struct UnionFind{
	vector<int> height, parent;
	
	void init(int n){
		height.assign(n, 0);
		parent.assign(n,0);
		for(int i =0; i<n;i++)
			parent[i] = i;
	}
	
	
	int getRoot(int n){
		if(parent[n] != n)
			return parent[n] = getRoot(parent[n]);
		return parent[n];
	}
	
	bool isSame(int n1, int n2){
		return getRoot(n1) == getRoot(n2);
	}
	
	void join(int n1, int n2){
		int r1 = getRoot(n1);
		int r2 = getRoot(n2);
		
		if(height[r1] > height[r2]){
			parent[r2] = r1;
			height[r1] = max(height[r1], height[r2]+1);
		}
		else{
			parent[r1] = r2;
			height[r2] = max(height[r2], height[r1]+1);	
		}
	}
	
};

vector<int> find_roads(int n, vector<int> u, vector<int> v ){
	vector<int> is_royal;
	int m = u.size();
	set<int> dontknow;
	for(int i =0; i< m; i++)
		dontknow.insert(i);
	
	/////////
	
	while(true){
		int curcheck = -1;
		int prevans = -1;
		int prevcheck = -1;
		
		UnionFind uf;
		uf.init(n);
		vector<int> roads;
		for(int i : is_royal){
				uf.join(u[i], v[i]);
				roads.push_back(i);
		}
		
		for(int i : dontknow){
			
			if(!uf.isSame(u[i], v[i]) && prevcheck != i){
				uf.join(u[i], v[i]);
				roads.push_back(i);
				curcheck = i;
				
				if(roads.size() == n-1)
					break;
			}
		}
		
		int ans = count_common_roads(roads);
		if(ans == n-1)
			return roads;
		if(prevans == -1){
			prevans = ans;
			continue;
		}
		
		if(ans == prevans+1){
			dontknow.erase(prevcheck);
			is_royal.push_back(prevcheck);
		}
		else if(ans==prevans -1){
			dontknow.erase(prevcheck);
		}
		prevcheck = curcheck;
	}
	return vector<int>();
	
	
}

컴파일 시 표준 에러 (stderr) 메시지

simurgh.cpp: In function 'std::vector<int> find_roads(int, std::vector<int>, std::vector<int>)':
simurgh.cpp:71:21: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   71 |     if(roads.size() == n-1)
      |        ~~~~~~~~~~~~~^~~~~~

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