이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/stdc++.h"
#define ll long long int
#define pb push_back
#define pii pair<int,int>
#define ff first
#define ss second
#define sz size()
const int N = 2e5 + 1;
using namespace std;
ll n, k, x[N], y[N], ans, p[N], q[N], mn = 1e15, cnt, cnt1, sum, sum1;
char s[N], d[N];
vector <pii> v;
vector <int> v1;
multiset <int> l, r;
bool cmp(pii a, pii b){
if(a.ff + a.ss <= b.ff + b.ss) return 1;
else return 0;
}
void med(int a){
if(l.empty()) {
l.insert(a);
sum += a;
return;
}
if((int)l.sz <= (int)r.sz){
auto t = r.begin();
ll aa = *t;
if(a > aa) {
sum += aa;
l.insert(aa);
sum1 += a;
sum1 -= aa;
r.erase(aa);
r.insert(a);
}
else{
l.insert(a);
sum += a;
}
}
else{
auto t = l.end();
t--;
ll aa = *t;
if(a < aa){
sum -= (ll)aa;
sum += a;
l.erase(aa);
l.insert(a);
r.insert(aa);
sum1 += aa;
}
else{
r.insert(a);
sum1 += a;
}
}
}
int main() {
// freopen ("input.txt", "r", stdin);
ios::sync_with_stdio(false); cin.tie(nullptr);
cin >> k >> n;
for(int i = 1; i <= n; i++){
cin >> s[i] >> x[i] >> d[i] >> y[i];
if(s[i] != d[i]){
ans++;
v.pb({x[i],y[i]});
}
else ans += abs(x[i]-y[i]);
}
sort(v.begin(), v.end(), cmp);
for(int i = 0; i < (int)v.sz; i++){
med(v[i].ff);
med(v[i].ss);
auto t = l.end();
t--;
int med1 = *t;
p[i] = (cnt*med1 - sum) + (sum1 - cnt1*med1);
}
cnt = sum = cnt1 = sum1 = 0;
l.clear();r.clear();
for(int i = (int)v.sz-1; i >= 0; i--){
med(v[i].ff);
med(v[i].ss);
auto t = l.end();
t--;
int med1 = *t;
q[i] = (cnt*med1 - sum) + (sum1 - cnt1*med1);
}
for(int i = 0; i < (int)v.sz; i++){
mn = min(mn, p[i] + q[i+1]);
}
mn = min(mn,q[0]);
// cout << sum << " " << sum1;
// return 0;a
if(k == 1) mn = q[0];
// assert (q[0] == p[(int)v.sz - 1]);
cout << mn + ans;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |