uva1152 - 4 Values whose Sum is 0 入门经典II 第八章 例题8-3

题目链接：http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=246&page=show_problem&problem=3593

分析：求a+b+c+d=0。分别枚举a+b，-c-d。和书上的思想一样。

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;

const int maxn = 4000 + 10;
int a[maxn],b[maxn],c[maxn],d[maxn],ab[maxn*maxn];
int main(){
    int T;
    cin >> T;
    while(T--){
        int N;
        cin >> N;
        for(int i=0;i<N;i++){
            cin>>a[i]>>b[i]>>c[i]>>d[i];
        }

        for(int i=0;i<N;i++)
            for(int j=0;j<N;j++)
                ab[i*N+j]=a[i]+b[j];

        sort(ab,ab+N*N);
        long long cnt=0;//用int也能过，不过还是要注意一下，毕竟在比赛中犯过同样的错误
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                cnt+=upper_bound(ab,ab+N*N,-c[i]-d[j]) -
                    lower_bound(ab,ab+N*N,-c[i]-d[j]);
            }
        }

        printf("%lld\n",cnt);
        if(T) printf("\n");
    }
    return 0;
}

汝佳提供的代码：

// UVa1152 4 Values Whose Sum is Zero
// Rujia Liu
#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn = 4000 + 5;
int n, c, A[maxn], B[maxn], C[maxn], D[maxn], sums[maxn*maxn];

int main() {
  int T;
  scanf("%d", &T);
  while(T--) {
    scanf("%d", &n);
    for(int i = 0; i < n; i++)
      scanf("%d%d%d%d", &A[i], &B[i], &C[i], &D[i]);
    c = 0;
    for(int i = 0; i < n; i++)
      for(int j = 0; j < n; j++)
        sums[c++] = A[i] + B[j];
    sort(sums, sums+c);
    long long cnt = 0;
    for(int i = 0; i < n; i++)
      for(int j = 0; j < n; j++)
        cnt += upper_bound(sums, sums+c, -C[i]-D[j]) - lower_bound(sums, sums+c, -C[i]-D[j]);
    printf("%lld\n", cnt);
    if(T) printf("\n");
  }
  return 0;
}