제출 #312402

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
312402		jainbot27	기지국 (IOI20_stations)	C++17	8 / 100	977 ms	1040 KiB

stations

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

#define f first
#define s second
#define pb push_back
#define ar array
#define all(x) x.begin(), x.end()
#define siz(x) (int)x.size()

#define FOR(x, y, z) for(int x = (y); x < (z); x++)
#define ROF(x, z, y) for(int x = (y-1); x >= (z); x--)
#define F0R(x, z) FOR(x, 0, z)
#define R0F(x, z) ROF(x, 0, z)
#define trav(x, y) for(auto&x:y)

using ll = long long;
using vi = vector<int>;
using vl = vector<long long>;
using pii = pair<int, int>;
using vpii = vector<pair<int, int>>;

template<class T> inline bool ckmin(T&a, T b) {return b < a ? a = b, 1 : 0;}
template<class T> inline bool ckmax(T&a, T b) {return b > a ? a = b, 1 : 0;}
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

const char nl = '\n';
const int mxN = 1001;
const int MOD = 1e9 + 7;
const long long infLL = 1e18;

vi sol, adj[mxN]; bool vis[mxN]; int num = 0;

void dfs(int u, bool b){
    vis[u] = 1;
    if(!b) sol[u] = num++;
    trav(v, adj[u]) if(!vis[v]) dfs(v, b^1);
    if(b) sol[u] = num++;
}
vi label(int n, int k, vi x, vi y){
    F0R(i, n) vis[i] = 0, num= 0, sol.clear();
    sol.resize(n);
    F0R(i, n-1){
        adj[x[i]].pb(y[i]);
        adj[y[i]].pb(x[i]);
    }
    dfs(0, 0);
    return sol;
}
int find_next_station(int s, int t, vi sz){
    int x = sz[0] , y = sz.back(); 
    if(s < x ) {
        if(t < s || t > y ) return y;
        F0R(i, siz(sz)) if(sz[i] >= t) return sz[i];
    }
    if(t > s || t < x ) return x;
    R0F(i, siz(sz)) if(sz[i] <= t) return sz[i];
}

// int main(){
//     int n, k; cin >> n >> k;
//     vi U, V;
//     F0R(i, n-1){
//         int uu, vv; cin >> uu;
//         U.pb(uu);
//     }
//     F0R(i, n-1){
//         int uu, vv; cin >> vv;
//         V.pb(vv);
//     }

//     vi ANS = label(n, k, U, V);
//     F0R(i, n){
//         cout << ANS[i] << "\n";
//     }
// }

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

stations.cpp: In function 'int find_next_station(int, int, vi)':
stations.cpp:59:1: warning: control reaches end of non-void function [-Wreturn-type]
   59 | }
      | ^

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