牛客小白月赛25 AOE还是单体？

链接：https://ac.nowcoder.com/acm/contest/5600/A
来源：牛客网

题目描述
牛可乐准备和 个怪物厮杀。已知第 个怪物的血量为 aia_iai​ 。
牛可乐有两个技能：
第一个技能是蛮牛冲撞，消耗 ，可以对任意单体怪物造成 点伤害。
第二个技能是蛮牛践踏，消耗 ，可以对全体怪物造成 点伤害。
牛可乐想知道，将这些怪物全部击杀，消耗 的最小值的多少？
输入描述:

第一行两个正整数 和 ，分别代表怪物的数量、每次蛮牛践踏消耗的 值。
第二行 个正整数 aia_iai​ ，分别代表每个怪物的血量。

输出描述:

一个正整数，代表消耗 的最小值。

示例1
输入
复制

5 2
2 4 5 6 3

输出
复制

11

说明

先对3号怪物用1次蛮牛冲撞，对4号怪物用2次蛮牛冲撞，此时消耗mp为3，怪物的血量是2，4，4，4，3。
然后用4次蛮牛践踏，击杀全部怪物，消耗的mp总量为11。

显然题 :

• 多次践踏后非零个数会小于践踏需要的mp

• 当非0个数大于蛮牛践踏消耗的mp$(即cn{t}_{非零}>x)$时,使用践踏更优

• 当非0个数小于x时,使用对每个怪物冲撞更优
  \

大到小排个序列
所以,只需要找前$x$大的值,多次践踏把$[x,n]$大消耗成0,
剩下的怪物全部用冲撞,非0数字累加即可

#define debug
#ifdef debug
#include <time.h>
#include "/home/majiao/mb.h"
#endif


#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <math.h>
#define MAXN ((int)2e5+7)
#define ll long long int
#define int long long int
#define INF (0x7f7f7f7f)
#define QAQ (0)

using namespace std;

#ifdef debug
#define show(x...) \
do { \
	cout << "\033[31;1m " << #x << " -> "; \
	err(x); \
} while (0)
void err() { cout << "\033[39;0m" << endl; }
#endif

template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }

int n, m, Q, K, a[MAXN];

signed main() {
#ifdef debug
	freopen("test", "r", stdin);
	clock_t stime = clock();
#endif
	//非0个数大于等于x就用践踏
	//如果x太大,就不用
	//找前x大的数
	//6-3, 5-3  mp=3*2 + 3+2 = 11
	scanf("%lld %lld ", &n, &K);
	int sum = 0;找前x大的数

	for(int i=1; i<=n; i++) 
		scanf("%lld ", a+i), sum += a[i];
	if(K > n) { //注意K>n的情况,用践踏不划算,所以全用冲撞
		printf("%lld\n", sum);
		return 0;
	}
	sort(a+1, a+1+n, greater<int>());
	int kth = a[K];
	int ans = K * kth;
	for(int i=1; i<K; i++)
		ans += (a[i] - kth);
	printf("%lld\n", ans);

#ifdef debug
	clock_t etime = clock();
	printf("rum time: %lf 秒\n",(double) (etime-stime)/CLOCKS_PER_SEC);
#endif 
	return 0;
}