题目大意:在4个都有n个元素的集合中,每个集合选出一个元素,使得4个数和为0。问有几种方案。
题目分析:开始想到暴力枚举,对四组数枚举,复杂度O(n^4)无法接受,后来对前三组数枚举累加,在第四组中找和的个数,如果对 第四组数据排序并二分查找,算法复杂度为O(n^3 logn)无法接受;最后采用二分。任选两组求和,剩下两组求和,枚举第一 组中 每一个和sum,在第二组和中查找-sum的个数,累加起来便得答案。
第一次用map映射前两组的sum,超时:
#include <cstdio>
#include <map>
using namespace std;
const int maxn = 4000 + 10;
int qua[4][maxn];
int main()
{
int T, first = 1;
scanf("%d", &T);
while(T--) {
int n;
scanf("%d", &n);
for(int i=0; i<n; i++) {
for(int j=0; j<4; j++) {
scanf("%d", &qua[j][i]);
}
}
map<int, int> mp;
for(int i=0; i<n; i++) {
for(int j=0; j<n; j++) {
mp[qua[0][i]+qua[1][j]]++;
}
}
int cnt = 0;
for(int i=0; i<n; i++) {
for(int j=0; j<n; j++) {
cnt += mp[-qua[2][i]-qua[3][j]];
}
}
if(first)first=0;
else putchar('\n');
printf("%d\n", cnt);
}
return 0;
}
第二次直接用数组存放sum,循环外sort,查找边界即可:
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 4000 + 10;
int qua[4][maxn];
int sum[maxn*maxn];
int main()
{
int T, first = 1;
scanf("%d", &T);
while(T--) {
int n;
scanf("%d", &n);
for(int i=0; i<n; i++) {
for(int j=0; j<4; j++) {
scanf("%d", &qua[j][i]);
}
}
int c = 0;
for(int i=0; i<n; i++) {
for(int j=0; j<n; j++) {
sum[c++] = qua[0][i]+qua[1][j];
}
}
sort(sum, sum+c);
int cnt = 0;
for(int i=0; i<n; i++) {
for(int j=0; j<n; j++) {
int t = -qua[2][i]-qua[3][j];
cnt += upper_bound(sum, sum+c, t) - lower_bound(sum, sum+c, t);
}
}
if(first)first=0;
else putchar('\n');
printf("%d\n", cnt);
}
return 0;
}