今天上午看了深搜和宽搜,看不懂,下午打算看看三分搜索,结果也是没有进行完,反而是看了点前奏——二分搜索。顺带做了一道用到查找的题,就把新学到的二分查找用了一下。
题目是杭电上面的。题目的大意是:给定三组数,给定一个目标数。在每组数中挑选出一个数相加,看是否存在这样一组数,如果存在,输出yes,如果不存在,输出no。用三重循环肯定是超时的,于是,先两重循环,两个数相加,再查找在第三组数中是否存在这样一个数。
代码实现:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 510
using namespace std;
__int64 d[N*N];
int BinarySearch(__int64 a[], int n, __int64 goal)
{
int left = 0, right = n-1;
int mid;
while(left <= right){
mid = (left+right)/2;
if(a[mid] == goal) break;
if(a[mid] > goal) right = mid-1;
else left = mid+1;
}
if(left > right) return -1;
else
return mid;
}
int main()
{
int l ,n , m, ct = 1;
__int64 a[N], b[N], c[N],x;
int i , j ,k;
while(~scanf("%d%d%d" , &l, &n, &m)){
int num = 0;
for(i =0; i<l; i++)
scanf("%I64d", &a[i]);
for(j = 0; j<n; j++)
scanf("%I64d", &b[j]);
for(k =0; k<m; k++)
scanf("%I64d", &c[k]);
for(i =0; i<l; i++)
for(j = 0; j<n; j++)
d[num++] = a[i] +b[j];
sort(d, d+num);
sort(c, c+m);
int s;
scanf("%d", &s);
printf("Case %d:\n", ct++);
while(s--){
scanf("%I64d", &x);
if(x<d[0]+c[0] || x>d[num-1]+c[m-1])
printf("NO\n");
else{
__int64 key;
for(i =0; i<m; i++){
key = x-c[i];
if(BinarySearch(d,num,key)!=-1)
{
printf("YES\n");
break;
}
}
if(i==m) printf("NO\n");
}
}
}
return 0;
}