제출 #349943

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
349943		evn	기지국 (IOI20_stations)	C++14	0 / 100	3051 ms	2097156 KiB

stations

#include <bits/stdc++.h>
#include "stations.h"
using namespace std;
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define sz(a) a.size()
typedef long long ll;
typedef pair<int, int> pii;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
template<class T> using oset=tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
int timer = 0;
int depth[1005];
int in[1005];
int out[1005];
vector<int> adj[1005];
void dfs(int u, int p){
	in[u] = timer++;
	for(int v : adj[u]){
		if(v != p){
			depth[v] = depth[u]+1;
			dfs(v, u);
		}
	}
	out[u] = timer++;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
	vector<int> labels(n);
	for(int i = 0; i < u.size(); i++){
		int a, b;
		a = u[i];
		b = v[i];
		adj[a].pb(b);
		adj[b].pb(a);
	}
	dfs(0, -1);
	for (int i = 0; i < n; i++) {
		if(depth[i] % 2 == 0)labels[i] = in[i];
		else{labels[i] = out[i];}
	}
	return labels;
}

int find_next_station(int s, int t, vector<int> c) {
	int lessThan = 0;
	for(int u : c){
		if(u < s)lessThan++;
	}
	if(lessThan == c.size()){
		//out
		if(c.size() == 1){
			return c[0];
		}
		int in = c[1];
		in--;
		if(t >= in && t <= s){
			//node is in this subtree
			for(int i =1; i < c.size(); i++){
				int out = (i == c.size()-1 ? s-1:c[i+1]-1);
				if(t >= c[i] && t <= out){
					return c[i];
				}
			}
		}
		else{
			//go to parent
			return c[0];
		}
	}
	else{
		//this is the in value
		if(c.size() == 1){
			return c[0];
		}
		int out = c[c.size() - 1]+1;
		if(t >= s && t <= out){
			//node is in this subtree
			if(s == 0){
				for(int i = 0; i < c.size(); i++){
					int in = (i == 0 ? s+1:c[i-1]+1);
					if(t >= in && t <= c[i]){
						return c[i];
					}
				}
			}
			else{
				for(int i = 1; i < c.size(); i++){
					int in = (i == 1 ? s+1 : c[i-1] + 1);
					if(t >= in && t <= c[i])return c[i];
				}	
			}
		}
		else{
			//go to parent
			return c[0];
		}
	}
	return c[0];
}

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

stations.cpp: In function 'std::vector<int> label(int, int, std::vector<int>, std::vector<int>)':
stations.cpp:32:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |  for(int i = 0; i < u.size(); i++){
      |                 ~~^~~~~~~~~~
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:52:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |  if(lessThan == c.size()){
      |     ~~~~~~~~~^~~~~~~~~~~
stations.cpp:61:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   61 |    for(int i =1; i < c.size(); i++){
      |                  ~~^~~~~~~~~~
stations.cpp:62:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |     int out = (i == c.size()-1 ? s-1:c[i+1]-1);
      |                ~~^~~~~~~~~~~~~
stations.cpp:82:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   82 |     for(int i = 0; i < c.size(); i++){
      |                    ~~^~~~~~~~~~
stations.cpp:90:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   90 |     for(int i = 1; i < c.size(); i++){
      |                    ~~^~~~~~~~~~

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