Submission #133830

# Submission time Handle Problem Language Result Execution time Memory
133830 2019-07-21T13:41:25 Z MetB Roller Coaster Railroad (IOI16_railroad) C++14
Compilation error
0 ms 0 KB
#include <algorithm>
#include <iostream>
#include <string.h>
#include <cstdlib>
#include <vector>
#include <string>
#include <bitset>
#include <math.h>
#include <queue>
#include <stack>
#include <set>
#include <map>
 
typedef long long ll;
typedef long double ld;
 
const ll MOD = 1e9 + 7, INF = 1e18 + 1;
 
using namespace std;
 
int p[1000000], sz[1000000];
 
set <int> mp;
 
struct stop
{
	int x;
	bool op, left;
 
	bool operator < (const stop& b)
	{
		if (x == b.x) return op > b.op;
		else return x < b.x;
	}
};
 
struct edge
{
	int u, v;
	ll c;
 
	bool operator < (const edge& b)
	{
		return c < b.c;
	}
};
 
vector <stop> v;
 
int find (int x)
{
	if (p[x] == x) return x;
	else return p[x] = find (p[x]);
}
 
void unite (int a, int b)
{
	a = find (a);
	b = find (b);
 
	if (a == b) return;
 
	if (sz[a] < sz[b]) swap (a, b);
 
	p[b] = a;
 
	sz[a] += sz[b];
}
 
int ind (int a, vector <ll>& s)
{
	return lower_bound (s.begin(), s.end(), a) - s.begin ();
}
 
ll plan_roller_coaster (vector <int> s, vector <int> t)
{
	ll ans = 0, balance = 0;
 
	vector <int> coord;
 
	for (int i = 0; i < s.size (); i++)
	{
		if (s[i] > t[i])
		{
			v.push_back ({t[i], true, true});
			v.push_back ({s[i], false, true});
		}
		else
		{
			v.push_back ({s[i], true, false});
			v.push_back ({t[i], false, false});
		}
 
		mp.insert (s[i]);
		mp.insert (t[i]);
	}
 
	for (auto a : mp)
		coord.push_back (a);
 
	for (int i = 0; i < coord.size (); i++)
	{
		sz[i] = 1;
		p[i] = i;
	}
 
	v.push_back ({1, true, true});
	v.push_back ({coord.back (), false, true});
 
	for (int i = 0; i < s.size (); i++)
		unite (ind (s[i], coord), ind (t[i], coord));
 
	sort (v.begin(), v.end());
 
	vector <edge> tr;
 
	for (int i = 0; i < v.size (); i++)
	{
		if (v[i].left ^ v[i].op) balance++;
		else balance--;
 
		vector <edge> tr;
 
		if (i != v.size () - 1) 
		{
			ans += max (0LL, balance * ll (v[i+1].x - v[i].x));
			if (balance) unite (ind (v[i].x, coord), ind (v[i+1].x, coord));
			else tr.push_back ({ind (v[i].x, coord), ind (v[i+1].x, coord), (v[i+1].x - v[i].x)});
		}
	}
 
	sort (tr.begin(), tr.end());
 
	for (edge e : tr)
	{
		if (find (e.u) != find (e.v))
		{
			ans += e.c;
			unite (e.u, e.v);
		}
	}
 
	return ans;
}

Compilation message

railroad.cpp: In function 'll plan_roller_coaster(std::vector<int>, std::vector<int>)':
railroad.cpp:81:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < s.size (); i++)
                  ~~^~~~~~~~~~~
railroad.cpp:101:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < coord.size (); i++)
                  ~~^~~~~~~~~~~~~~~
railroad.cpp:110:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < s.size (); i++)
                  ~~^~~~~~~~~~~
railroad.cpp:111:26: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
   unite (ind (s[i], coord), ind (t[i], coord));
                          ^
railroad.cpp:70:5: note: in passing argument 2 of 'int ind(int, std::vector<long long int>&)'
 int ind (int a, vector <ll>& s)
     ^~~
railroad.cpp:111:45: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
   unite (ind (s[i], coord), ind (t[i], coord));
                                             ^
railroad.cpp:70:5: note: in passing argument 2 of 'int ind(int, std::vector<long long int>&)'
 int ind (int a, vector <ll>& s)
     ^~~
railroad.cpp:117:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < v.size (); i++)
                  ~~^~~~~~~~~~~
railroad.cpp:124:9: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if (i != v.size () - 1) 
       ~~^~~~~~~~~~~~~~~~
railroad.cpp:127:42: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
    if (balance) unite (ind (v[i].x, coord), ind (v[i+1].x, coord));
                                          ^
railroad.cpp:70:5: note: in passing argument 2 of 'int ind(int, std::vector<long long int>&)'
 int ind (int a, vector <ll>& s)
     ^~~
railroad.cpp:127:65: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
    if (balance) unite (ind (v[i].x, coord), ind (v[i+1].x, coord));
                                                                 ^
railroad.cpp:70:5: note: in passing argument 2 of 'int ind(int, std::vector<long long int>&)'
 int ind (int a, vector <ll>& s)
     ^~~
railroad.cpp:128:42: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
    else tr.push_back ({ind (v[i].x, coord), ind (v[i+1].x, coord), (v[i+1].x - v[i].x)});
                                          ^
railroad.cpp:70:5: note: in passing argument 2 of 'int ind(int, std::vector<long long int>&)'
 int ind (int a, vector <ll>& s)
     ^~~
railroad.cpp:128:65: error: invalid initialization of reference of type 'std::vector<long long int>&' from expression of type 'std::vector<int>'
    else tr.push_back ({ind (v[i].x, coord), ind (v[i+1].x, coord), (v[i+1].x - v[i].x)});
                                                                 ^
railroad.cpp:70:5: note: in passing argument 2 of 'int ind(int, std::vector<long long int>&)'
 int ind (int a, vector <ll>& s)
     ^~~
railroad.cpp:128:88: error: no matching function for call to 'std::vector<edge>::push_back(<brace-enclosed initializer list>)'
    else tr.push_back ({ind (v[i].x, coord), ind (v[i+1].x, coord), (v[i+1].x - v[i].x)});
                                                                                        ^
In file included from /usr/include/c++/7/vector:64:0,
                 from railroad.cpp:5:
/usr/include/c++/7/bits/stl_vector.h:939:7: note: candidate: void std::vector<_Tp, _Alloc>::push_back(const value_type&) [with _Tp = edge; _Alloc = std::allocator<edge>; std::vector<_Tp, _Alloc>::value_type = edge]
       push_back(const value_type& __x)
       ^~~~~~~~~
/usr/include/c++/7/bits/stl_vector.h:939:7: note:   no known conversion for argument 1 from '<brace-enclosed initializer list>' to 'const value_type& {aka const edge&}'
/usr/include/c++/7/bits/stl_vector.h:953:7: note: candidate: void std::vector<_Tp, _Alloc>::push_back(std::vector<_Tp, _Alloc>::value_type&&) [with _Tp = edge; _Alloc = std::allocator<edge>; std::vector<_Tp, _Alloc>::value_type = edge]
       push_back(value_type&& __x)
       ^~~~~~~~~
/usr/include/c++/7/bits/stl_vector.h:953:7: note:   no known conversion for argument 1 from '<brace-enclosed initializer list>' to 'std::vector<edge>::value_type&& {aka edge&&}'