原题信息
资源限制
时间限制:1.0s 内存限制:64.0MB
问题描述
学校里有一个水房,水房里一共装有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 个整数,表示接水所需的总时间。
样例输入
5 3
4 4 1 2 1
样例输出
4
思路分析
首先看了一下题,第一反应暴力解决,因为看上去数据不大。
我就直接一个一个找,接到水的w[i]就减一,然后t++;
但是尝试了几次,都是九十分,最后那个一万的数据还是过不了,于是认真思考了一下。
这里注意它接水的人员是有顺序的,并不是我们可以控制的,所以我们得到的结果其实并不是理论最短的时间。
暴力不行,算法哪里可以优化呢?
其实可以换个角度,建立一个水龙头的数组,存储接水人的序号,大多数情况下,接水的人是满的。
那么我们不用一个一个减它接水量,只要求出他们里的最小值就行了,然后一个for循环一次减一堆,效率就能大大提高,时间复杂度显著下降。
后续的处理,其实很简单,接完水的水头龙用一个变量存储当前排队的人的序号,让这个人去接水就可以了。接下来就是,出现水龙头没人接的情况时,就不用再计算了,直接找一个最大的接水量加在时间上就可以了。
接下来是参考代码:
#include <iostream>
#include <algorithm>
using namespace std;
int main()
{
int n, m,t=0,q,ram;
cin >> n >> m;
static int w[10005],s[105];
for (int i = 0; i < n; i++)
{
cin >> w[i];
}
for (int i = 0; i <m; i++)
{
s[i] = i;
}
q = m;
int k = n,l;
while (1)
{
l = 0;
ram = 500;
for (int i = 0; i < m; i++)
{
if (w[s[i]]<ram)
{
ram = w[s[i]];
}
}
if (ram != 0)
{
for (int i = 0; i < m; i++)
{
w[s[i]] -= ram;
if (w[s[i]] == 0)
{
k--;
if (q < n)
{
s[i] = q++;
}
}
}
t += ram;
}
else
{
for (int i = 0; i < m; i++)
{
if (w[s[i]] > ram)
{
ram = w[s[i]];
}
}
t += ram;
break;
}
}
cout << t;
return 0;
}