思路
先用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");
}
}