POJ 2785 4 Values whose Sum is 0

最新推荐文章于 2025-08-12 17:21:36 发布

weixin_30485799

最新推荐文章于 2025-08-12 17:21:36 发布

阅读量56

点赞数

CC 4.0 BY-SA版权

文章标签： c/c++

原文链接：http://www.cnblogs.com/Yu2012/archive/2012/05/05/2483605.html

本文介绍了一种解决四数组求和问题的算法思路，通过枚举左边两个数的和及右边两个数的和，并利用二分查找进行配对计数。文章提供了C++实现代码，并对比了不同排序方法的效率。

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

枚举左边两个数的和以及右边两个数的和，用二分查找将其配对并计数。Hash的方法晚点再写。

/*Accepted 49208K 5188MS C++ 1146B 2012-05-04 23:56:30 */

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
#define MAXN 4005

/*int cmp( const void *_p, const void *_q)
{
    int *p = ( int *)_p;
    int *q = ( int *)_q;
    return *p - *q;
}*/

int a[MAXN], b[MAXN], c[MAXN], d[MAXN];
int x[MAXN * MAXN], y[MAXN * MAXN];
int n, m, ans;

int bs( int num)
{
    int left, right, mid, i, cnt;
    left = 0, right = m;
    while( left < right)
    {
        mid = ( left + right) >> 1;
        if( num + y[mid] >= 0)
            right = mid;
        else left = mid + 1;
    }
    for( i = left, cnt = 0; i < m && y[i] + num == 0; i ++, cnt ++);
    return cnt;
}

int main()
{
    while( scanf( "%d", &n) == 1)
    {
        for( int i = 0; i < n; i ++)
            scanf( "%d%d%d%d", &a[i], &b[i], &c[i], &d[i]);
        m = 0;
        for( int i = 0; i < n; i ++)
            for( int j = 0; j < n; j ++)
            {
                x[m] = a[i] + b[j];
                y[m] = c[i] + d[j];
                m ++;
            }
        sort( x, x + m);
        sort( y, y + m);
        //qsort( x, m, sizeof x[0], cmp);
        //qsort( y, m, sizeof y[0], cmp);
        ans = 0;
        for( int i = 0; i < m; i ++)
            ans += bs( x[i]);
        printf( "%d\n", ans);
    }
    return 0;
}