package leetcodejava.array;

import org.junit.Assert;
import org.junit.Test;

/**
 * This is the solution of No. 45 problem in the LeetCode,
 * the website of the problem is as follow:
 * https://leetcode-cn.com/problems/jump-game-ii
 * <p>
 * The description of problem is as follow:
 * ==========================================================================================================
 * 给定一个非负整数数组，你最初位于数组的第一个位置。
 * 数组中的每个元素代表你在该位置可以跳跃的最大长度。
 * 你的目标是使用最少的跳跃次数到达数组的最后一个位置。
 * <p>
 * 示例:
 * <p>
 * 输入: [2,3,1,1,4]
 * 输出: 2
 * 解释: 跳到最后一个位置的最小跳跃数是 2。
 *      从下标为 0 跳到下标为 1 的位置，跳 1 步，然后跳 3 步到达数组的最后一个位置。
 * 说明:
 * 假设你总是可以到达数组的最后一个位置。
 * <p>
 * 来源：力扣（LeetCode）
 * ==========================================================================================================
 *
 * @author zhangyu (zhangyuyu417@gmail.com)
 */
public class JumpGame45 {

    @Test
    public void jumpGameTest() {
        int[] nums = {2, 3, 1, 1, 4};
        int result = jump(nums);
        System.out.println(result);
        Assert.assertEquals(result, 2);
    }

    /**
     * 跳到数组最后
     *
     * @param nums 数组
     * @return 步数
     */
    public int jump(int[] nums) {
        int length = nums.length;
        int end = 0;   // 这一跳截止点
        int maxPosition = 0;  // 下一跳最远距离
        int result = 0;
        for (int i = 0; i < length - 1; i++) {
            maxPosition = Math.max(maxPosition, i + nums[i]);
            if (i == end) {
                end = maxPosition;
                result++;
            }
        }
        return result;
    }
    
      /**
     * 跳到数组最后
     *
     * @param nums 数组
     * @return 步数
     */
    public int jump2(int[] nums) {
        if( nums == null || nums.length <= 1) {
            return 0;
        }
        int reach = 0;
        int nextReach = nums[0];
        int step = 0;
        for(int i = 0; i < nums.length; i++) {
            nextReach = Math.max(i + nums[i], nextReach);
            if( nextReach >= nums.length - 1) {
                return step + 1;
            }
            // 现在能达到的点
            if (i == reach) {
                step++;
                // 下一次能达到的点
                reach = nextReach;
            }
        }
        return step;
    }
}
