HDOJ/HDU 2141 Can you find it? 二分搜索优化

题目:http://acm.hdu.edu.cn/showproblem.php?pid=2141

直接三重循环肯定超时了啊,O(n^3)是不允许的,现在把前两个数组加起来的每种情况保存,再用X-C[i]的值去做二分搜索,这样可以把复杂度降低到O(n^2+nlogn)就可以AC了,不过做二分搜索的时候一定要注意边界情况,不然会WA很多次,尼玛的!!

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<iostream> 
using namespace std;
int a[505],b[505],c[505];
int sum[505*505];
int l,n,m,k;
int flag;
void binary(int x)
{
	int left,right,mid;
	left=0,right=k-1;
	while(left<=right)
	{
		mid=(left+right)>>1;
		if(sum[mid]>x)
			right=mid-1;
		else if(sum[mid]<x)
			left=mid+1;
		else
			{flag=1;return ;}
	}
	return ;
}

int main()
{
	int i,j,q,x,cnt=1;
	while(cin>>l>>n>>m)
	{
		for(i=0;i<l;i++)
			scanf("%d",&a[i]);
		for(i=0;i<n;i++)
			scanf("%d",&b[i]);
		for(i=0;i<m;i++)
			scanf("%d",&c[i]);
		k=0;
		for(i=0;i<l;i++)
			for(j=0;j<n;j++)
			{
				sum[k++]=a[i]+b[j];
			}
		sort(sum,sum+k);
		scanf("%d",&q);
		printf("Case %d:\n",cnt++);
		while(q--)
		{
			scanf("%d",&x);
		    flag=0;
			for(i=0;i<m;i++)
			{
				binary(x-c[i]);
				if(flag){
					printf("YES\n");
					break;
				}		
			}
			if(!flag)
				printf("NO\n");
		}
	}
	return 0;
}


你可能感兴趣的:(二分查找,算法,搜索)