[51nod] 1267 4个数和为0 暴力+二分

第1行，1个数N，N为数组的长度(4 <= N <= 1000)
第2 - N + 1行：A[i]（-10^9 <= A[i] <= 10^9)

如果可以选出4个数，使得他们的和为0，则输出"Yes"，否则输出"No"。

5
-1
1
-5
2
4

Yes

暴力+排序+二分

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;

int a[1100];

int main()
{
//freopen("1.txt", "r", stdin);
int n;
scanf("%d", &n);
for (int i = 0; i < n; i++)
scanf("%d", &a[i]);
sort(a, a+n);

for (int i = 0; i < n; i++)
for (int j = i+1; j < n; j++) {
int k, m, sum;
k = j+1;
m = n-1;
while (k < m) {
sum = a[i]+a[j]+a[k]+a[m];
if (sum < 0) {
k++;
}
else if (sum > 0) {
m--;
}
else {
k++; m--;
printf("Yes\n");
return 0;
}
}
}
printf("No\n");

return 0;
}

还有个很好的思路，先把数组中任意两个数存起来，再从小到大排序，类似二分的思想从两头开始查找，是否满足sum1+sum2==0 并且两个坐标不重叠，每个点只选一次

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;

int a[1100];
struct node
{
int x, y;
int sum;
}P[1100*1100];

int cmp(node a, node b)
{
return a.sum < b.sum;
}

int main()
{
freopen("1.txt", "r", stdin);
int n;
scanf("%d", &n);
for (int i = 0; i < n; i++)
scanf("%d", &a[i]);

int k = 0;
for (int i = 0; i < n; i++)
for (int j = i+1; j < n; j++) {
P[k].x = a[i];
P[k].y = a[j];
P[k++].sum = a[i]+a[j];
}
sort(P, P+k, cmp);
int l = 0, r = k-1;
int flag = 0;
while (l < r) {
if (P[l].sum+P[r].sum==0 && P[l].x!=P[r].x && P[l].y!=P[r].y
&& P[l].x!=P[r].y && P[l].y != P[r].x) {
flag = 1;
break;
}
else if (P[l].sum+P[r].sum < 0)
l++;
else r--;
}
if (flag) printf("Yes\n");
else printf("No\n");

return 0;
}