Submission #933167

#	Time	Username	Problem	Language	Result	Execution time	Memory
933167		thunopro	Stations (IOI20_stations)	C++14	5 / 100	611 ms	48128 KiB

stations

#include<bits/stdc++.h>
#include "stations.h"
using namespace std ; 
#define maxn 1000009
#define ll long long 
#define pb push_back 
#define fi first 
#define se second 
#define left id<<1
#define right id<<1|1 
#define re exit(0); 
#define _lower(x) lower_bound(v.begin(),v.end(),x)-v.begin()+1
const int mod = 1e9+7 ;
const int INF = 1e9 ; 

typedef vector<int> vi ; 
typedef pair<int,int> pii ; 
typedef vector<pii> vii ; 

template < typename T > void chkmin ( T &a , T b ) { if ( a > b ) a = b ; } 
template < typename T > void chkmax ( T &a , T b ) { if ( a < b ) a = b ; } 

void add ( int &a , int b ) 
{
	a += b ; 
	if ( a >= mod ) a -= mod ; 
	if ( a < 0 ) a += mod ; 
}

void rf () 
{
	freopen ("bai1.inp","r",stdin) ;
}

int _pow ( int a , int n ) 
{
	if ( n == 0 ) return 1 ; 
	int res = _pow (a,n/2) ; 
	if ( n % 2 ) return 1ll*res*res%mod*a%mod ; 
	else return 1ll*res*res%mod ; 
}

//vi label ( int N , int K , vi U , vi V ) ; 
//int find_next_station ( int s , int t , int c ) ; 
//
//int main () 
//{
//	ios_base::sync_with_stdio(0); 
//	cin.tie(0);cout.tie(0); 
//	rf () ; 
//	int n , k ; 
//	vi u , v ; 
//	cin >> n >> k ; 
//	for ( int i = 0 ; i < n-1 ; i ++ ) 
//	{
//		int _u , _v ; cin >> _u >> _v ; 
//		u . pb (_u) , v . pb (_v) ; 
//	}
//	vi res = label (n,k,u,v) ; 
//}

vi adjList [maxn] ; 
int n , k ; 

vi label ( int N , int K , vi U , vi V ) 
{
	for ( int i = 0 ; i < n ; i ++ ) adjList [i] . clear () ; 
	n = N , k = K ; 
	for ( int i = 0 ; i < n-1 ; i ++ ) 
	{
		adjList [U[i]] . pb (V[i]) ; 
		adjList [V[i]] . pb (U[i]) ; 
	}
	int root = -1 ; 
	for ( int i = 0 ; i < n ; i ++ ) if ( adjList [i].size () == 1 ) root = i ; 
	vi res(n,-1) ; 
	int num = 0 ; 
	stack <int> q ; 
	q . push (root) ; 
	while ( q.size () ) 
	{
		int u = q.top() ; q.pop() ; res [u] = num ++ ;
		for ( auto v : adjList [u] ) if ( res [v] == -1 ) q . push (v) ; 
	}
	return res ; 
}

int find_next_station ( int s , int t , vi c ) 
{
	int res = -1 ; 
	for ( auto x : c ) 
	{
		if ( s < t ) 
		{
			if ( s <= x && x <= t ) res = max (res,x) ; 
		}
		else if ( s > t ) 
		{
			if ( s >= x && x >= t ) return x ; 
		}
	}
	return res ; 
}

Compilation message (stderr)

stations.cpp: In function 'void rf()':
stations.cpp:32:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   32 |  freopen ("bai1.inp","r",stdin) ;
      |  ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~

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