字节跳动2019春招试题七——机器人跳跃

字节跳动2019春招试题七——机器人跳跃

题目来自牛客网，感谢！

现在有编号为0到N的N+1座建筑，编号为0的建筑高度为0个单位，编号为i的建筑的高度为H_i个单位。
起初，机器人在编号为0的建筑处，每一步跳到下一个建筑。假设机器人在k号建筑，且它当前能量值是E，下一步它将跳到k+1号建筑。它将按H_k+1与E之差得到或者失去能量。如果H_k+1>E那么机器人就失去H_k+1-E的能量值，否则它将得到E-H_k+1的能量值。
机器人的目标是到达N号建筑，在这个过程中，能量值不能为负数。请问机器人出发的最少能量。

输入格式：

第一行输入N，代表有N组数据
第二行输入N个以空格分隔的整数，H1、H2、H3、…、Hn分别代表对应序号建筑的高度

输出格式：

第一行输出完成游戏所需的最少单位的初始能量

输入范例：

3
1 6 4

输出范例：

3

这道题在代码上没有难度，难度主要在数学上。
根据建筑k到建筑k+1跳跃过程中的能量改变规则，可以得到
$E_{k+1}=2E_k-H_{k+1}$。由$E_0=E$及$E_{k+1}\ge 0$，得到
$E_{k+1}=2(2(\dots (2E-H_1)\dots -H_k)-H_{k+1}\ge 0$
简化得到
$E_{k+1}=2^{k+1}E-(2^k{H}_1+2^{k-1}{H}_2+\dots +h_{k+1})\ge 0$
从而
$E\ge \frac{2^k{H}_1+2^{k-1}{H}_2+\dots +h_{k+1}}{2^{k+1}}={\sum }_{i=1}^{k+1}{2}^{-i}{H}_i$
由于E为一个整数，可以得到
$E=\lceil {\sum }_{i=1}^{k+1}{2}^{-i}{H}_i\rceil$

我的代码实现：

import java.util.Scanner;

public class RobotJump {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		double e = 0;
		for (int i = 0; i < n; i++)
			e += (double) sc.nextInt() * Math.pow(2, -i - 1);
		sc.close();
		System.out.println((int) Math.ceil(e));
	}
}