题目描述与示例
题目描述
跳房子，也叫跳飞机，是一种世界性的儿童游戏。 游戏参与者需要分多个回合按顺序跳到第1格直到房子的最后一格
跳房子的过程中，可以向前跳，也可以向后跳。
假设房子的总格数是count，小红每回合可能连续跳的步教都放在数组steps中，请问数组中是否有一种步数的组合，可以让小红两个回合跳到最后一格? 如果有，请输出索引和最小的步数组合。
注意:
- 数组中的步数可以重复，但数组中的元素不能重复使用。
- 提供的数据保证存在满足题目要求的组合，且索引和最小的步数组合是唯一的。
输入描述
第一行输入为每回合可能连续跳的步数，它是整数数组类型。
第二行输入为房子总格数count，它是int整数类型。
输出描述
返回索引和最小的满足要求的步数组合(顺序保持steps中原有顺序)
备注
count ≤ 10000 ≤ steps.length ≤ 5000-100000000 ≤ steps ≤ 100000000
示例一
输入
[1,4,5,2]
7
输出
[5,2]
说明
示例二
输入
[-1,2,4,9,6]
8
输出
[-1,9]
解题思路
注意，本题和LeetCode1、两数之和几乎完全一致。区别在于需要输出索引和最小的两个数字。
代码
Python
# 题目：2023B-跳房子I
# 分值：200
# 作者：许老师-闭着眼睛学数理化
# 算法：哈希表
# 代码看不懂的地方，请直接在群上提问
# 输入步数列表，注意需要去除最开头和最末尾的中括号
nums = list(map(int, input()[1:-1].split(",")))
target = int(input())
# 初始化索引和最大值，这里可以取inf，也可以取nums长度乘2
min_idx_sum = len(nums)*2
# 构建哈希表，储存每一种步数首次出现的下标
hash_dic = dict()
# 初始化答案列表
ans = [0, 0]
# 遍历nums
for i, num in enumerate(nums):
# 计算target-num的值rest_num
rest_num = target-num
# 若rest_num位于哈希表中，说明其曾经出现过
# rest_num和num相加等于所需要的结果target
if rest_num in hash_dic:
# 若此时min_idx_sum大于两者下标和
# 则更新min_idx_sum和ans
if min_idx_sum > hash_dic[rest_num] + i:
min_idx_sum = hash_dic[rest_num] + i
# 注意题目要求两个元素保持原有顺序
ans = [rest_num, num]
# 若num不位哈希表中，说明它第一次出现，记录其下标
# 由于我们希望两数的下标和尽可能小
# 故对于重复出现的数字，只记录其第一次出现的下标即可
if num not in hash_dic:
hash_dic[num] = i
# 输出答案，注意要按照格式输出
print(f"[{ans[0]},{ans[1]}]")
Java
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
String input = scanner.nextLine();
input = input.substring(1, input.length() - 1); // Remove square brackets
String[] numStrings = input.split(",");
int[] nums = new int[numStrings.length];
for (int i = 0; i < nums.length; i++) {
nums[i] = Integer.parseInt(numStrings[i].trim());
}
int target = scanner.nextInt();
int minIdxSum = nums.length * 2;
HashMap<Integer, Integer> hashDic = new HashMap<>();
int[] ans = new int[2];
for (int i = 0; i < nums.length; i++) {
int num = nums[i];
int restNum = target - num;
if (hashDic.containsKey(restNum)) {
if (minIdxSum > hashDic.get(restNum) + i) {
minIdxSum = hashDic.get(restNum) + i;
ans[0] = restNum;
ans[1] = num;
}
}
if (!hashDic.containsKey(num)) {
hashDic.put(num, i);
}
}
System.out.println("[" + ans[0] + "," + ans[1] + "]");
}
}
C++
“`C++
#include <iostream>
#include <vector>
#include <unordered_map>
#include <sstream>
using namespace std;
int main() {
string input;
getline(cin, input);
input = input.substr(1, input.length() – 2); // Remove square brackets
istringstream iss(input);
vector<int> nums;
string numStr;
<pre><code>while (getline(iss, numStr, ',')) {
nums.push_back(stoi(numStr));
}
int target;
cin >> target;
int minIdxSum = nums.size() * 2;
unordered_map<int, int> hashDic;
vector<int> ans(2);
for (int i = 0; i < nums.size(); i++) {
int num = nums[i];
int restNum = target – num;
if (hashDic.find(restNum) != hashDic.end()) {
if (minIdxSum > hashDic[restNum] + i) {
minIdxSum = hashDic[restNum] + i;
ans[0] = restNum;
ans[1] = num;
}
}
if (hashDic.find(num) == hashDic.end()) {
hashDic[num] = i;
}
}
cout << "[" << ans[0] << "," << ans[1] << "]" << endl;
return 0;
</code></pre>
}
“`
时空复杂度
时间复杂度：O(``N``)。仅需一次遍历数组。
空间复杂度：O(``N``)。哈希表所占空间。
说明
华为OD机试有三道题⽬，第⼀道和第⼆道属于简单或中等题，分值为 100 分，第三道为中等或困难题，分值为 200分，总分为 400 分。
机试分数越⾼评级越⾼，⼯资也就越⾼。
关于华为 OD 机试更加详细的介绍可以查看这篇⽂章：华为OD机考须知
关于机考题目汇总可以看这篇文章：华为OD机试真题 2023 A+B+C+D卷 + 2024新卷（Python&Java&C++）⽬录汇总（每⽇更新）
