算法提高 打水问题

时间限制：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。
　　第一个龙头打水的人总等待时间
　　第二个龙头打水的人总等待时间
　　第三个龙头打水的人总等待时间
　　所以总的等待时间

分析：首先先对整个队伍进行排序
然后再把学生一个一个的插入到那几个队伍中
每一插入一名学生都插入到要求等待时间最小的那个队伍里面
然后计算他要等待的时间
直到所有的学生都插入到队伍中后,统计结束

#include <stdio.h>
#include <algorithm>
using namespace std;
int d[50000];//等待的时间
int x[50000];//接水的人
int main()
{
    int n,r;
    scanf("%d %d",&n,&r);
    int sum=0;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&x[i]);
    }
    sort(x,x+n);
 for(int i=0;i<n;i++)
    {
        sort(d,d+r);
        sum+=(d[0]);
        d[0]+=x[i];
    }
    printf("%d",sum);
}

但是, 


好像不用每一次都排序来寻找要插入到那个队列


因为在这个问题上


学生的插入是有顺序的并且是按照从第一个到最后一个再从第一个开始的顺序插入的


所以可以把每一次插入的排序给去掉

#include<stdio.h>
#include<algorithm>
using namespace std;
int n;
int m;
int a[10000];
int sum;
int main()
{
  scanf("%d %d",&n,&m);
  for(int i=0;i<n;i++)
  {
    scanf("%d",&a[i]);
  }
  sort(a,a+n);
  for(int i=0;i<m;i++)
  {
    int temp=a[i];
    for(int j=i+m;j<n;j+=m)
    {
      sum+=(temp);
      temp+=a[j];
    }
  }
  printf("%d",sum);
}


算法提高 打水问题_等待时间