Submission #530512

#	Time	Username	Problem	Language	Result	Execution time	Memory
530512		LucaDantas	Roller Coaster Railroad (IOI16_railroad)	C++17	30 / 100	1328 ms	33552 KiB

railroad

#include "railroad.h"
#include <bits/stdc++.h>
using namespace std;

constexpr int maxn = 4e5+10;

struct DSU {
	int pai[maxn], peso[maxn];
	DSU() { for(int i = 0; i < maxn; i++) pai[i] = i, peso[i] = 1; }
	int find(int x) { return pai[x] == x ? x : pai[x] = find(pai[x]); }
	void join(int a, int b) {
		a = find(a), b = find(b);
		if(a == b) return;
		if(peso[a] < peso[b])
			swap(a, b);
		pai[b] = a;
		peso[a] += peso[b];
	}
} dsu;

map<int,int> qtd;

struct Itv { int l, r; bool operator<(const Itv& o) { return r-l > o.r-o.l; } };
vector<Itv> itv;

int pai[maxn];
bool vis[maxn];

long long plan_roller_coaster(vector<int> s, vector<int> t) {
	int n = (int) s.size(), ans = 0;
	
	for(int i = 0; i < n; i++) {
		qtd[s[i]]++, qtd[t[i]]--;
		itv.push_back({min(s[i], t[i]), max(s[i], t[i])});
	}

	for(auto it = qtd.begin(); it != qtd.end(); ++it) {
		if(it != qtd.begin())
			it->second += prev(it)->second;

		if(it->second != 1 && next(it) != qtd.end()) {
			itv.push_back({it->first, next(it)->first});
			if(it->second > 1)
				ans += (it->second-1) * (next(it)->first - it->first);
		}
	}

	// SUBTASK 3 <---------------------------------------------------------------------------------------------------------------->
	if(ans) return ans; // se eu tive q colocar alguma aresta pra trás então não da pra fazer com 0

	// agora que eu tenho todos os intervalos, tenho que checar as intersecções
	
	sort(itv.begin(), itv.end()); // ordeno pelo de maior tamanho

	int coord = 0;
	for(auto& it : qtd)
		it.second = ++coord;

	iota(pai+1, pai+coord+1, 1); // todo mundo começa apontando pro próximo
	
	for(auto [l, r] : itv) {
		l = qtd[l], r = qtd[r];
		vis[l] = 1, vis[r] = 1;
		while(l < r) {
			if(vis[l])
				dsu.join(l, r);
			int nxt = pai[l];
			pai[l] = r;
			l = nxt;
		}
	}

	int comp = dsu.find(qtd[itv[0].l]);

	for(int i = 0; i < itv.size(); i++)
		if(dsu.find(qtd[itv[i].l]) != comp) return 1; // pra subtask q só quer saber se dá com zero bastar printar 1
	
	// se não deu ruim no for acima então todo mundo está ligado na mesma componente e dá pra fazer 0
	return 0;
}

Compilation message (stderr)

railroad.cpp: In function 'long long int plan_roller_coaster(std::vector<int>, std::vector<int>)':
railroad.cpp:75:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<Itv>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   75 |  for(int i = 0; i < itv.size(); i++)
      |                 ~~^~~~~~~~~~~~

• Subtask 1: 0 / 11
• Subtask 2: 0 / 23
• Subtask 3: 30 / 30
• Subtask 4: 0 / 36