HDU 2141--Can you find it?(二分查找基础好题)

题意:能不能从给定的三个序列中找出满足ai+bj+ck==给定的x值,能则输出YES,否则输出NO。
思路:暴搜绝对会超时,由于区间长度为静态,考虑排序后二分(二分的前提一定是有序)。二分有两种选择,其中一种会超时:1.在c中二分查找 x-a[i]-b[j] 是否存在,那么时间复杂度为n * m * log( p )(设a,b,c数组的长度为n,m,p),会超时;
2.还有一种更好的方法,先预处理出a,b,数组中所有可能组合情况的和(放入ab数组),在ab中二分查找 x-c[i]是否存在,时间复杂度为p * log ( n * m);

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
//朴素暴搜时间复杂度为n^3会TIE;
//对于区间为静态(即无增删操作时),考虑二分查找;
//这里二分可以总值减去前两个值去二分第三个值,但是这样时间复杂度有n*m*logp,查询有k次,
//这样做还是会TLE(为什么我知道,因为我这样做TLE了...)
//故二分应该预处理出前两项,用总值减去第三项,取预处理出来的数组中二分,
//这样复杂度变为了p*log(n*m),还要注意类型要开long long; 

ll a[666],b[666],c[666],ab[666*666];//ab数组用于预处理a,b两数组的和(数组要开得足够大); 
int main()
{
	int i,j;
	int n,m,p;
	int t,k=0;
	ll x;
	while(scanf("%d%d%d",&n,&m,&p)!=EOF)
	{
		for(i=1;i<=n;i++)
		{
			scanf("%lld",&a[i]);
		}
		for(i=1;i<=m;i++)
		{
			scanf("%lld",&b[i]);
		}
		for(i=1;i<=p;i++)
		{
			scanf("%lld",&c[i]);
		}
		int tot=0;
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=m;j++)
			{
				ab[++tot]=a[i]+b[j];//预处理出a,b两数组的和; 
			}
		}
		sort(ab+1,ab+1+tot);//二分的条件为序列有序,故一定要先排序; 
		sort(c+1,c+1+p);
		scanf("%d",&t);//t次询问; 
		printf("Case %d:\n",++k);
		while(t--)
		{
			scanf("%lld",&x);
			if((x<ab[1]+c[1])||(x>ab[tot]+c[p]))//若x比最小的还小或者x比最大的还大就不用搜了
			{
				printf("NO\n");
				continue;
			 } 
			bool flag=true;//flag==false时为找到了满足条件的数;
			for(i=1;i<=p&&flag;i++)
			{
				ll ans=x-c[i];
				int l=1;
				int r=tot;//对预处理出来的ab数组进行二分;
				while(l<=r)
				{
					int mid=(l+r)>>1;
					if(ab[mid]==ans)
					{
						flag=false;//找到符合条件的解; 
						break;
					}
					else
					if(ab[mid]<ans)
					{
						l=mid+1;
					}
					else
					{
						r=mid-1;
				    }
				}  
			 }
			 if(!flag)
			 {
			 	printf("YES\n");
			 }
			 else
			 {
			 	printf("NO\n");
			 }
		}
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值