2021牛客暑期多校训练营1 Game of Swapping Numbers

Posted 2021-08-26 一只特立独行的猫

2021牛客暑期多校训练营1 Game of Swapping Numbers
题目大意：
给出两个长度为n的数组，需要进行k次操作，使得两个数组的每一位的差值的和最大。思路：ma存放被减数，mi存放减数，被减数的最小值如果比减数的最大值小，就进行一次交换，带来的收益是2*(mi[max]-ma[min])。

#include<iostream>
#include<algorithm>

using namespace std;

int a[500005];
int b[500005];
int ma[500005];
int mi[500005];
long long ans;

int main() {
	int n, k;
	cin >> n >> k;
	for (int i = 0; i < n; i++) {
		cin>>a[i];
	}
	for (int i = 0; i < n; i++) {
		cin >> b[i];
	}
	for (int i = 0; i < n; i++) {
		ma[i] = max(a[i], b[i]);//存放减数
		mi[i] = min(a[i], b[i]);//存放被减数
		ans += ma[i] - mi[i];//存放原始差值
	}
	sort(ma, ma + n);
	sort(mi, mi + n);
	for (int i = 0; i < k&&i<n; i++) {
		if (ma[i] < mi[n-i-1]) {
			ans += 2 * (long long)(mi[n - i - 1] - ma[i]);//交换减数与被减数会带来的收益
		}
		else {
			break;
		}
	}
	cout << ans << endl;
	return 0;
}