FZU 1707 等式数量

Accept: 207Submit: 1317
Time Limit: 1000 mSecMemory Limit : 32768 KB

Problem Description

集合是基本的数学概念，它是集合论的研究对象。关于集合论最简单的说法，是在最原始的集合论《朴素集合论》中的定义：集合就是“一堆东西”。集合里的“东西”，叫作元素。若x是集合A中的元素，记作 x∈A。集合中的元素是互不相同的。
表示相等关系的式子叫做等式。
给你一个包含有N个整数的集合，如果从集合中找到三个不同的整数a,b,c使得a+b=c，那么我们说这个集合包含一个等式。
你的任务是从集合中找出所有的等式数量。

Input

每组输入数据的第一行为一个整数N (3<=N<=5,000)，表示集合中一共有N个不同的整数。之后一行有N个整数，表示集合中的元素，元素的整数值大小为(-10,000,000~10,000,000)。

Output

对于每组数据输出一个整数占一行，表示该集合的等式数量。

Sample Input

4

3 1 2 5

Sample Output

2

//困了我一晚上，TLE>=4次......

#include<iostream> #include<algorithm> #include<functional> using namespace std; int main() { int n,a[5001]; while(scanf("%d",&n)!=EOF) { int i,left,right,count=0; for(i=0;i<n;i++) scanf("%d",&a[i]); sort(a,a+n,greater<int>()); for(i=0;i<n;i++) { left=0;right=n-1; while(right>left) { while(a[i]<a[left]+a[right] && left<n) left++; while(a[i]>a[left]+a[right] && right>=0) right--; if(right>left && a[i]==a[left]+a[right] && i!=left && i!=right) count++; left++; } } printf("%d/n",count); } return 0; }