刷题记录:牛客NC14670脸盆大哥的木桶

传送门:牛客

彩虹岛网红脸盆大哥最骄傲就是自己制作的木桶。一天𝑙𝑤𝑞拿了𝑛块木板，其中第𝑖块木板的高
度为ℎ𝑖，他希望脸盆大哥能够用这些木板制作出精美的木桶。脸盆大哥告诉𝑙𝑤𝑞制作一个木桶
需要𝑘块木板，并且所有桶的底面积为𝑠，底面的木板由𝑠𝑙𝑝提供。𝑙𝑤𝑞想知道用这些木块所制作
出来的木桶最多能够盛多少体积的水。
注意，木板不能叠在另一个木板上，且不需要考虑木桶具体是怎么由木板组成的，即是说1块
或2块木板也可以组成木桶，底面积仍为𝑠。
输入:
2
4 2 5
1 2 3 4
5 2 5
1 4 5 2 3
输出:
20
30

主要思路:
因为底面积是固定的,并且总的盛水量肯定是由最短的那个木板决定的,所以我们有一个贪心的想法,就是每次都将最短的几个放在一组,这样就能保证答案最优,为什么这样的贪心算法是正确的呢,请看下面证明:

假设我们现在有a,b,c,d这四个木板组合(注意此时我们的abcd四个是一个组合每组都包含很多个木板,只不过每一个木板组合的最短值我们分别设定为a,b,c,d),且我们是可以直接进行排序的,这个对于答案没有任何影响,我们不妨定义a<=b<=c<=d,那么如果此时我们将四个分成两组的话,有a+b,c+d和a+c,b+d,(其中的a+d,b+c的方法和第一种其实是答案是一样的,我们在这里忽略,另一种亦然)两种组法,并且我们显然的会发现前面那一组的答案是最优的,此时我们使用分治的想法去继续推这道题,四个数里面分成两组是这样最优,那么我们的现在的每一个组合都可以分成无数的两个的组合,且abcd中的每一个组都肯定是采用小小大大的配合最优,再递推一下,我们可以使用现在的abcd中两个的组合成的组合(此时abcd分别变成了a+b与c+d的两个组合)继续和其他组合继续使用这种方法组合(假设不止a+b,c+d两个组,当然要继续排序),那么我们就推出来无论木板的数目有几个,采用我们的贪心算法肯定是最优的

即先推出abcd的最优关系,再证明组成abcd的数组成abcd的最优关系,再证明abcd继续和其他组合的最优关系

证明之后我们就很容易实现了

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
int a[maxn];
bool cmp(int a,int b) {
	return a>b;
}
int main() {
	int T;T=read();
	int n,k,s;
	while(T--) {
		memset(a,0,sizeof(a));
		n=read();k=read();s=read();
		for(int i=1;i<=n;i++) {
			a[i]=read();
		}
		int sum=0;
		sort(a+1,a+n+1,cmp);
		for(int i=k;i<=n;i+=k) {
			sum+=s*a[i];
		}
		cout<<sum<<endl;
	}
	return 0;
}