国际空间研究组织 | ISRO CS 2008 |问题 79

题目描述

给定一个数组A和一个整数K，编写一个程序来查找数组A中是否存在两个数字，它们的和等于K。

函数声明为：int findSum(int A[], int N, int K);

输入格式

第一行包含一个整数T，表示测试用例的数量。

对于每个测试用例，第一行包含一个整数N，表示数组的长度。

第二行包含N个空格分隔的整数，表示数组A的元素。

第三行包含一个整数K，表示要查找的和。

输出格式

对于每个测试用例，如果可以找到两个数字，使它们的和等于K，则打印1，否则输出0。

输入样例：

2
4
1 2 4 3
8
4
2 4 6 8
5

输出样例：

1
0

解题思路

这道题可以使用哈希表来实现，具体步骤如下：

1. 创建一个哈希表，用来存储数组中出现的数字。
2. 遍历数组A，计算K与当前数字的差值diff。
3. 在哈希表中查找diff是否存在，如果存在，则说明可以找到一对数字与当前数字的和为K，返回1。
4. 如果不存在，则将当前数字插入到哈希表中。
5. 如果遍历完了整个数组A后仍然没有找到满足条件的数字，说明不存在这样的一对数字，返回0。

代码实现

```python
def findSum(A, n, K):
    hash_table = {}

    for i in range(n):
        diff = K - A[i]
        if diff in hash_table:
            return 1
        hash_table[A[i]] = True

    return 0

t = int(input())

for _ in range(t):
    n = int(input())
    A = list(map(int, input().split()))
    K = int(input())

    print(findSum(A, n, K))