2021-09-13 POJ2785四数之和

题目链接
最朴素的方法是$O(n^4)$地枚举四个数组的所有元素（注意每个数组都从0开始），不过很显然会TLE（尽管此题时限长）。如果对最后一个数组进行排序，将第四层循环改成二分查找，时间复杂度为$O(n^{3}logn)$，虽然明显改进但仍不是好的算法。
正确思路：将后两个数组每两个元素的和计算出来并存放在新数组sum[]中。求出前两个和subsum后，计算sum[]中有多少个元素为-subsum。复杂度$O(n^{2}logn)$
注意读入顺序。
sum要开到平方而不是二倍，因此此题可以说是用空间换时间。
参考代码：

#include <cstdio>
#include <algorithm>
const int M=4001;
using namespace std;
int a[M],b[M],c[M],d[M],sum_cd[M*M];

int main(){
    int n;
    scanf("%d",&n);
	for(int i=0; i<n; i++)  scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
    int k=0;
    for(int i=0; i<n; i++) {
        for(int j=0; j<n; j++) {
            sum_cd[k++]=c[i]+d[j];
        }
    }
    sort(sum_cd,sum_cd+k);
    long long res=0;
    for(int i=0; i<n; i++) {
        for(int j=0; j<n; j++) {
            int sum=a[i]+b[j];
            res+=(long long)(upper_bound(sum_cd,sum_cd+k,-sum)-lower_bound(sum_cd,sum_cd+k,-sum));
        }
    }
    printf("%lld",res);
    return 0;
}