FZU OJ 1707 等式数量

FZU的1707等式数量问题，折腾了好久，几次的超时和一次的wrong answer，终于在最后ac了。刚开始的思路是最普通的P(n,2)，再for数组找等号，慢得不行。再来sort后，用binary_search找等号还是慢。接着改用set存数据，还是p(n,2)，再set找等号，还是超时。哦哦……

反过来，逆向思考，先找等号右边，再找等号左边的式子，于是，还是先sort数组，再for循环，对每个数组元素查等号左边的两个数，为减少查找的复杂度，采用头尾查找的方式可省掉不少功夫，抓住头尾两数，一小一大的特点，若相加>当前数，则大数往左移；若相加<当前数，则小数往右移；若=成立，则count++；当小数与大数相遇时，退出此次查找。

代码如下：

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