Embedded_game/人工智能赛项/004_查找两个总和为特定值的索引.py

29 lines
836 B
Python
Raw Normal View History

2025-01-02 12:48:11 +08:00
# -*- coding:utf-8 -*-
# @Author len
# @Create 2023/12/14 21:09
def find_two_sum_indices(nums, target_sum):
# 创建一个字典来存储数组值到索引的映射
num_to_index = {}
for index, num in enumerate(nums):
# 计算当前数字与目标和的差
difference = target_sum - num
if difference in num_to_index:
# 如果差值在字典中,返回对应的索引和当前索引
return [num_to_index[difference], index]
num_to_index[num] = index
# 如果没有找到,则返回空列表
return []
# 假设的输入
array_length = int(input())
array = list(map(int, input().split(" ")))
specific_sum = int(input())
# 查找和为特定值的两个索引
result_indices = find_two_sum_indices(array, specific_sum)
# 输出结果
print(result_indices)