Submission #1195926

#TimeUsernameProblemLanguageResultExecution timeMemory
1195926simona1230기지국 (IOI20_stations)C++20
0 / 100
313 ms5264 KiB
#include "stations.h"
#include <bits/stdc++.h>
using namespace std;

vector<int> g[200001];
int in[200001],out[200001];
int d[200001];
vector<int> l;
int num;

void dfs(int i,int p)
{
    in[i]=num++;
    for(int j=0;j<g[i].size();j++)
    {
        int nb=g[i][j];
        if(nb==p)continue;
        d[nb]=d[i]+1;
        dfs(nb,i);
    }
    out[i]=num++;
    if(d[i]%2==0)l[i]=in[i];
    else l[i]=out[i];
}

std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v)
{
    num=0;
    l.clear();
    for(int i=0;i<n;i++)
        l.push_back(0);
    for(int i=0; i<n; i++)
        g[i].clear();
    for(int i=0; i<n-1; i++)
        g[u[i]].push_back(v[i]),g[v[i]].push_back(u[i]);

    dfs(0,-1);
    return l;
}

int find_next_station(int s, int t, std::vector<int> c)
{
    if(s==0)
    {
        for(int i=0;i<c.size();i++)
        {
            if(t<=c[i])return c[i];
        }
    }

    if(s<c[0])
    {
        for(int i=0;i<c.size()-1;i++)
        {
            if(t<=c[i])return c[i];
        }
        return c[c.size()-1];
    }

    for(int i=c.size()-1;i>0;i--)
    {
        if(c[i]<=t)return c[i];
    }
    return c[0];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...