查找等式的解

查找等式的解

题目描述：
有三个数列 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 500using 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;}