Trash Problem - 每天一把CF - 20200927

28补27

题目

原题链接:https://codeforc.es/problemset/problem/1418/D

思路

题目大意：现在我们有一个坐标轴OX，在这个轴上的某些位置有垃圾。我们有一个操作，将垃圾从一个坐标扫到其临近的一个坐标（x->x+1 OR x->x-1）。我们最终的目标是将所有垃圾扫到最多两个点上，求最少的操作步数。

这道题认真去思考一下，能够发现我们最终是要在坐标轴选定两个点，然后将其他点的垃圾一步步移到这两个点，很明显，从两端开始往中间移是必要的，同时我们使这两个点的距离尽量大，并且这两个所在的位置本来就是有垃圾的吗，这样就没有多余操作，所以最终的最小操作步数就是垃圾的最大最小坐标差减去相邻垃圾间的最大距离。

代码实现

#include <iostream>
#include <set>
using namespace std;

typedef long long ll;

set<ll> s;
multiset<ll> m;
int n, q;

ll sum = 0;

void add(int x) {
	s.insert(x);
	auto it = s.find(x);
	if (next(it) != s.end())
		sum += *next(it) - *it, m.insert(*next(it) - *it);
	if (it != s.begin())
		sum += *it - *prev(it), m.insert(*it - *prev(it));
	if (next(it) != s.end() && it != s.begin())//prev(it) != --s.begin()
		sum -= *next(it) - *prev(it), m.erase(m.find(*next(it) - *prev(it)));
}

void erase(int x) {
	auto it = s.find(x);
	if (next(it) != s.end())
		sum -= *next(it) - *it, m.erase(m.find(*next(it) - *it));
	if (it != s.begin())
		sum -= *it - *prev(it), m.erase(m.find(*it - *prev(it)));
	if (next(it) != s.end() && it != s.begin())
		sum += *next(it) - *prev(it), m.insert(*next(it) - *prev(it));

	s.erase(x);
}

void get_ans() {
	if (m.size() < 2)
		cout << 0 << endl;
	else
		cout << sum - *--m.end() << endl;
}

int main() {
	ios::sync_with_stdio(false);
	cin >> n >> q;

	ll tp;
	for (int i = 1; i <= n; i++)
	{
		cin >> tp;
		add(tp);
	}

	get_ans();
	for (int i = 1; i <= q; i++)
	{
		ll a, b;
		cin >> a >> b;
		if (a == 1)
			add(b);
		else
			erase(b);
		get_ans();
	}
	return 0;
}