【Python】洛谷P2141[NOIP2014 普及组] 珠心算测验题解

原题链接

• P2141

题目描述

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

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

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

(本题目为2014NOIP普及T1)

输入格式

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

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

输出格式

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

输入输出样例

输入 #1

4
1 2 3 4

输出 #1

2

说明/提示

【样例说明】

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

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

【数据说明】

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

代码实现

n = int(input())
num_list = set(map(int, input().split()))
print(len(set(a+b for a in num_list for b in num_list if a!=b).intersection(num_list)))

代码部分参考了这位大佬：Findegil 的博客，这算法实在是太强了，再次献上我的膝盖orz