Submission #1001226

#	Time	Username	Problem	Language	Result	Execution time	Memory
1001226		vjudge1	Safety (NOI18_safety)	C++17	0 / 100	2003 ms	262144 KiB

safety

// #pragma optimize ("g",on)
// #pragma GCC optimize("inline")
// #pragma GCC optimize ("Ofast")
// #pragma GCC optimize ("unroll-loops")
// #pragma GCC optimize ("03")
// #pragma GCC target ("sse,sse2,sse3,ssse3,sse4,popcnt,abm,avx2,mmx,fma,avx,tune=native")

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>

#define file(s) freopen(s".in", "r", stdin); freopen(s".out", "w", stdout);
#define adiyer(); ios_base::sync_with_stdio(0); cin.tie(0);
#define bitcount(n) __builtin_popcountll(n)
#define puts(x) cout << (x ? "YES\n" : "NO\n");
#define ent (i == n ? '\n' : ' ')
#define all(x) x.begin(), x.end()
#define md ((l + r) >> 1)
#define rv(v) ((v << 1) | 1)
#define lv(v) (v << 1)
#define rs(v) rv(v), md + 1, r
#define ls(v) lv(v), l, md
#define len(s) (int) s.size()

#define yes {cout << "YES\n"; return; }
#define no {cout << "no\n"; return; }
#define skip continue
#define pb push_back
#define S second
#define F first

// #define int long long

using namespace std;
using namespace __gnu_pbds;

typedef long long ll;
typedef long double ld;
typedef vector < ll > vll;
typedef pair < ll, ll > pll;
typedef vector < pair < ll, ll > > vpll;
typedef tree < ll, null_type, less < ll >, rb_tree_tag, tree_order_statistics_node_update > ordered_set;

const int dx[8] = {1, 0, -1, 0, 1, 1, -1, -1};
const int dy[8] = {0, 1, 0, -1, -1, 1, -1, 1};
const int N = 2e5 + 12;
const int K = 500 + 12;
const int MAX = 2e5;
const int mod = 998244353;
const ll inf = 1e9 + 10;

mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

ll n, k, res;
ll a[N];

multiset < ll > s1, s2;

void output(){
	cin >> n >> k;
	for(ll i = 1; i <= n; i++) cin >> a[i];
	s1.insert(a[1]), s2.insert(a[1]);
	for(ll i = 2; i <= n; i++){
		ll x = (a[i] + (i - 1) * k), y = (a[i] - (i - 1) * k);
		if(x >= *s1.rbegin() && y <= *s2.begin()){
			s1.insert(x), s2.insert(y);
			skip;
		}
		while(len(s1) && x < *s1.rbegin()){
			s1.insert(x), s1.insert(x);
			s2.insert(*s1.rbegin() - (i - 1) * k * 2);
			res += abs(*s1.rbegin() - x);
			s1.erase(s1.find(*s1.rbegin()));
		}
		while(len(s2) && a[i] - (i - 1) * k > *s2.begin()){
			s2.insert(y), s2.insert(y);
			s1.insert(*s2.begin() + (i - 1) * k * 2);
			res += abs(*s2.begin() - y);
			s2.erase(s2.find(*s2.begin()));
		}
	}
	cout << res;
}

const bool cases = 0;

signed main(){
//  file("disrupt");
    adiyer();
    int tt = 1;
    if(cases) cin >> tt;
    for(int i = 1; i <= tt; i++){
//      cout << "Case " << i << ":\n";
        output();
    }
}

• Subtask 1: 0 / 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