poj2785 4 Values whose Sum is 0

最新推荐文章于 2022-02-18 17:51:58 发布

nacrt

最新推荐文章于 2022-02-18 17:51:58 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： # 二分查找文章标签：折半枚举双向搜索

本文链接：https://blog.youkuaiyun.com/qq_31975227/article/details/70153557

二分查找专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种解决四数组和为0问题的有效算法。通过对数组C和D的所有可能组合求和并排序，然后利用二分查找法找到与A和B组合相反数相等的和，从而高效地计算出所有和为0的组合数量。

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

原题链接： poj2785 4 Values whose Sum is 0

题意：给出元素都为n的4个数组A,B,C,D,要从每个数组中各取一个数，求这四个数和为0的次数，一个数列有多个相同数字时，当作不同数字看待

//思路先枚举C,D的和排序，再枚举A，B的和在C，D和中进行折半查找C+D=-(A+B)的部分 
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAX_N = 4000 + 5;
//输入 
int n;
int A[MAX_N], B[MAX_N], C[MAX_N], D[MAX_N];
int CD[MAX_N * MAX_N]; //存C、D中的所有和 
int main(){
	int n, i, j;
	scanf("%d", &n);
	for(i = 0;i < n;i ++)
		scanf("%d%d%d%d", &A[i], &B[i], &C[i], &D[i]); 
	for(i = 0;i < n;i ++){	//枚举C、D的和存入CD中 
		for(j = 0;j < n;j ++)
			CD[i * n + j] = C[i] + D[j];
	}
	sort(CD, CD + n * n);	//对和排序 
	long long res = 0;
	for(i = 0;i < n;i ++){
		for(j = 0;j < n;j ++){
			int cd = -(A[i] + B[j]);
			//取出C和D中和为cd的部分（CD中>cd的第一个位置与>=cd的第一个位置相减） 
			res += upper_bound(CD, CD + n * n, cd) - lower_bound(CD, CD + n * n, cd);
		}
	}
	printf("%lld\n", res);
	return 0;
}