《挑战程序设计竞赛》 P2 题目：抽签优化，优化，再优化

最新推荐文章于 2022-02-09 12:57:02 发布

原创最新推荐文章于 2022-02-09 12:57:02 发布 · 358 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

信息学奥赛（NOIP）同时被 2 个专栏收录

34 篇文章

订阅专栏

《挑战程序设计竞赛》

13 篇文章

订阅专栏

本文介绍了一种通过预处理和二分查找实现的高效四数求和算法。该算法首先将所有可能的两数之和进行排序并存储，然后对于每一对数，使用二分查找来判断是否存在第三个数使得三个数之和等于指定的目标值。

样例：
输入：
3
10
1 3 5

输出：
Yes

输入：
3
9
1 3 5

输出：
No

#include <bits/stdc++.h>

using namespace std;
const  int MAX_N=51;
int k[MAX_N];

int kk[MAX_N * MAX_N];

int main()
{
      int n, m;  //n个数。取四次， 要求和为m；
      cin >>n >>m;
      for(int i=0;i<n;i++)
            cin >> k[i];
      for(int i=0;i<n;i++)
            for(int j=0;j<n;j++){
                  kk[ i*n + j] = k[i] + k[j];
            }

      sort(kk,kk+n*n);

      for(int i=0;i<n;i++)
            for(int j=0;j<n;j++){
                  if( binary_search( kk, kk+ n*n, m-k[i]-k[j] ) ){
                        cout << "Yes";
                        return 0;
                  }
            }
     cout << "No";
    return 0;
}