蓝桥杯 算法提高 打水问题 逻辑策略 贪心

算法提高 打水问题

时间限制：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

http://lx.lanqiao.cn/problem.page?gpid=T224

贪心的策略。

为了让等待时间最短，所以让小的先去接水

为了让总体等待时间最短，所以让小的去等待时间小的地方接水

这样 下次在这个水龙头的接水等待时间最短

#include <iostream>
#include <string>
#include <cstring>
#include <stdio.h>
#include <cmath>
#include <algorithm>
using namespace std;
int a[10000];
int d[10000];
int main()
{
int n,m;
while(cin>>n>>m)
{
memset(d,0,sizeof(d));
memset(a,0,sizeof(a));
for(int i=0;i<n;i++)
{
cin>>a[i];
}
sort(a,a+n);//排序 优先让小的接水

int sum=0;
for(int i=0;i<n;i++)
{
sort(d,d+m);//排序，让小的去等待时间最短的地方接水
sum+=d[0];
d[0]+=a[i];//将最小的那个水龙头加上等待时间
}
cout<<sum<<endl;
}

}