poj 2785 二分的方法

最新推荐文章于 2019-01-20 19:56:14 发布

ming_514

最新推荐文章于 2019-01-20 19:56:14 发布

阅读量389

点赞数 2

CC 4.0 BY-SA版权

分类专栏：贪心

本文链接：https://blog.youkuaiyun.com/qq_36819130/article/details/72469462

贪心专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种解决四数求和为零问题的有效算法。通过将问题转换为两个两数之和的问题，并利用二分查找提高效率，避免了四重循环的复杂度。文章详细展示了如何使用C++实现该算法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个题的大题意思就是需要从四列中随机找四个数，他们的和为零，四重循环是不可能的，想了想分成两组去做，发现也超时了，后来就用了二分的方法过的。

也是挺头疼的，也学习一波二分。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include <iostream>
using namespace std;
int a[4][4444],n;
int num1[16100000],num2[16100000],c;
int erfen(int left,int right,int k)
{
    int i;
    while(left<=right)
    {
        int mid=(left+right)/2;
        int num=0;
        if(num2[mid]==k)
        {
            num=1;
            for(i=mid-1; i>=0&&num2[i]==k; i--)  num++;
            for(i=mid+1; i<n*n&&num2[i]==k; i++)  num++;
            return num;
        }
        else if(num2[mid]>k)
            right=mid-1;
        else left=mid+1;
    }
    return 0;

}
int main()
{
    int i,j;
    while(scanf("%d",&n)!=EOF)
    {
        int num=0;
        c=0;
        for(i=0; i<n; i++)
        {
            scanf("%d %d %d %d",&a[0][i],&a[1][i],&a[2][i],&a[3][i]);
        }
        for(i=0; i<n; i++)
            for(j=0; j<n; j++)
            {
                num1[c]=a[2][i]+a[3][j];
                num2[c++]=-(a[0][i]+a[1][j]);
            }
        sort(num2,num2+c);

        for(i=0; i<c; i++)
        {
            num+=erfen(0,n*n-1,num1[i]);
        }
        printf("%d\n",num);
    }
    return 0;
}