【错题集-编程题】AOE还是单体？（贪心）

牛客对应链接：AOE还是单体？ (nowcoder.com)

一、分析题目

• 如果使用一次 AOE 造成的伤害比消耗的蓝量多，那就使用。
• 否则就一直使用单体伤害。

二、代码

//值得学习的代码
#include <iostream>
#include <algorithm>using namespace std;const int N = 2e5 + 10;int n, x;
int arr[N];int main()
{cin >> n >> x;for(int i = 1; i <= n; i++) cin >> arr[i];sort(arr + 1, arr + 1 + n);long long ret = 0;int index = max(0, n - x); // 处理 x 过⼤的情况ret += arr[index] * x;for(int i = index + 1; i <= n; i++) ret += arr[i] - arr[index];cout << ret << endl;return 0;
}//重新写一遍后的代码
#include <iostream>
#include <algorithm>
using namespace std;typedef long long LL;
const int N=200010;
int a[N];int main()
{int n, x;cin >> n >> x;for(int i=1; i<=n; i++) cin >> a[i];sort(a+1, a+n+1);LL ret=0;int k=max(0, n-x);ret+=a[k]*x;for(int i=k+1; i<=n; i++)ret+=a[i]-a[k];cout << ret << endl;return 0;
}

三、反思与改进

没有弄清楚不同变量之间的关系，应该通过样例来理清楚它们之间的关系，而不是靠着运气来 +1/-1。