问题:由于楼层并不高,在繁忙时段,每次电梯从一层往上走时,我们只允许停到其中某一层。所有乘客再从这层去往自己的目的楼层。
电梯停在哪一层,能够保证乘客爬楼梯之和最少?
import java.util.Scanner;
public class BeautyPro18 {
/** * @param args */
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
int n = scanner.nextInt();
int[] nPerson = new int[n];
for (int i = 0; i < n; i++) {
nPerson[i] = scanner.nextInt();
}
System.out.println(getStayFloor(nPerson, n));
}
}
// nPerson[i]表示到第i层的乘客数目, n表示楼层
public static int getStayFloor(int[] nPerson, int n) {
int minStep = Integer.MAX_VALUE;
int targetFloor = -1;
for (int i = 0; i < n; i++) {
int stepNum = 0; // 记录所需的步数
for (int j = 0; j < i; j++) {
// 停留楼层上方所需的步数
stepNum += nPerson[j] * (i - j);
}
for (int j = i; j < n; j++) {
// 停留楼层下方所需的步数
stepNum += nPerson[j] * (j - i);
}
if (targetFloor == -1 || minStep > stepNum) {
minStep = stepNum;
targetFloor = i;
}
}
return targetFloor + 1; // 停留楼层
}
// 方法2:假设当前楼梯停在第i层。N1代表在停留楼层以下的乘客数,N2表示在停留楼层的乘客,N3表示在停留楼层上方的乘客
// 当N1 > N2+N3 电梯停在i-1层较好 当N1 + N2 < N3 ,停在i+1层比较好。其他情况下,停在第i层
public static int getStayFloor1(int[] nPerson, int n) {
int N1 = 0;
int N2 = nPerson[0];
int N3 = 0;
int minStep = 0;
int targetFloor = 0;
// 初始化停留在第一层时,N3和所需步数minStep
for (int i = 1; i < n; i++) {
N3 += nPerson[i];
minStep += nPerson[i] * i;
}
for (int i = 1; i < n; i++) {
if (N1 + N2 < N3) {
targetFloor = i;
minStep += (N1 + N2 - N3);
N1 += N2;
N2 = nPerson[i];
N3 -= nPerson[i];
} else {
break;
}
}
return targetFloor + 1;
}
}