UVA - 1152 4 Values whose Sum is 0问题分解，二分查找

题目：点击打开题目链接

思路：暴力循环显然会超时，根据紫书提示，采取问题分解的方法，分成A+B与C+D，然后采取二分查找，复杂度降为O(n2logn)

AC代码：

#include <bits/stdc++.h>

using namespace std;

const int maxn = 4005;


int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T, n, ans; cin >> T;
    int A[maxn], B[maxn], C[maxn], D[maxn];
    while(T--) {
        ans = 0;

        vector<int> vec;
        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++)
                vec.push_back(A[i] + B[j]);
        sort(vec.begin(), vec.end());

        for(int i = 0; i < n; i++)
            for(int j = 0; j < n; j++)
                ans += upper_bound(vec.begin(), vec.end(), -(C[i] + D[j])) - lower_bound(vec.begin(), vec.end(),-(C[i] + D[j]));

        cout << ans << endl;
        if(T) cout << endl;
    }
    return 0;
}