华为OD机试 Python 跳房子I

描述

小红玩一个叫“跳房子”的游戏，每个房子都有自己的格子。小红需要在两轮里跳到最后一个格子。她每次可以跳很多步，也可以往前或往后跳。现在的问题是：给定一些可选的跳跃步数，是否有两个步数使得小红能够跳到最后一个格子？如果可以，哪两个步数的索引之和最小？

输入说明：

输入的第一行是一个整数，表示房子的总格数。
输入的第二行是一系列整数，代表小红可以选择的跳跃的步数。
输出说明：

输出两个整数，表示可以使小红跳到最后一个格子的两个步数。

限制：

房子的总格数不超过1000。
可选择的跳跃步数不超过5000。
每步跳跃的步数在-100000000到100000000之间。
例子：

输入：
7
[1,4,5,2,2]

输出：
[5, 2]

代码

def find_min_index_sum_pair(count, steps):
    step_map = {