【校OJ】选网线

暑假学校OJ上的题目。

一道很有意思的二分。

 

题意：三个数组，每个数组各选一个数出来看是否能组成目标数。

题解：前两个数组两两的和组合一下，二分第三个数组，找是否能组成目标数。

 

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn = 1001;

int l,n,m;
int a[maxn];
int b[maxn];
int c[maxn];
int sum[maxn];
int cnt = 1;

int find(int x){    
    for(int i = 1; i <= m; i ++){
        int l = 1,r = cnt+1;
        int mid;
        while( l <= r){
            mid = ( l + r) >> 1;
            if(c[i] + sum[mid] == x){
                return true;
            } 
            else if(c[i] + sum[mid] < x){
                l = mid + 1; 
            }
            else if(c[i] + sum[mid] > x){
                r = mid - 1;
            }
        }
    }
    return false;
} 

int main(){
    cin>>l>>n>>m;
    for(int i = 1; i <= l ;i++){
        cin>>a[i];
    }
    for(int i = 1; i <= n ;i++){
        cin>>b[i];
    }
    for(int i = 1; i <= m ;i++){
        cin>>c[i];
    }
    for(int i = 1; i <= l ;i++){
        for(int j = 1; j <= n ;j++){
            sum[cnt++] = a[i]+b[j];
        }
    }
    sort(sum+1,sum+1+n);
    int s;
    cin>>s;
    while(s--){
        int x;
        cin>>x;
        if(find(x)){
            cout<<"Yes"<<endl;
        }
        else{
            cout<<"No"<<endl;
        }
    }
        
    return 0;
}