题目大意:给你n * k 个木板,让你组成有n个木桶,每个木桶有k个木板。每个木桶的体积
是这个木桶的木板中最短的那个。并且任意两个木桶的体积的差必须<=l。
求如何组装才能使n个木桶的体积和最大。输出体积和。
解题思路:
首先我们先将所有的木板从小到大分成n个块。如果我们想要实现体积最大化,我们需要选出
每个块最短的那个木板作为由本块木板组成得木桶的体积。
但是题目上限制了任何两个木桶的体积的差必须小于l,所以如果本块最短的那个木板如果不符合
条件就往前面找到符合条件且最长的那个木板作为本块木桶的体积即可。
代码如下:
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#define mmset(a,b) memset(a,b,sizeof(a))
#define ll long long
using namespace std;
const int N = 1e5 + 5;
int data[N];
int mark[N] = {0};
/*
4 2 1
2 2 1 2 3 2 2 3
outputCopy
7
inputCopy
2 1 0
10 10
outputCopy
20
inputCopy
1 2 1
5 2
outputCopy
2
inputCopy
3 2 1
1 2 3 4 5 6
outputCopy
0
*/
int main()
{
int n,k,m,flag = 0;
scanf("%d %d %d",&n, &k, &m);
for(int i = 1; i <= n * k; i++)
{
scanf("%d",&data[i]);
}
sort(data + 1, data + 1 + n * k);
ll sum = 0;
int base = data[1];
if(data[n]-data[1] > m)
{
printf("0\n");
return 0;
}
for(int i = 0; i <= (n - 1); i++)
{
int j = i * k + 1;
if(data[j] > base + m)
{
for(; j >= 1; j--)
{
if(mark[j] == 0 && data[j] <= base + m)
{
mark[j] = 1;
sum += data[j];
break;
}
}
}
else
{
mark[j] = 1;
sum += data[j];
}
}
if(flag == 1)
{
printf("0\n");
}
else
{
printf("%lld\n",sum);
}
return 0;
}