查找等式的解

查找等式的解

题目描述：
有三个数列 a b c ，有一个数 x，计算是否存在a b c 中的在3个整数a b c 使得a+b+c=x成立。
输入：
第一行为三个整数，分别为a b c 数列的长度
第二行到第四行分别为a b c 数列中的值
第五行为要测试的 x 的个数 n
接下来的 n 行为 n 个测试数据
输出：
n行输出，如果能找到这样的 a b c ，输出yes， 如果不存在这样的 a b c 输出no
思路：
先计算出a+b的值，然后在所有a+b中二分查找是否有满足的值使得满足当前的c和x.

#include <iostream>
#include <algorithm>
#define MAX 500
using namespace std;

int main()
{
    //输入数据
    int A[MAX], B[MAX], C[MAX];
    int sum[MAX * MAX];
    int la, lb, lc;
    int N, X[MAX];
    bool answer[MAX];
    cin>>la>>lb>>lc;
    for(int i = 0 ; i < la ; i++)
        cin>>A[i];
    for(int i = 0 ; i < lb ; i++)
        cin>>B[i];
    for(int i = 0 ; i < lc ; i++)
        cin>>C[i];
    cin>>N;
    for(int i = 0 ; i < N ; i++)
        cin>>X[i];
    int k = 0;
    for(int i = 0 ; i < la ; i++)
        for(int j = 0 ; j < lb ; j++)
            sum[k++] = A[i] * B[j];
    sort(sum, sum+k);
    int l, r, S;
    for(int n = 0 ; n < N ; n++){
        answer[n] = false;
        for(int i = 0 ; i < lc ; i++){
            S = X[n] - C[i];
            l = 0, r = k - l;
            while(r >= l){
                int m = (r + l) / 2;
                if(sum[m] == S){
                    answer[n] = true;
                    break;
                }else if(sum[m] > S){
                    r = m-1;
                }else{
                    l = m+1;
                }
            }
            if(answer[n])
                break;
        }
    }
    for(int i = 0 ; i < N ; i++)
        if(answer[i])
            cout<<"YES\n";
        else
            cout<<"NO\n";
    system("pause");
    return 0;
}