题解 | CF1328F. Make k Equal (思维,前缀 & 后缀和)
题目链接:Here
题意:把
思路:
我们可以枚举,把
如果相同的数大于
有三种情况,一种是只动前面,一种只动后面,还有就是前后都动
因为是改变最大或最小的数,所以我们只有把所有小于
然后接着考虑,在什么情况下可以动前面呢,当然是他前面的数大于
以只动前面为例
化简一下发现
就是
只动后面同理
动两边,这时相等的数的个数恰好为
记录好前缀和 和(后缀和?)就可以用
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll inf = 1e17;
ll a[200009];
ll cnt[200009];
ll sumq[200009], sumh[200009];
int main() {
int n, k;
ll ans = inf;
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i++) {
scanf("%lld", &a[i]);
}
sort(a + 1, a + 1 + n);
for (int i = 1; i <= n; i++) sumq[i] = sumq[i - 1] + a[i];
for (int i = n; i >= 1; i--) sumh[i] = sumh[i + 1] + a[i];
for (int i = 1; i <= n; i++) {
if (a[i] == a[i - 1])cnt[i] = cnt[i - 1] + 1;
else cnt[i] = 1;
if (cnt[i] >= k) {
puts("0");
return 0;
}
}
for (int i = 1; i <= n; i++) {
if (i >= k) {
ll tem1 = i * (a[i] - 1) - sumq[i] + k;
ans = min(tem1, ans);
}
if (n - i + 1 >= k) {
ll tem2 = n - i + 1;
tem2 = sumh[i] - tem2 * (a[i] + 1) + k;
ans = min(tem2, ans);
}
if (i < k && (n - i + 1) < k) {
ll tem3 = i * a[i] - sumq[i] + sumh[i] - (n - i + 1) * a[i] - (n - k);
ans = min(tem3, ans);
}
}
printf("%lld\n", ans);
return 0;
}
便捷写法
int main() {
cin.tie(nullptr)->sync_with_stdio(false);
int n, k;
cin >> n >> k;
vector<ll> a(n);
for (ll &x : a) cin >> x;
sort(a.begin(), a.end());
for (int i = 0; i + k - 1 < n; ++i) {
if (a[i] == a[i + k - 1])
return printf("0\n"), 0;
}
ll lcost = 0;
ll rcost = 0;
for (int i = 0; i < k; ++i) {
lcost += a[k - 1] - a[i];
rcost += a[n - 1 - i] - a[n - k];
}
for (int j = k; j < n; ++j) {
if (a[k - 1] == a[j]) lcost--;
if (a[n - k] == a[n - 1 - j]) rcost--;
}
ll sum = 0;
for (int i = 0; i < n - 1 - i; ++i) sum += a[n - 1 - i] - a[i];
ll ans = min(sum - (n - k), min(lcost, rcost));
cout << ans;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 上周热点回顾(3.3-3.9)
2020-08-18 Problem 550A - Two Substrings
2020-08-18 # 0x56 动态规划-状态压缩DP
2020-08-18 git - 简易指南