求和问题可以被看做是以下的公式,给定 A,B,C,D 四个列表,计算有多少四元组满足 (a, b, c, d) ∈ A × B × C × D 且 a + b + c + d = 0。我们推测所有的列表都有 n 个数字。
注:不同的四元组是指元素位置不一样的四元组
数据范围 n<=2e
样例输入
输入的第一个数字指明有 T 组。每一组这样描述,第一行是列表大小 n, 然后有 n 行。每一行都有四个整型数字,分别属于 A,B,C,D 四列。
样例输出
对于每一个测试用例,统计有多少个四元组满足他们的和是 0 。每一组数据一行。
Sample Input
Copy to Clipboard
1 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
Copy to Clipboard
5
/*
* @Description: To iterate is human, to recurse divine.
* @Autor: Recursion
* @Date: 2022-03-23 09:44:28
* @LastEditTime: 2022-03-23 12:03:40
*/
#include<bits/stdc++.h>
using namespace std;
int T,n,sum;
int a[10000],b[10000],c[10000],d[10000];
int ab[200000001],cd[200000001];
// int a[10000][10000];
// void read()
// {
// for(int i = 1;i <= n;i ++)
// for(int j = 1;j <= 4;j ++)
// cin >> a[i][j];
// }
// void dfs(int x)
// {
// if(x == 4 + 1){
// if(sum == 0)
// ans++;
// return;
// }
// for(int i = 1;i <= n;i ++){
// sum += a[i][x];
// dfs(x + 1);
// sum -= a[i][x];
// }
// }
// int main()
// {
// cin >> T;
// while(T--){
// cin >> n;
// read();
// dfs(1);
// cout << ans << endl;
// }
// }
void read()
{
for(int i = 0;i < n;i ++)
cin >> a[i] >> b[i] >> c[i] >> d[i];
}
void solve()
{
for(int i = 0;i < n;i ++)
for(int j = 0;j < n;j ++){
ab[i*n + j] = a[i] + b[j];
cd[i*n + j] = c[i] + d[j];
}
sort(ab,ab + n*n);
sort(cd,cd + n*n);
long long int ans = 0;
for(int i = 0;i < n*n;i ++){
int temp = -ab[i];
ans += upper_bound(cd,cd + n*n, temp) - lower_bound(cd,cd + n*n,temp);
}
/*
upper_bound(first, last, val) 寻找在数组或容器的[first,last)范围第一个大于val的元素位置
lower_bound(first, last, val)寻找在数组或容器的[first,last)范围第一个大于等于val的元素位置
使用时必须为有序的数组或容器
相减得到相等个数
lower_bound:
功能:查找非递减序列[first,last) 内第一个大于或等于某个元素的位置。
返回值:如果找到返回找到元素的地址否则返回last的地址。(这样不注意的话会越界,小心)
用法:int t=lower_bound(a+l,a+r,key)-a;(a是数组)。
upper_bound:
功能:查找非递减序列[first,last) 内第一个大于某个元素的位置。
返回值:如果找到返回找到元素的地址否则返回last的地址。(同样这样不注意的话会越界,小心)
用法:int t=upper_bound(a+l,a+r,key)-a;(a是数组)。
*/
cout << ans << endl;
}
int main()
{
int t;
cin >> t;
while(t--){
cin >> n;
read();
solve();
}
}