小白暑假加训备战网络赛day8-CSDN博客

本文链接：https://blog.csdn.net/zhiyixi170/article/details/149105920

生老病死自然规律珍惜当下享受当下

文章目录

一、思维 / 二分
总结

一、思维 / 二分

思路：所有情况遍历时间复杂度是 O (n ^ 4)，所以我们可以拆成2个人一组去暴力枚举，时间复杂度 O (n ^ 2)，记录两个数字之和出现的次数，对于前两个人可以取得的值 x ，判断是否存在 -x, 这样两个相加和为 0 ，组合数量为 num1[x] * num2[-x]，题目的标签是二分，时间复杂度为 O (n ^ 2 * logn)，但是该方法完全可以实现 O (n ^ 2) 的时间复杂度通过这道题

代码：

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long 
#define ff first
#define ss second
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin >> n;
    vector<vector<int>>a(n + 5, vector<int>(5));
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= 4; j++)
        {
            cin >> a[i][j];
        }
    }
    unordered_map<int, int>v, mp;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            int ans = a[i][1] + a[j][2];
            v[ans]++;
        }
    }
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            int ans = a[i][3] + a[j][4];
            mp[ans]++;
        }
    }
    int sum = 0;
    for (auto i : v)
    {
        
        sum += i.ss * mp[-i.ff];
    }
    cout << sum;
}