洛谷-珠心算测验python

P2141 [NOIP 2014 普及组] 珠心算测验

题目背景

NOIP2014 普及 T1

题目描述

珠心算是一种通过在脑中模拟算盘变化来完成快速运算的一种计算技术。珠心算训练，既能够开发智力，又能够为日常生活带来很多便利，因而在很多学校得到普及。

某学校的珠心算老师采用一种快速考察珠心算加法能力的测验方法。他随机生成一个正整数集合，集合中的数各不相同，然后要求学生回答：其中有多少个数，恰好等于集合中另外两个（不同的）数之和？

最近老师出了一些测验题，请你帮忙求出答案。

输入格式

共两行，第一行包含一个整数 $n$，表示测试题中给出的正整数个数。

第二行有 $n$ 个正整数，每两个正整数之间用一个空格隔开，表示测试题中给出的正整数。

输出格式

一个整数，表示测验题答案。

输入输出样例 #1

输入 #1

4
1 2 3 4

输出 #1

2

说明/提示

【样例说明】

由 $1+2=3,1+3=4$，故满足测试要求的答案为 $2$。

注意，加数和被加数必须是集合中的两个不同的数。

【数据说明】

对于 $100\%$ 的数据，$3\le n\le 100$，测验题给出的正整数大小不超过 $10,000$。

n = int(input())  # 读取正整数个数
numbers = list(map(int, input().split()))  # 读取整数列表

# 转换为集合，以便快速查找
set_numbers = set(numbers)

result = 0
# 使用一个集合来存储已经满足条件的和，避免重复统计
valid_sums = set()

# 遍历所有不同的 (a, b) 组合
for i in range(n):
    for j in range(i + 1, n):
        sum_of_two = numbers[i] + numbers[j]  # 计算两个数的和
        if sum_of_two in set_numbers and sum_of_two not in valid_sums:
            result += 1  # 满足条件的结果加1
            valid_sums.add(sum_of_two)  # 标记该和已经被计算过

print(result)  # 输出满足条件的数的个数

我们引入了一个新的集合 valid_sums 来记录所有已经统计过的和。这样，在每次计算出一个和 sum_of_two 时，我们检查它是否已经被统计过。如果统计过，就跳过，否则就统计并将其加入 valid_sums。