Description
你的任务是从集合中找出所有的等式数量。
Input
每组输入数据的第一行为一个整数N (3<=N<=5,000),表示集合中一共有N个不同的整数。之后一行有N个整数,表示集合中的元素,元素的整数值大小为(-10,000,000~10,000,000)。
Output
对于每组数据输出一个整数占一行,表示该集合的等式数量。
Sample Input
43 1 2 5
Sample Output
2
这题:直接做会超时用容器也超时了。
用头尾查找(头和尾同时进行处理)
//头尾查找
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[5005];
int main()
{
int n,i;
while(scanf("%d",&n)!=EOF)
{
for(i=0;i<n;i++)
scanf("%d",a+i);
sort(a,a+n);
int sum=0,l,r;
for(i=0;i<n;i++)
{
l=0;r=n-1;
while(l<r)
{
if(i==l) l++;
else if(i==r) r--;
else if(a[i]<a[r]+a[l])r--;
else if(a[i]>a[r]+a[l]) l++;
else {sum++;r--;}
}
}
printf("%d\n",sum);
}
return 0;
}