2024牛客五一集训派对day4 G.Game of Swapping Numbers

题目

思路：

补充证明：

1、当n > 2时，至少有两个区间是ai > bi（设为正） 或 ai < bi（设为逆）；

2、当 最优操作执行完后，任意两个区间都有交集；

3、又因为当两个区间有交集（无论是相交还是包含）并且同正或同逆时，对这两个区间进行操作不会影响答案

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fi first
#define se second
#define lson p << 1
#define rson p << 1 | 1
const int maxn = 1e6 + 5, inf = 1e18, maxm = 4e4 + 5;
const int N = 1e6;
const int mod = 1e9 + 7;
// const int mod = 998244353;
//const __int128 mod = 212370440130137957LL;
//  int a[1005][1005];
// bool vis[505][505];
int n, m;
int a[maxn];
int b[maxn];
string s;

// struct Node{
//     int a, b;
//     // int val, id;
//     bool operator<(const Node &u)const{
//         
//     }
// }node[maxn];

//long long ? maxn ? n? m?
void solve(){
    int res = 0;
    int k;
    cin >> n >> k;
	for(int i = 1; i <= n; i++){
		cin >> a[i];
	} 
	for(int i = 1; i <= n; i++){
		cin >> b[i];
	}
	vector<int> l(n + 5), r(n + 5);
	for(int i = 1; i <= n; i++){
		l[i] = min(a[i], b[i]);
		r[i] = max(a[i], b[i]);
	}
	if(n == 2){
		if(k % 2 == 1){
			swap(a[1], a[2]);
		}
		res = abs(a[1] - b[1]) + abs(a[2] - b[2]);
		cout << res << '\n';
		return; 
	}
    sort(l.begin() + 1, l.begin() + n + 1, greater<int>());
	sort(r.begin() + 1, r.begin() + n + 1);
	for(int i = 1; i <= n; i++){
		res += abs(a[i] - b[i]);
	}
	for(int i = 1; i <= min(n, k); i++){//让最大的l和最小的r配对（即让两个无交集的、距离最大的区间配对），进行一次操作，然后第二大的l和第二小的l配对,进行一次操作，...,每进行一次操作，产生2*(l[i]-r[i])的贡献,
                                        //每进行一次操作，设新生成的两个区间为[l[i], r[i]], [l[j], r[j]],l[i]和l[j]会小于所有r，同理r[i],r[j]会大于所有l，因此这两个区间对答案不再有贡献
		res += max(0LL, 2LL * (l[i] - r[i]));
	}
    
	cout << res << '\n';
}   
    
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);

    int T = 1;
//    cin >> T;
    while (T--)
    {
        solve();
    }
    return 0;
}