This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#define int long long
#define float double
#define pb push_back
#define F first
#define S second
#define T int t; cin >> t; while(t--)
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
using namespace std;
/// Benzema in the best player in the world
const int inf = 8e18;
const int N = 1e6 + 6;
const int M = 1e3 + 3;
const int LOG = 31;
const int mod = 1e9 + 7;
const float pi = atan(1) * 4;
int n, k, a[N], b[N], cnt[2 * N];
vector<int> v;
main()
{
IOS
cin >> n >> k;
for(int i = 1; i <= n; i++)
cin >> a[i];
for(int i = 1; i <= n; i++)
cin >> b[i];
int pos = 0, posn = 0, neg = 0, negn = 0, zn = 0;
for(int i = 1; i <= n; i++)
{
if (a[i] < b[i])
neg += b[i] - a[i], negn++, cnt[b[i] - a[i]]++;
else if (a[i] > b[i])
pos += a[i] - b[i], posn++;
else
zn++;
}
int x = 0, ans = neg + pos;
for(int i = 1; i <= 2e6; i++)
{
neg -= negn;
negn -= cnt[i];
pos += posn + zn;
posn += zn;
zn = cnt[i];
if (ans > neg + pos)
x = i, ans = neg + pos;
cnt[i] = 0;
}
neg = 0, negn = 0, posn = 0, pos = 0, zn = 0;
for(int i = 1; i <= n; i++)
{
if (b[i] < a[i])
neg += a[i] - b[i], negn++, cnt[a[i] - b[i]]++;
else if (b[i] > a[i])
pos += b[i] - a[i], posn++;
else
zn++;
}
for(int i = 1; i <= 2e6; i++)
{
neg -= negn;
negn -= cnt[i];
pos += posn + zn;
posn += zn;
zn = cnt[i];
if (ans > neg + pos)
x = -i, ans = neg + pos;
cnt[i] = 0;
}
for(int i = 1; i <= n; i++)
a[i] += x, v.pb(abs(a[i] - b[i]));
sort(v.begin(), v.end(), greater<int>());
for(int i = 0; i < k; i++)
ans -= v[i];
cout << ans;
}
///use scanf
Compilation message (stderr)
simfonija.cpp:21:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
21 | main()
| ^~~~
# | 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... |
# | 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... |