西电网络赛 - G

#include <iostream>
#include <algorithm>
#include <string.h>
using namespace std;
/****************************************************************************************************************
画外音（学神满早上说用贪心，然后想了一天没想出来，晚上打球回来 WC 的时候竟然想出来了）
题意：给定 n 个文件，每次最多合并 k 个文件，输出最小的文件总数
思路：
1，贪贪贪，数值大的文件合并次数尽可能少，尽量多合并数值小的文件。
2，每次合并完之后，将新合并的文件放入原文件堆，重新排序，再次处理

eg：
输入： 7  4
1 2 3 4 5 6 7 8 9 10 11
过程：
1 2 ）3 4 5 ）6 7 8 ）9 10 11    <合并 1 2，花费 1+2=3>
3 3 4 5 ）6 7 8 ）9 10 11        <合并 3 3 4 5，花费 3+3+4+5=15>
6 7 8 9 ）10 11 15               <合并 6 7 8 9，花费 6+7+8+9=30>
10 11 15 30                      <合并 10 11 15 30，花费 10+11+15+30=66>
合并结束，总花费 3+15+30+66=114

****************************************************************************************************************/
int main()
{
int N,M;
long long a[2050];
while(cin>>N>>M)
{
memset(a,0,sizeof(a));
long long sum=0;
for(int i = 1;i <= N;i ++)
cin>>a[i];

sort(a+1,a+N+1);

if(N == 1)
cout<<"0"<<endl;
else if(N <= M){
for(int i = 1;i <= N;i ++)
sum+=a[i];
cout<<sum<<endl;
}
else{

int le=N,mod,id=1;

while(le >  0){

mod=le;
while(mod > M)
mod=mod-(M-1);

long long flag=0;
for(int i = id;i <id+mod;i ++)
flag+=a[i];

id=id+mod-1;
a[id]=flag;
sort(a+id,a+N+1);
sum+=flag;

if(le > mod)
le=le-mod+1;
else
le=0;
}
cout<<sum<<endl;
}
}
return 0;
}