Codeforces 425A multiset使用

题意

• 给你一个数列，你可以最多swap两个元素k次，问你交换后数列的子串中，和最大的可以到多大

思路

• 比较暴力的算法，每次看(i,j)这个子串，让它外面的元素有序，内部的也有序，那么最多迭代min(k, 外面元素的个数，内部元素的个数)次，如果外面最大的>里面最小的，那么就交换，否则结束迭代
• 为了优化下复杂度，也为了编程简单，使用了mutiset来维护一个有序的序列
• 复杂度为O(k * n^2 * log(n) )

实现

#include<bits/stdc++.h>
using namespace std;
const int maxn = 204;
int a[maxn];
int n,k;
multiset<int> set_in,set_out;
multiset<int>::iterator it_in,it_out;
const int inf = 0x3f3f3f3f;

int main(){
	cin>>n>>k;
	for (int i=0;i<n;i++)
		scanf("%d",&a[i]);
	int ans = -inf;
	for (int i=0;i<n;i++){
		set_in.clear();
		set_out.clear();
		for (int j=0;j<n;j++){
			set_out.insert(a[j]);
		}
		int sum = 0;
		for (int j=i;j<n;j++){
			set_in.insert(a[j]);
			set_out.erase(set_out.lower_bound(a[j]));
			sum += a[j];
			if (set_out.size() == 0){
				ans = max(ans,sum);
				continue;
			}
			int t; 
			int now = sum;
			for (t=0,it_in=set_in.begin(),it_out=--set_out.end();t<min((int)set_out.size(),min(k,(int)set_in.size()));t++,it_in++,it_out--){
				if (*it_in >= *it_out){
					break;
				}
				now += *it_out - *it_in; 
			}
			ans = max(ans,now);
		}
	}
	cout << ans <<"\n";
	
	return 0;
}