华为机试 - 跳房子I

华为机试 - 跳房子I

题目描述

跳房子，也叫跳飞机，是一种世界性的儿童游戏。游戏参与者需要分多个回合按顺序跳到第1格直到房子的最后一格。跳房子的过程中，可以向前跳，也可以向后跳。假设房子的总格数是count，小红每回合可能连续跳的步数都放在数组steps中，请问数组中是否有一种步数的组合，可以让小红两个回合跳到量后一格？如果有，请输出索引和最小的步数组合。

注意：

• 数组中的步数可以重复，但数组中的元素不能重复使用。
• 提供的数据保证存在满足题目要求的组合，且索引和最小的步数组合是唯一的。

输入描述

第一行输入为每回合可能连续跳的步数，它是 int 整数数组类型。

第二行输入为房子总格数count，它是 int 整数类型。

• count ≤ 1000
• 0 ≤ steps.length ≤ 5000
• -100000000 ≤ steps ≤ 100000000

输出描述

返回索引和最小的满足要求的步数组合（顺序保持steps中原有顺序）

示例1

输入

[1,4,5,2,2]
7

输出

[5, 2]

示例2

输入

[-1,2,4,9,6]
8

输出

[-1, 9]

解题思路

本题实际转化为在数组中找到两个数字，使得数字之和等于目标值，也就是 两数之和 等于目标值，输出要求为满足索引和最小。

暴力求解

public class JumpStep {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int[] steps = Arrays.stream(
                in.nextLine().replaceAll("[^\\-0-9,]", "").split(",")
        ).mapToInt(Integer::parseInt).toArray();
        int aimStep = in.nextInt();
        // 维护一个最小索引和用作判断
        int minIndexSum = steps.length * 2;
        // 维护输出结果
        int firstNum = 0, secondNum = 0;
        // 暴力遍历所有组合
        for (int i = 0; i < steps.length; i++) {
            for (int j = i + 1; j < steps.length; j++) {
                // 满足两数之和等于目标值
                if(steps[i] + steps[j] == aimStep) {
                    if(i + j < minIndexSum) {
                        minIndexSum = i + j;
                        firstNum = steps[i];
                        secondNum = steps[j];
                    }
                }
            }
        }
        System.out.println(
                minIndexSum == steps.length * 2 ? "" : ("[" + firstNum + "," + secondNum + "]")
        );
    }
}

哈希表

public class JumpStep {

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int[] steps = Arrays.stream(
                in.nextLine().replaceAll("[^\\-0-9,]", "").split(",")
        ).mapToInt(Integer::parseInt).toArray();
        int aimStep = in.nextInt();
        // 维护一个最小索引和用作判断
        int minIndexSum = steps.length * 2;
        // 维护输出结果
        int firstNum = 0, secondNum = 0;
        // 存储数字与索引
        Map<Integer, Integer> numIndexMap = new HashMap<>();
        for (int i = 0; i < steps.length; i++) {
            Integer otherIndex = numIndexMap.get(aimStep - steps[i]);
            // 如果在map中找到了目标值（aimStep - steps[i]），说明当前的 steps[i] + steps[otherIndex] = aimStep
            if(null != otherIndex) {
                // 如果索引和小于前面符合条件的索引和，则更新结果
                if(i + otherIndex < minIndexSum) {
                    minIndexSum = i + otherIndex;
                    firstNum = steps[otherIndex];
                    secondNum = steps[i];
                }
            }
            // 此处需要考虑情况，数组中存在重复数值，比如说 2, 3, 2, 4, 3 aim=6
            // 咱们不能在遍历时直接 put，这样相同数值的下标将会被更新
            // 咱们需要计算最小索引和，那么对于重复数值，只需要维护最小的索引即可
            if(!numIndexMap.containsKey(steps[i])) {
                numIndexMap.put(steps[i], i);
            }
        }
        System.out.println(
            minIndexSum == steps.length * 2 ? "" : ("[" + firstNum + "," + secondNum + "]")
        );
    }
}