Submission #1000652

#	Time	Username	Problem	Language	Result	Execution time	Memory
1000652		vjudge1	Safety (NOI18_safety)	C++17	100 / 100	37 ms	4704 KiB

safety

//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("Ofast,unroll-loops,fast-math,O3")
 
#include<bits/stdc++.h>	
 
using namespace std;
 
#define all(a) a.begin(), a.end()                                                   
#define rall(a) a.rbegin(), a.rend()                 
#define sz(a) (int)a.size()
#define s second
#define f first
 
using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
 
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
     
vector<pii> rid = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
vector<pii> dir = {{-1, -1}, {-1, 1}, {1, -1}, {1, 1}};
 
const int N = 1e6 + 1, mod = 998244353;
 
const ll inf = 1e9;
 
double eps = 1e-15;
                                                
bool flg = 0;

void slv() {
	ll n, h;
	cin >> n >> h;
	int s[n];
	for (int i = 0; i < n; i++) cin >> s[i];
	priority_queue<ll> tl, tr;
	ll ans = 0;
	tl.push(s[0]);
	tr.push(-s[0]);
	for (int i = 1; i < n; i++) {
		ll x = tl.top() - h * i;
		ll y = (-tr.top()) + h * i;
		if (x <= s[i] && s[i] <= y) {
			tl.push(s[i] + h * i);
			tr.push(-(s[i] - h * i));
		}else if (y < s[i]) {
			tr.pop();
			tl.push(y + h * i);
			tr.push(-(s[i] - h * i));
			tr.push(-(s[i] - h * i));
			ans += (s[i] - y);
		}else {
			tl.pop();
			tr.push(-(x - h * i));
			tl.push(s[i] + h * i);
			tl.push(s[i] + h * i);
			ans += (x - s[i]);
		}
	}	
	cout << ans;
}       	                                                                    
 
main() {
	//freopen("input.txt", "r", stdin);                                                                                     
	//freopen("output.txt", "w", stdout);                                                                                     
	ios_base::sync_with_stdio(0);	                                                                                       
	cin.tie(0);
	int tp = 1;
	if (flg) cin >> tp;
	while (tp--) {  	
		slv();
	}
}
//wenomechainsama      

Compilation message (stderr)

safety.cpp:63:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   63 | 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