hdu 2141

最新推荐文章于 2024-05-14 11:49:30 发布

原创最新推荐文章于 2024-05-14 11:49:30 发布 · 384 阅读

0 ·

CC 4.0 BY-SA版权

查找专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一个使用二分查找解决的问题，即从三个已知数组中各选一个数相加是否能等于给定的目标数。通过双重循环组合两数组元素求和，并使用二分查找在第三个数组中验证是否存在补足目标值的数。

今天上午看了深搜和宽搜，看不懂，下午打算看看三分搜索，结果也是没有进行完，反而是看了点前奏——二分搜索。顺带做了一道用到查找的题，就把新学到的二分查找用了一下。

题目是杭电上面的。题目的大意是：给定三组数，给定一个目标数。在每组数中挑选出一个数相加，看是否存在这样一组数，如果存在，输出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;
}