算法课

/*你的朋友提议玩一个游戏：将写有数字的n个纸片放入口袋中．你可以从口袋中抽取4次纸片，每次记下纸片上的数字后都将其放回口袋中如果这4个数字的和是m，就是你赢，否则就是你的朋友赢。你挑战了好几回，结果一次也没赢过，于是怒而撕破口袋，取出所有纸片，检查自己是否真的有赢的可能性。请你编写一个程序，判断当纸片上所写的数字是k1，k2，…，kn时，是否存在抽取4次和为m的方案。如果存在，输出彻；否则，输出NO.
限制条件:   1≤n ≤ 50;   1 ≤ m ≤ 10^8;   1 ≤ Ki ≤ 10^8

输入样例： 3 10 1 3 5             (格式-n,m, k1,k2,..kn)
输出样例：YES

输入样例：3 9 1 3 5 
输出样例：NO
*/

#include<iostream>
#include<algorithm>

using namespace std;

int n, m, i, j;

int kk[1000000];//开大点防止爆掉        
int bs(int b){
    int left = 0, right = n*n-1;
    while(left < right){
        int k = left+(right-left)/2;
        if(kk[k] == b)
            return 1;
        if( m >kk[k])
            left = k+1;
        else
            right = k;
                
    }

    return 0;
}

int main(){
    
    cin >> n >> m;
    int a[n];
    for(i = 0; i < n; i++){
        cin >> a[i];
    }
    for ( i=0; i < n; i++)
        for  ( j=0; j < n; j++)
            kk[i*n + j] = a[i]+a[j];//枚举组合
    sort(kk, kk+n*n); 
    for(i = 0; i < n; i++)
        for(j = 0; j < n; j++){
                if( bs(m-a[i]-a[j]) ){
                    cout << "YES";
                    return 0;
                }
            } 
        
        
    cout << "NO";
    return 0;
}