Codeforces 425A multiset使用

luke2834

已于 2024-01-25 11:34:21 修改

阅读量536

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM_Codeforces ACM_STL模板文章标签： codeforces mutiset stl

于 2015-12-29 14:08:18 首次发布

本文链接：https://blog.csdn.net/luke2834/article/details/50426273

ACM_Codeforces 同时被 2 个专栏收录

34 篇文章

订阅专栏

ACM_STL模板

8 篇文章

订阅专栏

本文介绍了一种解决特定子串和最大化问题的方法。通过维护内外两组元素的有序集合，并利用multiset进行优化，实现了复杂度为O(k*n^2*log(n))的算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意

给你一个数列，你可以最多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;
}