Submission #1000649

#	Time	Username	Problem	Language	Result	Execution time	Memory
1000649		vjudge1	Safety (NOI18_safety)	C++17	3 / 100	58 ms	8352 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;
	ll s[n];
	for (int i = 0; i < n; i++) cin >> s[i];
	multiset<ll> tl, tr;
	ll ans = 0;
	tl.insert(s[0]);
	tr.insert(s[0]);
	for (ll i = 1; i < n; i++) {
		ll x = (*--tl.end()) - h * i;
		ll y = (*tr.begin()) + h * i;
		if (x <= s[i] && s[i] <= y) {
			tl.insert(s[i] + h * i);
			tr.insert(s[i] - h * i);
		}else if (y < s[i]) {
			tr.erase(tr.begin());
			tl.insert(y + h * i);
			tr.insert(s[i] - h * i);
			ans += s[i] - y;
		}else {
			tl.erase(--tl.end());
			tr.insert(x - h * i);
			tl.insert(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:61:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   61 | main() {
      | ^~~~

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