Week4 B 四个数列

题目描述：

ZJM 有四个数列 A,B,C,D，每个数列都有 n 个数字。ZJM 从每个数列中各取出一个数，他想知道有多少种方案使得 4 个数的和为 0。
当一个数列中有多个相同的数字的时候，把它们当做不同的数对待。

输入格式：

第一行：n（代表数列中数字的个数） （1≤n≤4000）
接下来的 n 行中，第 i 行有四个数字，分别表示数列 A,B,C,D 中的第 i 个数字（数字不超过 2 的 28 次方）

Sample Input：

6
-45 22 42 -16
-41 -27 56 30
-36 53 -37 77
-36 30 -75 -46
26 -38 -10 62
-32 -54 -6 45

输出格式：

输出不同组合的个数。

Sample Output：

5

思路：

若对数据和一一枚举，则会到达n的四次方的复杂度，时间复杂度过高。因此，我们先使用前两个数组进行枚举，在sum数组中记录下他们的和，然后再枚举后两个数组的和，然后在sum数组中寻找他们的相反数的个数。在寻找相反数的个数时，我们选择对sum数组进行排序，然后再进行二分，找出该相反数的第一个和最后一个位置，从而求出该相反数的个数。如此可以将时间复杂度降低。

代码：

#include <iostream>
#include <algorithm>

using namespace std;

int findmax(int x,int n,int *a)
{
	int l = 0, r = n - 1, ans = -1;
	while (l <= r)
	{
		int mid = (l + r) >> 1;
		if (a[mid] == x)
		{
			ans = mid;
			l = mid + 1;
		}
		else if (a[mid] > x)r = mid - 1;
		else l = mid + 1;
	}
	return ans;
}

int findmin(int x, int n, int *a)
{
	int l = 0, r = n - 1, ans = -1;
	while (l <= r)
	{
		int mid = (l + r) >> 1;
		if (a[mid] == x)
		{
			ans = mid;
			r = mid - 1;
		}
		else if (a[mid] > x)r = mid - 1;
		else l = mid + 1;
	}
	return ans;
}

int main()
{
	int n;
	while (scanf("%d",&n)!=EOF)
	{
		int count = 0;
		int *a = new int[n];
		int *b = new int[n];
		int *c = new int[n];
		int *d = new int[n];
		for (int i = 0; i < n; i++)
			cin >> a[i] >> b[i] >> c[i] >> d[i];
		int *sum = new int[n*n]();
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++)
				sum[i*n + j] = a[i] + b[j];
		sort(sum, sum + n * n);
		for (int i = 0; i < n; i++)
			for (int j = 0; j < n; j++)
			{
				int last = -1;
				int tp = -(c[i] + d[j]);
				int max = findmax(tp, n*n, sum);
				int min = findmin(tp, n*n, sum);
				int ans = max - min + 1;
				if (min != last && max != last)
					count += ans;
			}
		cout << count << endl;
	}
	return 0;
}