a+b+c+d==0

求和问题可以被看做是以下的公式,给定 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();
    }
}

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值