蓝桥杯算法提高 打水问题

原题：

  算法提高 打水问题  

时间限制：1.0s   内存限制：512.0MB
    

问题描述

　　N个人要打水，有M个水龙头，第i个人打水所需时间为Ti，请安排一个合理的方案使得所有人的等待时间之和尽量小。

输入格式

　　第一行两个正整数N M 接下来一行N个正整数Ti。

　　N,M<=1000，Ti<=1000

输出格式

　　最小的等待时间之和。（不需要输出具体的安排方案）

样例输入

7 3

3 6 1 4 2 5 7

样例输出

11

提示

　　一种最佳打水方案是，将N个人按照Ti从小到大的顺序依次分配到M个龙头打水。

　　例如样例中，Ti从小到大排序为1，2，3，4，5，6，7，将他们依次分配到3个龙头，则去龙头一打水的为1，4，7；去龙头二打水的为2,5；去第三个龙头打水的为3,6。

　　第一个龙头打水的人总等待时间 = 0 + 1 + (1 + 4) = 6

　　第二个龙头打水的人总等待时间 = 0 + 2 = 2

　　第三个龙头打水的人总等待时间 = 0 + 3 = 3

　　所以总的等待时间 = 6 + 2 + 3 = 11

分析：

使用num
数组记住输入，然后进行排序，n.m次循环中，进行m次循环，使得sum+=num[..],并使num[..]加到后面对应的num[..+m]上去。注意边界问题

代码:

#include <algorithm>
#include <iostream>
using namespace std;
int main()
{
int n,m;
cin>>n>>m;
int num
,sum = 0;
for(int i = 0;i < n;i++)
cin>>num[i];
sort(num,num + n);
for(int i = 0;i < n;i += m)
{
for(int j = 0;j < m;j++)
{
if(i + j >= n)
break;
if(i + j + m >= n)
continue;
sum += num[i + j];
num[i + j + m] += num[i + j];
}
}
cout<<sum;
return 0;
}