51Nod - 1267 4个数字和为0(尺取法)

给出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"。 Sample Input
5
-1
1
-5
2
4
Sample Output
Yes

我一一开始写的深搜,不出意外的超时了,然后看了看别人的代码才想开可以把两个数相加的结果存到数组里然后排序,然后用双指针从两头开始查找。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
ll q[10000];
struct Node
{
	ll x,y,w;
}s[1005 * 1005];
bool compare(Node x,Node y)
{
	return x.w < y.w;
}
int main()
{
	ll k;
	scanf("%lld",&k);
	int u = 0;
	for(ll i = 0 ; i < k ; i++)
	{
		scanf("%lld",&q[i]);
	}	
	for(ll i = 0 ; i < k ; i++)
	{
		for(ll j = i + 1 ; j < k ; j++)
		{
			s[u].x = i;
			s[u].y = j;
			s[u++].w = q[i] + q[j];
		}
	}
	sort(s,s + u,compare);
	ll l = 0,r = u - 1;
	int flag = 0;
	while(l <= r)
	{
		if(s[l].w + s[r].w == 0)
		{
			if(s[l].x != s[r].x && s[l].x != s[r].y && s[l].y != s[r].x && s[l].y != s[r].y)
			{
				flag = 1;
				break;
			}
			else
			{
				l++;
				r--;
			}
		}
		if(s[l].w + s[r].w > 0)
		{
			r--;
		}
		if(s[l].w + s[r].w < 0)
		{
			l++;
		}
	}
	if(flag == 1)
	{
		printf("Yes\n");
	}
	else
	{
		printf("No\n");
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值