51nod 1267 4个数和为0

思路：先n^2预处理出所有两两的组合，那么就变成要找a+b==0的问题了，那么排序O(n)去找就可以了

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int maxn = 1005*1005;
struct Node
{
	LL v;
	int x,y;
}node[maxn];
bool cmp(Node a,Node b){return a.v<b.v;}
LL a[1005];
bool check(int i,int j)
{
	if(node[i].x!=node[j].x && node[i].x!=node[j].y && node[i].y!=node[j].x && node[i].y!=node[j].y)
		return true;
	return false;
}
int main()
{
    int n;
	scanf("%d",&n);
	for(int i = 1;i<=n;i++)scanf("%lld",&a[i]);
	int tot = 1;
	for(int i = 1;i<=n;i++)
	{
		for(int j = i;j<=n;j++)
		{
			if(i==j)continue;
            node[tot].v=a[i]+a[j];
			node[tot].x=i;
			node[tot++].y=j;
		}
	}
	sort(node+1,node+tot,cmp);
	int i=1;
	int j=tot-1;
	bool flag = false;
	while(i<j)
	{
		//printf("%lld %d %d   %lld %d %d\n",node[i].v,node[i].x,node[i].y,node[j].v,node[j].x,node[j].y);
		if(node[i].v+node[j].v>0)
			j--;
		else if (node[i].v+node[j].v<0)
			i++;
		else if(node[i].v+node[j].v==0 && check(i,j))
		{
			flag = true;
			printf("Yes\n");
			break;
		}
		else 
		{
			i++;
			j--;
		}
	}
	if(!flag)printf("No\n");
}

1267 4个数和为0

基准时间限制：1 秒 空间限制：131072 KB 分值: 20  难度：3级算法题

 收藏

 关注

给出N个整数，你来判断一下是否能够选出4个数，他们的和为0，可以则输出"Yes"，否则输出"No"。

Input

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

Output

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

Input示例

5
-1
1
-5
2
4

Output示例

Yes