NUMPY095 题解：找和为目标值的元素对

题目

找出数组中所有和为目标值的元素对。

解题思路

1. 读取数组和目标值
2. 对每个元素，寻找补数
3. 输出所有满足条件的对

代码

import numpy as np

n, target = map(int, input().split())
arr = np.array(list(map(int, input().split())))
result = []
for i in range(n):
    complement = target - arr[i]
    indices = np.where(arr == complement)[0]
    for j in indices:
        if j > i:
            result.append(f"{arr[i]} {arr[j]}")
print('\n'.join(result) if result else "No pairs found")

运行示例

输入：

5 6
1 2 3 4 5

分析：

• $1 + 5 = 6$
• $2 + 4 = 6$

输出：

1 5
2 4

核心知识点

1. $\texttt{np.where}$ —— 条件索引