题目描述与示例

题目描述

跳房子，也叫跳飞机，是一种世界性的儿童游戏。 游戏参与者需要分多个回合按顺序跳到第1格直到房子的最后一格

跳房子的过程中，可以向前跳，也可以向后跳。

假设房子的总格数是count，小红每回合可能连续跳的步教都放在数组steps中，请问数组中是否有一种步数的组合，可以让小红两个回合跳到最后一格? 如果有，请输出索引和最小的步数组合。

注意:

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

输入描述

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

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

输出描述

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

备注

• count ≤ 1000
• 0 ≤ 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_numrest_num = target-num# 若rest_num位于哈希表中，说明其曾经出现过# rest_num和num相加等于所需要的结果targetif rest_num in hash_dic:# 若此时min_idx_sum大于两者下标和# 则更新min_idx_sum和ansif 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 bracketsString[] 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++

#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 bracketsistringstream iss(input);vector<int> nums;string numStr;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;
}

时空复杂度

时间复杂度：O(N)。仅需一次遍历数组。

空间复杂度：O(N)。哈希表所占空间。

华为OD算法/大厂面试高频题算法练习冲刺训练

• 华为OD算法/大厂面试高频题算法冲刺训练目前开始常态化报名！目前已服务100+同学成功上岸！

• 课程讲师为全网50w+粉丝编程博主@吴师兄学算法 以及小红书头部编程博主@闭着眼睛学数理化

• 每期人数维持在20人内，保证能够最大限度地满足到每一个同学的需求，达到和1v1同样的学习效果！

• 60+天陪伴式学习，40+直播课时，300+动画图解视频，300+LeetCode经典题，200+华为OD真题/大厂真题，还有简历修改、模拟面试、专属HR对接将为你解锁

• 可上全网独家的欧弟OJ系统练习华子OD、大厂真题

• 可查看链接 OD算法冲刺训练课程表 & OD真题汇总(持续更新)

• 绿色聊天软件戳 od1336了解更多