제출 #393080

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
393080		REALITYNB	기지국 (IOI20_stations)	C++14	8 / 100	1144 ms	656 KiB

stations

#include <bits/stdc++.h> 
#include "stations.h"
#define pii pair<int,int> 
#define in first 
#define out second
#define mp make_pair 
using namespace std; 

vector<int> label(int n ,int k , vector<int> u , vector<int> v){
	vector<int> ans(n) ; 
	vector<int> adj[n] ; 
	for(int i=0;i<n-1;i++) {
		adj[u[i]].push_back(v[i]) ; 
		adj[v[i]].push_back(u[i]) ; 
	}
	int tim =0; 
	function<void(int,int,int)> dfs = [&](int a, int p,int num ){
		ans[a]=num ; 
		bool flg = 0 ; 
		for(int&x:adj[a]){
			if(x!=p) {
				dfs(x,a,num*2+flg) ; 
				flg=1;
			}
		}
	}; 
	dfs(0,0,1); 
	return ans ; 
}
vector<int> get_root_path(int x){
	vector<int> a ; 
	while(x){
		a.push_back(x) ; 
		x/=2 ;
	}
	reverse(a.begin(),a.end()) ; 
	return a ; 
}
vector<int> get_path(int s , int t){
	vector<int> a = get_root_path(s) , b = get_root_path(t) ; 
	//for(int x : a) cout << x << " "; 
	int lca = 0 ;

	for(int i=0;i<min(a.size(),b.size());i++){
		if(b[i]==a[i]){
			lca = i ; 
			continue ; 
		} 
	}
	vector<int> ans ; 
	for(int i=a.size()-1;i>lca;i--){
		ans.push_back(a[i]) ; 
	}
	ans.push_back(a[lca]) ; 
	/*cout << endl ; 
	for(int x : b) cout <<  x<< " " ; 
		cout << lca << " "; */
	for(int i=lca+1;i<b.size();i++) ans.push_back(b[i]) ; 
	//cout << endl ; 
	return ans ; 
}
int find_next_station(int s ,int t, vector<int> ne){
	vector<int>  a= get_path(s,t) ; 
	return a[1] ; 
}
/*int main(){
	cout << "4 7 :" ;
	vector<int> a = get_path(4,7) ; 
	for(int x : a) cout << x << " "; 
	cout << endl ; 
	return 0 ; 
}*/

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

stations.cpp: In function 'std::vector<int> label(int, int, std::vector<int>, std::vector<int>)':
stations.cpp:16:6: warning: unused variable 'tim' [-Wunused-variable]
   16 |  int tim =0;
      |      ^~~
stations.cpp: In function 'std::vector<int> get_path(int, int)':
stations.cpp:44:15: warning: comparison of integer expressions of different signedness: 'int' and 'const long unsigned int' [-Wsign-compare]
   44 |  for(int i=0;i<min(a.size(),b.size());i++){
      |              ~^~~~~~~~~~~~~~~~~~~~~~~
stations.cpp:58:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   58 |  for(int i=lca+1;i<b.size();i++) ans.push_back(b[i]) ;
      |                  ~^~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 8 / 8
• Subtask 3: 0 / 16
• Subtask 4: 0 / 10
• Subtask 5: 0 / 61