你的朋友提议玩一个游戏:将写有数字的n个纸片放入口袋中,你可以从口袋中抽取4次纸片,每次记下纸片上的数字后都将其放回口袋中。如果这4个数字的和是m,就是你赢,否则就是你的朋友赢。你挑战了好几回,结果一次也没赢过,于是怒而撕破口袋,取出所有纸片,检查自己是否真的有赢的可能性。请你编写一个程序,判断当纸片上所写的数字是k1, k2, …, kn时,是否存在抽取4次和为m的方案。如果存在,输出Yes;否则,输出No。(1<=n<=1000,1<=m<=1e8,1<=k<=1e8)
题解:枚举先抽前两次的和的所有可能, 并存入一个数组a。二分查找后两次之和是否可能等于数组a的值。(n^2logn)
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1e6 + 10;
int a[maxn], n, k[maxn];
/*二分查找*/
bool binary_search(int x) {
int l = 0, r = n * n;
while (r - l >= 1) {
int i = (l + r) / 2;
if (a[i] == x)
return true;
else if (a[i] < x)
l = i + 1;
else
r = i;
}
}
int main()
{
int m;
int ok = 0;
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[n * i + j] = k[i] + k[j];
sort(a, a + n * n);
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++) {
if (binary_search(m - k[i] - k[j])) {
ok = 1;
break;
}
}
if (ok)
printf("Yes\n");
else
printf("No\n");
return 0;
}