挑战程序设计竞赛——抽签进阶
限制条件:
- 1 <= n <= 1000
- 1 <= m <= 1e8
- 1 <= ki <= 1e8
首先,进阶版抽签,与普通法版的区别是:n的数量级变大,导致我们不能暴力枚举。要用更优的枚举方法 二分搜索 O(long n)
其次,如果你们对二分搜索不明白的话,我后续会出专题,感兴趣的可以关注我。
这里我来讲一下这题为什么用二分,有一下两点:
- 多个数的枚举
- n <= 1000
思路: 原始枚举公式是ka + kb + kc + kd = =m,现在把公式变形,变成kc + kd = m - ka - kb,两重循环枚举kc + kd,再用两重循环枚举 + 二分搜索 m - ka - kb
以下是代码部分:
#include <bits/stdc++.h>
using namespace std;
const int maxl = 1005;
int n, m, k[maxl], a[maxl * maxl], b[maxl * maxl];
bool flag = false;
bool binary_search(int x){//二分搜索
int l = 0, r = n * n - 1;
while (l <= r){
int mid = l + (r - l) / 2;
if (a[mid] == x)
return true;
else if (a[mid] < x)
l = mid + 1;
else
r = mid - 1;
}
return false;
}
int main(){
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++){
a[i * n + j] = k[i] + k[j];
}
sort(a, a+ n * n);
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++){
int d = m - k[i] - k[j];
if (binary_search(d)){
flag = true;
}
}
cout << (flag ? "Yes" : "No"); //三目运算