二分贪心--04

最新推荐文章于 2025-03-09 21:53:43 发布

原创最新推荐文章于 2025-03-09 21:53:43 发布 · 284 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据 #二分查找

二分贪心专栏收录该内容

29 篇文章

订阅专栏

题目大概：

输入n行数据，每行数据有四个数，共四组数，从四组数中挑出一个，组成一列数，找出四组数中相加为零的数列的个数。

思路：

前两列数相加，有n*n个和，后两列数相加，有n*n个和，找出两组和相加为零的个数，可以把一组和排序，然后用upper_bound()和lower_bound()函数，求每一个和为零的个数。

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
int a[4000],b[4000],c[4000],d[4000],ab[16000000],cd[16000000];

int main()
{int n;

scanf("%d",&n);

for(int i=0;i<n;i++)
{scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
}int k=0;
 for(int i=0; i<n; ++i)
        for(int j=0;j<n;++j)
            {cd[k++]=c[i]+d[j];
            }
    sort(cd,cd+n*n);
    long long sum=0;
    for(int i=0; i<n; ++i)
        for(int j=0; j<n; ++j)
        {
            int ad=-(a[i]+b[j]);

            sum+=upper_bound(cd,cd+n*n,ad)-lower_bound(cd,cd+n*n,ad);
        }


printf("%d",sum);
    return 0;
}