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

来源：互联网发布：锁屏也能抢红包的软件编辑：程序博客网时间：2024/05/22 00:28

题目：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;}