洛谷第三场选拔赛.B 最大值和

题目链接

思路

先用vis数组标记每一个数所在的位置，再得到1一直加到n的和，看sum是否小于等于n-1，若是，则直接将a中最大的值（n）和（n-sum）交换位置，然后按次序输出a；若不是，则定义l=1，r=n，当sum>0时，看sum是否大于等于r-l，是则sum-=（r-l），并交换a中数r和l的位置，否则让r–，最后顺序输出a。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int mod = 998244353;
 
ll a[101000];
map<int,int>vis;
int main()
{
	int n;
	scanf("%d",&n);
	ll sum;
	scanf("%lld",&sum);
	ll k = 0;
	for(int i = 1; i <= n; i++)
	{
		scanf("%lld",&a[i]);
		k += a[i];
		vis[a[i]] = i;
	}
	sum -= k;
	if(sum <= n-1)
	{
		int f1,f2;
		for(int i = 1; i <= n; i++)
		{
			if(a[i] == n)
			{
				f1 = i;
			}
			if(a[i] == (n - sum))
			{
				f2 = i;
			}
		}
		int l = a[f1];
		a[f1] = a[f2];
		a[f2] = l;
		for(int i = 1; i <= n; i++)
		printf("%lld ",a[i]);
	}
	else
	{
		int l = 1,r = n;
		while(sum>0)
		{
			if(sum >= r - l)
			{
				int p = a[vis[r]];
				a[vis[r]] = a[vis[l]];
				a[vis[l]] = p;
				sum -= (r-l);
				r--,l++;
			}
			else
			{
				r--;
			}
		}
		for(int i = 1; i <= n; i++)
		printf("%lld ",a[i]);
		printf("\n");
	}
}