Submission #796790

#	Time	Username	Problem	Language	Result	Execution time	Memory
796790		alexander707070	Stations (IOI20_stations)	C++14	100 / 100	780 ms	872 KiB

stations

#include<bits/stdc++.h>
#define MAXN 1007
using namespace std;

int n,st,et,cnt;
vector<int> sol,v[MAXN],w;
int tim,in[MAXN],out[MAXN],z[2*MAXN];

void dfs(int x,int p,int dep){
    in[x]=tim; tim++;
    for(int i=0;i<v[x].size();i++){
        if(v[x][i]!=p)dfs(v[x][i],x,dep+1);
    }
    out[x]=tim; tim++;

    if(dep%2==0)sol[x]=in[x];
    else sol[x]=out[x];
}

vector<int> label(int N, int k,vector<int> from,vector<int> to){
    n=N;
    sol.resize(n);
    for(int i=0;i<n;i++)v[i].clear();
    tim=0;

    for(int i=0;i<n-1;i++){
        v[from[i]].push_back(to[i]);
        v[to[i]].push_back(from[i]);
    }

    dfs(0,-1,0);
    
    w=sol;
    sort(w.begin(),w.end());
    
    for(int i=0;i<w.size();i++){
        z[w[i]]=i;
    }
    
    for(int i=0;i<sol.size();i++)sol[i]=z[sol[i]];

    return sol;
}

int find_next_station(int s, int t,vector<int> c){
    if(s==0){
        st=s+1;
        for(int i=0;i<c.size();i++){
            if(t>=st and t<=c[i])return c[i];
            st=c[i]+1;
        }
    }else if(s<c[0]){
        st=s+1;
        for(int i=0;i<c.size()-1;i++){
            if(t>=st and t<=c[i])return c[i];
            st=c[i]+1;
        }
        return c[c.size()-1];
    }else{
        et=s-1;
        for(int i=c.size()-1;i>=1;i--){
            if(t>=c[i] and t<=et)return c[i];
        }
        return c[0];
    }
}

Compilation message (stderr)

stations.cpp: In function 'void dfs(int, int, int)':
stations.cpp:11:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   11 |     for(int i=0;i<v[x].size();i++){
      |                 ~^~~~~~~~~~~~
stations.cpp: In function 'std::vector<int> label(int, int, std::vector<int>, std::vector<int>)':
stations.cpp:36:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   36 |     for(int i=0;i<w.size();i++){
      |                 ~^~~~~~~~~
stations.cpp:40:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |     for(int i=0;i<sol.size();i++)sol[i]=z[sol[i]];
      |                 ~^~~~~~~~~~~
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:48:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |         for(int i=0;i<c.size();i++){
      |                     ~^~~~~~~~~
stations.cpp:54:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   54 |         for(int i=0;i<c.size()-1;i++){
      |                     ~^~~~~~~~~~~
stations.cpp:66:1: warning: control reaches end of non-void function [-Wreturn-type]
   66 | }
      | ^

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