제출 #182385

#제출 시각아이디문제언어결과실행 시간메모리
182385CaroLindaRoller Coaster Railroad (IOI16_railroad)C++14
34 / 100
2094 ms492928 KiB
#include "railroad.h"
#include <bits/stdc++.h>

#define debug printf
#define lp(i,a,b) for(int i = a ; i < b ; i++ )
#define ff first
#define ss second
#define pb push_back
#define mk make_pair
#define ll long long
#define sz size()
#define pii pair<int,int>
#define all(x) x.begin(),x.end()
#define tiii tuple<int,int,int>
#define mkt make_tuple

const int MAXN = 2e5+10 ;
const ll inf = 1e18+10 ;

using namespace std ;

int n , Key , comp ;
int bit[MAXN*2] , realVal[MAXN*2] , myComp[MAXN*2] , pai[MAXN*2] ;
map<int,int> mp ;
vector<int> adj[MAXN*2] ;
ll ans ;
bool vis[MAXN*2] ;
vector< tuple<int,int,int> > edges ;

void upd(int pos, int val) { for(int i = pos ; i < MAXN*2 ; i += (i&-i) ) bit[i] += val ; }
int qry(int pos)
{
	int tot = 0 ;
	for(int i = pos ; i > 0 ; i -= (i&-i) )
		tot += bit[i] ;
	return tot;
}

void dfs(int cur)
{

	myComp[cur] = comp ;
	vis[cur] = true ;

	for(int viz : adj[cur])
		if(!vis[viz])
			dfs(viz) ;

}

int find(int i)
{
	if( pai[i] == i ) return i ;
	return pai[i] = find(pai[i]);
}

void join(int a, int b,int cost)
{
	a = find(a) ;
	b = find(b) ;
	if(a == b) return ;
	ans += 1LL * cost ;
	if( rand()%2 ) pai[a] = b ;
	else pai[b] = a ;
}

long long plan_roller_coaster(std::vector<int> s, std::vector<int> t) 
{
    n = (int)(s.sz);

    for(int i : s ) mp[i] = 0 ;
    for(int i : t ) mp[i] = 0 ;

    for(auto &p : mp ) 
    {
    	p.ss = ++Key ;
    	realVal[Key] = p.ff ;
    }

	int val1,val2;
	lp(i,0,n)
	{
		s[i] = mp[ s[i] ] ;
		t[i] = mp[ t[i] ] ;

		if( s[i] == t[i] ) continue ;

		adj[ s[i] ].pb( t[i] ) ;

		if( s[i] > t[i] ) 
		{
			upd( t[i] , -1 ) ;
			upd( s[i], 1 ) ;
			continue ;
		}

		upd( s[i] , 1 ) ;
		upd( t[i], -1 ) ;

	}
	upd(1,-1) ;
	upd( Key, 1 ) ;
	adj[Key].pb(1) ;

	lp(i,1,Key)
	{

		int val = qry(i) ;

		if( val > 0 )
		{
			ans += 1LL*val*(realVal[i+1]-realVal[i]) ;
			for(int j = 0 ; j < val ; j++ ) adj[i+1].pb(i) ;
		}
		else if( val < 0 )
			for(int j = 0 ; j < abs(val) ; j++ ) adj[i].pb(i+1) ;

	}

	lp(i,1,Key+1)
		if( !vis[i] )
		{
			comp ++ ;
			dfs(i) ;
		}

	lp(i,1,Key)
		if( myComp[i] != myComp[i+1] )
			edges.pb( mkt( realVal[i+1] - realVal[i] , myComp[i] , myComp[i+1] ) ) ;

	sort(all(edges)) ;
	lp(i,1,comp+1) pai[i] = i ;

	for(auto p : edges)
		join( get<1>(p) , get<2>(p) , get<0>(p) ) ;

    return ans ;
}

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

railroad.cpp: In function 'long long int plan_roller_coaster(std::vector<int>, std::vector<int>)':
railroad.cpp:80:6: warning: unused variable 'val1' [-Wunused-variable]
  int val1,val2;
      ^~~~
railroad.cpp:80:11: warning: unused variable 'val2' [-Wunused-variable]
  int val1,val2;
           ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...