四个数列
-
题意:ZJM 有四个数列 A,B,C,D,每个数列都有 n 个数字。ZJM 从每个数列中各取出一个数,他想知道有多少种方案使得 4 个数的和为 0。
当一个数列中有多个相同的数字的时候,把它们当做不同的数对待。
请你帮帮他吧! -
输入输出:
Input
第一行:n(代表数列中数字的个数) (1≤n≤4000)
接下来的 n 行中,第 i 行有四个数字,分别表示数列 A,B,C,D 中的第 i 个数字(数字不超过 2 的 28 次方)
Output
输出不同组合的个数。
-
解题思路:
根据题意,需要在四个集合中分别拿出一个数,使得四个数的和为0。最简单的办法是对四个集合进行四重循环,将和为零的情况记录下来。不过这种方法的时间复杂度是O(n^4),显然不合适。故考虑优化。设四个集合为a,b,c,d,要使a+b+c+d=0,即
a+b= -(c+d),可以将a+b的所有答案记录下来,在c+d中查找,将复杂度降为O(n^2)。为了提高查找的效率,在查找时使用二分查找的方法,复杂度为O(logn)。
具体方法为:
开四个数组abcd分别记录四个集合的元素,ab和cd两个数组用来记录集合a+b和c+d的和。由于输入的数据量比较大所以用ios::sync_with_stdio(false)关闭兼容。先用双重循环将a+b和c+d的值分别存入数组ab和cd,对cd数组进行排序。将cd数组中的数依次取出,从ab数组中对该数进行二分查找。将s定位到小于要查找的数字的最后一个位置,因为可能会有多个满足条件的答案。从s开始向后遍历,如果满足条件和为零时,计数器ans+1,最后输出ans即为所求。 -
代码实现:
#include <iostream>
#include <algorithm>
using namespace std;
int a[4010],b[4010],c[