이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define int int64_t
#define pii array<int,2>
struct cmp {
bool operator()(const pii &x, const pii &y) {
return (x[0] + x[1]) < (y[0] + y[1]);
}
};
signed main() {
int k, n;cin>>k>>n;
vector<pii> dif;
int dis = 0;
for (int i=0;i<n;i++) {
char c1,c2;
int p1,p2;
cin>>c1>>p1>>c2>>p2;
if (c1 == c2) {
dis += abs(p1-p2);
}
else {
dif.push_back({p1,p2});
}
}
if (dif.size() == 0) {
cout<<dis<<"\n";
return 0;
}
if (k == 1 || n == 1) {
vector<int> al;
for (auto x:dif) {
al.push_back(x[0]);
al.push_back(x[1]);
}
sort(al.begin(),al.end());
int med = al[al.size() / 2];
for (int x:al) {
dis += abs(x - med);
}
dis += dif.size();
cout<<dis<<"\n";
return 0;
}
sort(dif.begin(),dif.end(), cmp());
multiset<int> m1, m2;
int s1 = 0, s2 = 0;
function<void(int)> insert=[&](int num) {
m1.insert(num);
s1 += num;
while ((m1.size() > m2.size() && ((m1.size() + m2.size()) % 2 == 0)) ||
(m1.size() - 1 > m2.size() && ((m1.size() + m2.size()) % 2 == 1))) {
num = *m1.rbegin();
m1.erase(m1.find(num));
m2.insert(num);
s1 -= num;
s2 += num;
}
};
vector<int> cst1;
vector<int> tmp1;
for (auto x:dif) {
insert(x[0]);
insert(x[1]);
int med = *m1.rbegin();
tmp1.push_back(med);
cst1.push_back((m1.size() * med - s1) + (s2 - m2.size() * med));
}
reverse(dif.begin(),dif.end());
vector<int> cst2;
vector<int> tmp2;
m1.clear();m2.clear();
s1=s2=0;
for (auto x:dif) {
insert(x[0]);
insert(x[1]);
int med = *m1.rbegin();
tmp2.push_back(med);
cst2.push_back((m1.size() * med - s1) + (s2 - m2.size() * med));
}
reverse(cst2.begin(),cst2.end());
reverse(tmp2.begin(), tmp2.end());
int res = 1e17;
for (int i=1;i<cst1.size();i++) {
res = min(res, cst1[i-1] + cst2[i]);
}
res += dif.size();
cout<<res+dis<<"\n";
}
컴파일 시 표준 에러 (stderr) 메시지
bridge.cpp: In function 'int main()':
bridge.cpp:102:19: warning: comparison of integer expressions of different signedness: 'int64_t' {aka 'long int'} and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
102 | for (int i=1;i<cst1.size();i++) {
| ~^~~~~~~~~~~~
# | 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... |