제출 #796938

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
796938		boyliguanhan	기지국 (IOI20_stations)	C++17	100 / 100	841 ms	784 KiB

stations

#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
int cnt = 0;
vector<int> adj[1010];
vector<int> labels;
void dfs(int n, int p, int dep=1) {
    if(dep%2)
        labels[n] = cnt++;
    for(auto j: adj[n])
        if(j!=p)
            dfs(j,n,dep+1);
    if(!(dep%2))
        labels[n] = cnt++;
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
    cnt = k = 0;
    for(int i = 0; i < n; i++)
        adj[i].clear();
    labels.resize(n);
	for(int i = 0; i < n-1; i++)
        adj[u[i]].push_back(v[i]), adj[v[i]].push_back(u[i]);
    dfs(0,-1);
	return labels;
}
bool inrange(int l, int r, int x) {
    if(l>r) swap(l,r);
    return l<=x&&x<=r;
}
int find_next_station(int s, int t, vector<int> c) {
	c.push_back(s);
     sort(c.begin(),c.end());
    if(c[0]==s)
        reverse(c.begin(),c.end());
    int p = c[0];
    int start = 1;
    if(s==0) start = 0;
    for(int i = start; i < c.size()-1; i++)
        if(inrange(c[i],c[i+1]+(c[i]<c[i+1]?-1:1),t))
            return c[i];
    return p;
}

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

stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:38:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   38 |     for(int i = start; i < c.size()-1; i++)
      |                        ~~^~~~~~~~~~~~

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