题目:一栋楼有6层,现在设计一种电梯调度算法:电梯在一楼让大家上电梯,然后根据大家选择要到的楼层算出某一楼层i,电梯在i层停下让所有人下电梯,然后大家爬楼梯达到自己的楼层。请问电梯停在哪一层,可以使得这一次的所有乘客爬楼层之和最短?
编程之美总共给了两种解答:
- 最简单的想法:遍历每一层,依次求出每一层所有人爬楼总和,总和最小的那一层即是答案,这样的话时间复杂度就是O(N * N)。
- 动态规划法:假设停在i层,有N1个人在i层之下,N2个人在i层,N3个人在i层之上,若电梯停在i + 1层,则爬楼层总和需要增加 N1 + N2 - N3层,若N1 + N2 - N3小于0,则i + 1层爬楼总和更小。所以按照这个思路,时间复杂度为O(N)。
动态规划法只需要一次不满足N1 + N2 - N3<0就可以退出的原因是:假设再上一层,则N1'=N1+N2,N2'=nperson[i+1],N3'=N3-nperson[i+1]。则N1'+N2’-N3'=N1 + N2 - N3+2*nperson[i+1]>=N1 + N2 - N3,以此类推.....因此越上则需要走的层数越多。
代码:
int person[];
int N1,N2,N3;
int minFloor = 0;
int targetFloor = 1;
for(int i = 2; i < n; i++)
{
N3 += person[i];
minFloor = person[i] * (i - 1);
}
for(int i = 2; i < n; i++)
{
if(N1 + N2 < N3)
{
targetFloor = i;
minFloor += (N1 + N2 - N3);
N1 += N2;
N2 = person[i];
N3 -= person[i];
}
else
break;
}
return targetFloor;
}