Submission #414476

#	Time	Username	Problem	Language	Result	Execution time	Memory
414476		grt	Palembang Bridges (APIO15_bridge)	C++17	63 / 100	2098 ms	3304 KiB

bridge

#include <bits/stdc++.h>
#define ST first
#define ND second
#define PB push_back

using namespace std;
using ll = long long;
using pi = pair<int,int>;
using vi = vector<int>;

const int nax = 100 * 1000 + 10;
const ll INF = 1e18;
int n, m, k;
int opn[nax], clo[nax];
vi all;
ll add = 0;

int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cin >> k >> n;
	for(int i = 0; i < n; ++i) {
		char t1, t2;
		int d1, d2;
		cin >> t1 >> d1 >> t2 >> d2;
		add += abs(d1 - d2);
		if(t1 != t2) {
			opn[m] = min(d1, d2);
			clo[m] = max(d1, d2);
			all.PB(d1);
			all.PB(d2);
			m++;
		}
	}
	sort(all.begin(), all.end());
	if(m == 0) {
		cout << add << "\n";
		return 0;
	}
	if(k == 1) {
		sort(opn, opn + m);
		sort(clo, clo + m);
		int ptr = 0, ptr2 = 0;
		ll Lsum = 0, Rsum = 0;
		ll ans = INF;
		for(int i = 0; i < m; ++i) Rsum += opn[i];
		for(int x : all) {
			while(ptr2 < m && clo[ptr2] < x) {
				Lsum += clo[ptr2];
				ptr2++;
			}
			while(ptr < m && opn[ptr] < x) {
				Rsum -= opn[ptr];
				ptr++;
			}
			ll d = (ll)x * ptr2 - Lsum - (ll)x * (m - ptr) + Rsum;
			ans = min(ans, d);
		}
		cout << ans * 2 + add + m << "\n";
	} else {
		ll ans = INF;
		for(int x : all) {
			ll cur = 0;
			priority_queue<pi>PQ;
			vi nopn, nclo;
			for(int i = 0; i < m; ++i) {
				if(opn[i] <= x) {
					if(clo[i] >= x) continue;
					cur += abs(x - clo[i]);
				} else {
					PQ.push({-(clo[i] + opn[i] - x), opn[i]});
					nopn.PB(opn[i]);
					nclo.PB(clo[i]);
				}
			}
			sort(nopn.begin(), nopn.end());
			sort(nclo.begin(), nclo.end());
			int ptr = 0, ptr2 = 0;
			ll Lsum = 0, Rsum = 0;
			int lft = 0;
			for(int i = 0; i < (int)nopn.size(); ++i) Rsum += nopn[i];
			for(int y : all) {
				if(y < x) continue;
				while(ptr2 < (int)nclo.size() && nclo[ptr2] < y) {
					Lsum += nclo[ptr2];
					ptr2++;
					lft++;
				}
				while(ptr < (int)nopn.size() && nopn[ptr] < y) {
					Rsum -= nopn[ptr];
					ptr++;
				}
				while(!PQ.empty()) {
					auto z = PQ.top();
					if(y < -z.ST) break;
					PQ.pop();
					lft--;
					cur += z.ND - x;
					z.ST = -z.ST;
					Lsum -= (z.ST - z.ND + x);
				}
				ll d = cur + Rsum - (ll)y * ((int)nopn.size() - ptr) + (ll)lft * y - Lsum;
				ans = min(ans, d);
			}
		}
		cout << ans * 2 + add + m << "\n";
	}
	
}

• Subtask 1: 8 / 8
• Subtask 2: 14 / 14
• Subtask 3: 9 / 9
• Subtask 4: 32 / 32
• Subtask 5: 0 / 37