接水的问题

题目描述

学校里有一个水房，水房里一共装有 m 个龙头可供同学们打开水，每个龙头每秒钟的供水量相等，均为 1。
现在有 n 名同学准备接水，他们的初始接水顺序已经确定。将这些同学按接水顺序从 1到 n 编号， i 号同学的接水量为 wi。接水开始时， 1 到 m 号同学各占一个水龙头，并同时打开水龙头接水。当其中某名同学 j 完成其接水量要求 wj 后，下一名排队等候接水的同学 k马上接替 j 同学的位置开始接水。这个换人的过程是瞬间完成的，且没有任何水的浪费。即j 同学第 x 秒结束时完成接水， 则 k 同学第 x+1 秒立刻开始接水。 若当前接水人数 n’不足 m，则只有 n’个龙头供水，其它 m−n’个龙头关闭。
现在给出 n 名同学的接水量，按照上述接水规则，问所有同学都接完水需要多少秒。

输入

第 1 行 2 个整数 n 和 m，用一个空格隔开，分别表示接水人数和龙头个数。
第 2 行 n 个整数 w1、 w2、……、 wn，每两个整数之间用一个空格隔开， wi 表示 i 号同学的接水量。

输出

输出只有一行， 1 个整数，表示接水所需的总时间。

样例输入 Copy

5 3
4 4 1 2 1

样例输出 Copy

4

提示

1 ≤ n ≤ 10000， 1 ≤ m ≤ 100 且 m≤ n；
1 ≤ wi ≤ 100。

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int student[10005];//每个学生的接水量
bool cmp(int a, int b) {
    return a>b;
}
bool cmp(int a,int b);
int main (){
    int n,m,a;
    int sum=0;
    cin>>n>>m;
    a=0;
    for(int i=0;i<n;i++){
        scanf("%d",&student[i]);
    }
    if(m>=n){  //判断是接水的人多还是水龙头多
        sort(student,student+m);
        sum=student[m-1];
    }
    else{

    while(a<n){  
        sort(student,student+m);
        int temp1;
        temp1=student[0];
        for(int i=0;i<m;i++){  //进行一轮接水
            int temp2;
            temp2=student[i];
            student[i]=temp2-temp1;
        }    
        
        if((n-a)>m){       //a用来判断一共有几个人接完水    
            sum+=temp1;    //放到循环外面，防止重复运算
            for(int i=0;i<m;i++){  //判断有多少人接完水了
                if(student[i]==0){
                    a++;
                    int temp3;
                    if((m+a-1)<n){           //防止数组越界
                        temp3=student[m+a-1];
                        student[i]=temp3;
                    }
                    
                }
            }
        }
        
        else {
            sort(student,student+m,cmp); //此时无人在排队接水了，只需知道还在接水人的最大接水量即可
            int temp5;
            temp5=student[0];
            sum=temp5+temp1+sum;
            break;
        }
        
    }
}
    cout<<sum<<endl;
    return 0;
}