题目大意:给定一个大小为n的整数数组,经过m次操作使得他们的和最小。
每次操作为:找到一个非负的整数Ai,并记录下这个数的位置k,从当前位置开始到最后一个数每一个数都减去Ai。
我的思路:直接看第二组例子,数组中有三个非负的整数并且最多可以操作3次,那么要使数组和最小,我们可以先找到最大的数9,然后每个数都减9,那么和就少了3 * 9,但时0变成了负,最后还剩的非负的数就只有4了,和再减去4 * 4。可以刚好看到此时结果就是答案。
但是现在面临两个问题就是每次先从大的数往后减就一定结果最小吗?第二个为什么不从最前面的非负整数开始减呢?
经过简单分析就可以知道每个数之间其实没什么影响,所以从最大数往后减一定最小。可以看出来,先操作前面的只会产生负面影响(因为先操作前面的只会让后面的 Ai 更小,则和更小。
那么我们可以记录每个数对和造成的影响为 ,那么只需要找出最大v的 m 个正整数,将总和减去就行。
代码:
#include<bits/stdc++.h>;
#define int long long
using namespace std;
const int N = 1e5 + 5;
int n, m;
int a[N];
void solve()
{
memset(a, 0, sizeof a);
cin >> n >> m;
vector<int> v;
int ans = 0;
for(int i = 1; i <= n; i++){
cin >> a[i];
ans += a[i];
if(a[i] >= 0){
v.push_back((n - i + 1) * a[i]);
}
}
sort(v.begin(), v.end(), greater<int>());
for(int i = 0; i < min(m, (int)v.size()); i++){
ans -= v[i];
}
cout << ans << endl;
}
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
int t = 1;
cin >> t;
while (t--) {
solve();
}
return 0;
}
好像写的有点多了,看起来有点乱。。。