比赛 2025.4.12 评测结果 AATAAATTTTTTTTTTTTTT
题目名称 OohMoo Milk 最终得分 25
用户昵称 LikableP 运行时间 30.088 s
代码语言 C++ 内存使用 1.92 MiB
提交时间 2025-04-12 08:56:35
显示代码纯文本
#include <cstdio>
#include <algorithm>
typedef long long ll;
using namespace std;

const int MAXN = 1e5 + 10;
const int MOD = 1e9 + 7;

int N, D;
int A, B;
ll m[MAXN];
ll ans;

int main() {
	freopen("Milk.in", "r", stdin);
	freopen("Milk.out", "w", stdout);
	scanf("%d %d", &N, &D);
	scanf("%d %d", &A, &B);
	for (int i = 1; i <= N; ++i) {
		scanf("%lld", &m[i]);
	}

	while (D--) {
		partial_sort(m + 1, m + A + 1, m + N + 1, [](int x, int y){return x > y;});
		for (int i = B + 1; i <= A; ++i) {
			m[i] += 1;
		}
	}
	
	for (int i = 1; i <= N; ++i) {
		(ans += m[i] % MOD * m[i] % MOD) %= MOD;
	}
	printf("%lld", ans);
	return 0;
}