Submission #961603

#	Time	Username	Problem	Language	Result	Execution time	Memory
961603		vjudge1	Safety (NOI18_safety)	C++17	100 / 100	156 ms	21088 KiB

safety

// #pragma GCC target("avx2")
// #pragma GCC optimize("O3")
// #include <x86intrin.h>

#include <bits/stdc++.h>
#include <chrono>
#include <random>

// @author: Vlapos

namespace operators
{
	template <typename T1, typename T2>
	std::istream &operator>>(std::istream &in, std::pair<T1, T2> &x)
	{
		in >> x.first >> x.second;
		return in;
	}

	template <typename T1, typename T2>
	std::ostream &operator<<(std::ostream &out, std::pair<T1, T2> x)
	{
		out << x.first << " " << x.second;
		return out;
	}

	template <typename T1>
	std::istream &operator>>(std::istream &in, std::vector<T1> &x)
	{
		for (auto &i : x)
			in >> i;
		return in;
	}

	template <typename T1>
	std::ostream &operator<<(std::ostream &out, std::vector<T1> &x)
	{
		for (auto &i : x)
			out << i << " ";
		return out;
	}

	template <typename T1>
	std::ostream &operator<<(std::ostream &out, std::set<T1> &x)
	{
		for (auto &i : x)
			out << i << " ";
		return out;
	}

	template <typename T1>
	std::ostream &operator<<(std::ostream &out, std::multiset<T1> &x)
	{
		for (auto &i : x)
			out << i << " ";
		return out;
	}
}

// name spaces
using namespace std;
using namespace operators;
// end of name spaces

// defines
#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
#define pii pair<int, int>
#define pll pair<ll, ll>
#define f first
#define s second
#define uint unsigned int
#define all(vc) vc.begin(), vc.end()
// end of defines

// usefull stuff

void boost()
{
	ios_base ::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
}

inline int getbit(int &x, int &bt) { return (x >> bt) & 1; }

const int dx4[4] = {-1, 0, 0, 1};
const int dy4[4] = {0, -1, 1, 0};
const int dx8[8] = {-1, -1, -1, 0, 0, 1, 1, 1};
const int dy8[8] = {-1, -0, 1, -1, 1, -1, 0, 1};

const ll INF = 2e18;
const int BIG = (1e9) * 2 + 100;
const int MAXN = (1e5) + 5;
const int MOD7 = (1e9) + 7;
const int MOD9 = (1e9) + 9;
const uint MODFFT = 998244353;

#define int ll

struct test
{
	multiset<int> decslp, incslp;
	int slp0 = 0, dpmn = 0, pmn = 0;
	int mvm = 0;

	void addSlope(int x)
	{
		assert(x >= -mvm);
		dpmn += abs(pmn - x);
		if (incslp.size() and (*incslp.begin()) + mvm <= x)
		{
			incslp.insert(x - mvm);
			incslp.insert(x - mvm);
		}
		else
		{
			decslp.insert(x + mvm);
			decslp.insert(x + mvm);
		}

		if (decslp.size() > -slp0)
		{
			auto cur = *decslp.rbegin() - mvm;
			decslp.erase(prev(decslp.end()));
			incslp.insert(cur - mvm);
		}
		if (decslp.size() < -slp0)
		{
			auto cur = *incslp.begin() + mvm;
			incslp.erase(incslp.begin());
			decslp.insert(cur + mvm);
		}

		int topm = *decslp.rbegin() - mvm;
		// cout << dpmn << " " << pmn << " " << topm << ": ";
		dpmn -= max(topm - pmn, 0LL);
		pmn = topm;
	}

	void solve(int testcase)
	{
		boost();
		int n, h;
		cin >> n >> h;

		for (int i = 0; i < n; ++i)
		{
			int x;
			cin >> x;
			slp0--;
			mvm += h;
			pmn -= h;
			addSlope(x);
		}

		cout << dpmn << "\n";
	}
};

main()
{
	boost();
	int q = 1;
	// cin >> q;
	for (int i = 0; i < q; i++)
	{
		test t;
		t.solve(i);
	}
	return 0;
}
//[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]//
//                                                                                    //
//                               Coded by Der_Vlἀpos                                  //
//                                                                                    //
//[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[[]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]]//

Compilation message (stderr)

safety.cpp: In member function 'void test::addSlope(long long int)':
safety.cpp:124:21: warning: comparison of integer expressions of different signedness: 'std::multiset<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
  124 |   if (decslp.size() > -slp0)
      |       ~~~~~~~~~~~~~~^~~~~~~
safety.cpp:130:21: warning: comparison of integer expressions of different signedness: 'std::multiset<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
  130 |   if (decslp.size() < -slp0)
      |       ~~~~~~~~~~~~~~^~~~~~~
safety.cpp: At global scope:
safety.cpp:163:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
  163 | main()
      | ^~~~

• Subtask 1: 3 / 3
• Subtask 2: 4 / 4
• Subtask 3: 9 / 9
• Subtask 4: 5 / 5
• Subtask 5: 6 / 6
• Subtask 6: 11 / 11
• Subtask 7: 11 / 11
• Subtask 8: 22 / 22
• Subtask 9: 29 / 29