Codeforces540B School Marks (思维 贪心 构造)

题目链接: School Marks

大致题意

有一个长度为$n$的数组（$n$为奇数）$a_1,a_2,...,a_n$

由于某些原因，数组丢失了一部分数，现在只剩下了前$k$个数 $a_1,a_2,...,a_k$

请你补全丢失的$n-k$个数，并使得补全后的数组能够满足以下性质：

1. 数组中的元素的取值范围为$[1,p]$。
2. 数组中所有元素的和不大于$x$
3. 数组中的中位数的值不小于$y$

如果补全方法不唯一，则任意合理方案均可。

解题思路

思维

我们考虑最终要求中位数不小于$y$, 等价于最终数组有$num=(n+1)/2$个数字是大于等于$y$的.

由于最终数组元素总和不能大于$x$, 因此如果数组内不够$num$个数组大于等于$y$, 则我们都填充$y$即可.

剩余$n-num$个数字应该是小于$y$的, 我们同理填充$1$即可.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 1234;
int a[N];
int main()
{
    int n, k, p, x, y; cin >> n >> k >> p >> x >> y;
    int all = 0, big = 0;
    rep(i, k) scanf("%d", &a[i]), all += a[i], big += a[i] >= y;

    vector<int> res;
    int mid = (n + 1) / 2;
	while (big < mid) {
		big++, all += y;
		res.push_back(y);
	}
	while (res.size() < n - k) res.push_back(1), all++;

	if (res.size() > n - k or all > x) puts("-1");
	else {
		for (auto& op : res) printf("%d ", op); puts("");
	}

    return 0;
}

END