AcWing 每日一题 3664. 数组补全（思维）

原题链接

数据范围
1≤n≤999，n 是奇数。
0≤k<n,
1≤p≤1000,
n≤x≤np,
1≤y≤p
输入样例1：
5 3 5 18 4
3 5 4
输出样例1：
4 1
输入样例2：
5 3 5 16 4
5 5 5
输出样例2：
-1

---

在一个数组中，大于等与中位数的数的个数一定大于等于这个数组大小的一半

我们在输入是计算一下有多少个数大于等于y（即为cnt）
对于不够一半的，我们用y来补
够了之后只要在前面补上若干个1就行

AC代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1007;
int a[maxn],ans[maxn],tot;
int main(){
    int n,k,p,sum,mid;
    scanf("%d %d %d %d %d",&n,&k,&p,&sum,&mid);
    int cnt=0;
    for(int i=1;i<=k;++i){
    	scanf("%d",&a[i]);
    	sum-=a[i];
    	if(a[i]>=mid)cnt++;
	}
	while(cnt<=n/2)ans[++tot]=mid,sum-=mid,cnt++;
	while(k+tot<n)ans[++tot]=1,sum--; 
	if(sum<0)printf("-1\n");
	else {
		for(int i=1;i<=tot;++i)
			printf("%d ",ans[i]);
		printf("\n");
		tot=0;
	}
    return 0;
}