四个数列(二分法查找)

该博客介绍了如何解决一个寻找四个数列中元素和为0的组合问题。通过优化策略,将原本的四重循环复杂度降低到O(n^2)并利用二分查找进一步提升效率至O(logn)。具体方法包括存储数列和、排序、二分查找等步骤。
摘要由CSDN通过智能技术生成

四个数列

  • 题意: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[
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值