题目链接:https://ac.nowcoder.com/acm/contest/57322/F
题目描述:
凡凡有一个好用的序列自动机,可以帮他快速改变一个序列。当给定一个序列 AAA 时,凡凡每次可以使用自动机的以下任意两种操作之一来改变当前序列,操作次数不限。
①:选择序列中任意一个数,将
的大小减 1。
②:选择序列中任意一个数,将除去
以外其他所有数的大小减 1。
请问为使该序列中所有数字的大小相等,凡凡至少需要使用多少次他的自动机。
________________________________________________________
解法一
操作① 代价为 1
操作② 令 x 为当前最小数的个数 就要进行x步 ,需要修改x + 1 ~ n,代价为(n - (x + 1) + 1) / x
联立方程 1 >= (n - x) / x = > x >= (n + 1) / 2 在 下标大于(n + 1) / 2时选择用操作①
关于对“x 为当前最小数的个数 就要进行x步”举例
0 0 0 3 5 ,此时最小数的个数为3
若执行操作① 用1的代价修改了元素值1
若执行操作②0 0 0 3 5->0 -1 -1 2 4 ->-1 -1 -2 1 3->-2 -2 -2 0 2 相当于让后两个数各减了1,但是执行了三步,代价为 ,比1小,故选择使用操作①
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define qwq ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
const int N = 200010, M = 200010, INF = 0x3f3f3f3f;
int n, m, k;
int a[N];
int main(){
qwq; int T;
cin >> n;
for (int i = 1; i <= n; ++ i)
cin >> a[i];
sort(a + 1, a + 1 + n);
//操作① 代价为 1
//操作② 令 x 为当前最小数的个数 需要修改x + 1 ~ n 代价为(n - (x + 1) + 1) / x
//方程 1 >= (n - x) / x => x >= (n + 1) / 2 在 下标大于n / 2时选择用操作①
ll ans = 0;
for (int i = 2; i <= n; ++ i) {
//a[i] - a[i - 1] 是将 a[i] 转为 a[i - 1] 需要的步数
if (i <= n / 2) ans += (a[i] - a[i - 1]) * (i - 1); // i - 1 为已有的最小值的数量
else ans += (a[i] - a[i - 1]), a[i] = a[i - 1]; // 执行操作①时要更新a[i]
}
cout << ans;
return 0;
}
——————————————————————
解法二
将操作②看作是将该点值+1
依次枚举终点,记录最小花费次数
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define qwq ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
const int N = 200010, M = 200010, INF = 0x3f3f3f3f;
int n, m, k;
int cnt[N]; // 记录每种值出现的次数
int main(){
qwq; int T;
cin >> n;
ll sum, ans;
for (int i = 1, x; i <= n; ++ i) {
cin >> x; sum += x; cnt[x] ++;
}
ans = sum; //初始令所有点移动到0,记录花费次数
int l = 0, r = n;
// l是小于等于终点的点的数量,r是大于终点的点的数量
for (int i = 0; i <= 2e5; ++ i) {
// 依次让终点上移 更新 l 和 r
l += cnt[i], r -= cnt[i];
//终点每次+1,使得所有小于等于l的点花费次数+1,所有大于r的数字距离-1
sum += l, sum -= r;
ans = min(sum, ans);// 每次更新最小花费
}
cout << ans;
return 0;
}