Atcoder beginner contest 128D - equeue

https://atcoder.jp/contests/abc128/tasks/abc128_d
思路：n,k都很小，可以暴力。先枚举用到的操作次数0-k，设此时操作次数为x。再枚举取的个数0-x，设取数的个数为y，经过推算可以发现可以取到两边的前(x+y)/2个（左边取最左边的a个，右边取最右边的b个，则a+b=(x+y)/2），且两边分别计算时也满足。当(x+y)/2不是整数时说明次数没用完，这在枚举到x-1时已经考虑到，可以忽略。最后枚举左边取几个，计算右边取几个，计算答案。时间复杂度是O(nk³)。其中统计我用的是multiset，比较方便。
还有一种做法比较简单：暴力枚举左边和右边取的数量，每次先将取到的加起来，在计算可以放回几个，在加起来的数中找出最小的负数放回（仅放回负数），最后统计答案。
做法一：

#include<cstdio>
#include<string>
#include<cstring>
#include<utility>
#include<cmath>
#include<map>
#include<queue>
#include<set>
#include<algorithm>
#include<vector>
#include<iostream>
#define ll long long
#define pii pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define inf 0x7fffffff
using namespace std;
int a[100];
multiset<int> s;
int ge(int x)
{
	int sum=0,n=0;
	__typeof(s.begin()) it=s.end();
	it--;
	for(;it!=s.begin();it--)
	{
		sum+=*it;
		n++;
		if(n==x)
		{
			break;
		}
	}
	if(n<x)
	{
		sum+=*(s.begin());
	}
	return sum;
}
void print()
{
	for(__typeof(s.begin()) it=s.begin();it!=s.end();it++)
	{
		printf("%d ",*it);
	}
	printf("\n");
}
int main()
{
	int i,j,k,n,m,x,y;
	scanf("%d%d",&n,&m);
	for(i=0;i<n;i++)
	{
		scanf("%d",&a[i]);
	}
	x=0;
	for(i=1;i<=m;i++)//uesd
	{
		if(i%2==0)
		{
			y=i/2+1;
			if(y>n)
			{
				continue;
			}
			for(j=2;j<=i;j+=2)//how many to choose
			{
				s.clear();
				for(k=0;k<y;k++)
				{
					s.insert(a[n-1-k]);
				}
				for(k=0;k<=y;k++)//left
				{
					x=max(x,ge(j));
					if(k==y)
					{
						break;
					}
					s.erase(s.find(a[n-y+k]));
					s.insert(a[k]);
				}
				y++;
				if(y>n)
				{
					break;
				}
			}
		}
		if(i%2==1)
		{
			y=i/2+1;
			if(y>n)
			{
				continue;
			}
			for(j=1;j<=i;j+=2)//how many to choose
			{
				s.clear();
				for(k=1;k<=y;k++)
				{
					s.insert(a[n-k]);
				}
				for(k=0;k<=y;k++)//left
				{
					x=max(x,ge(j));
					if(k==y)
					{
						break;
					}
					s.erase(s.find(a[n-y+k]));
					s.insert(a[k]);
				}
				y++;
				if(y>n)
				{
					break;
				}
			}
		}
	}
	printf("%d",x);
	return 0;
}

做法二：

#include<cstdio>
#include<string>
#include<cstring>
#include<utility>
#include<cmath>
#include<map>
#include<queue>
#include<set>
#include<algorithm>
#include<vector>
#include<iostream>
#define ll long long
#define pii pair<int,int>
#define mp make_pair
#define fi first
#define se second
#define inf 0x7fffffff
using namespace std;
int a[100];
vector<int> b;
int main()
{
	int i,j,k,n,m,x,y,l,sum,ans=0;
	scanf("%d%d",&n,&k);
	for(i=0;i<n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(i=0;i<=min(n,k);i++)
	{
		for(j=0;j<=min(n,k);j++)
		{
			if(i+j>n||i+j>k)
			{
				break;
			}
			b.clear();
			for(l=0;l<i;l++)
			{
				b.push_back(a[l]);
			}
			for(l=0;l<j;l++)
			{
				b.push_back(a[n-1-l]);
			}
			sort(b.rbegin(),b.rend());
			sum=0;
			for(l=0;l<i+j;l++)
			{
				sum+=b[l];
			}
			ans=max(ans,sum);
			x=k-i-j;
			y=i+j-1;
			while(x>0&&b.size()>0&&b[y]<0)
			{
				sum-=b[y];
				y--;
				x--;
			}
			ans=max(ans,sum);
		}
	}
	printf("%d",ans);
	return 0;
}