传送门
先看看代码。
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int n, k, a[500002], cnt;
int main() {
scanf("%d %d", &n, &k);
for(int i = 1; i <= n; i ++)
scanf("%d", &a[i]);
if(k == 0)
return ! printf("-1\n");
sort(a + 1, a + n + 1);
for(int i = n - k + 1; i <= n - k + (k >> 1); i ++)
swap(a[i], a[n - (i - n + k) + 1]);
for(int i = n; i >= 1; i --)
if(a[i] > cnt)
cnt ++;
if(cnt == k)
for(int i = 1; i <= n; i ++)
printf("%d ", a[i]);
else
printf("-1\n");
return 0;
}
开始解释方法的正确性。
首先,越排在前面就越可能是对的;a[]值越大,错误的可能性越大。因此将a数组从小到大排序,再将后面k个翻转,使后面k个错误的可能性越大。
可翻转不一定恰好有k个,故需要验证(注意:前面也有可能是错的)。
为什么这样排序?
1.个数<k 前面的数一定比后面k个数小,既然后面都不满足条件,换了位置就更不可能了。
2.个数>k 往后排,错误的结论是错的。往前排,可以证明后k个一定是错的,交换的根源来自于中间有错误的,如果满足,则还是大于k。
如有错误,请大佬在评论区指出。