题目描述
肥成在玩一个游戏:将写有数字的n个纸片放入口袋,你可以从口袋中抽取4次纸片,每次记下纸片上的数字后都将其放回口袋中。如果这四个数字的和是m,就是你赢,否则就是你的朋友赢。
你挑战了好几回,结果一次也没赢过,于是怒而撕破口袋,取出所有纸片,检查自己是否真的有赢的可能性。
纸片上所写的数字是k[1],k[2],k[3],...,k[n],请你编一个程序,判断是否存在抽取四次,四个数的和为m的可能,如果存在,输出Yes;否则,输出No。
输入格式
第一行,一个整数n
第二行,一个整数m
第三行,n个整数K[i]
数据范围
对于50%的数据有1≤n≤100,1≤K[i]≤1000,1≤m≤1000。
对于80%的数据有1≤n≤100,1≤K[i]≤10^6,1≤m≤10^6。
对于100%的数据有1≤n≤1000,1≤K[i]≤10^8,1≤m≤10^8。
输出格式
输出一行,如果存在,输出Yes,否则输出No。
样例解释
对于样例一,方案是1、3、3、3
对于样例二,方案是3、3、3、3
样例输入
3
10
1 3 5
样例输出
Yes
样例输入
4
12
3 3 4 4
样例输出
Yes
题目计算量不大,合理运用枚举,在条件不可能成立时”剪枝“(深搜串台??),可以大大提高算法效率
#include <iostream>
#include <algorithm>
int main() {
register unsigned short n;
register unsigned int m, * k;
std::cin >> n >> m;
k = new unsigned int[n];
for (register unsigned short i = 0; i < n; i++) {
std::cin >> k[i];
}
std::sort(k, k + n);
for (register unsigned short i = 0; i < n; i++) {
if (k[i] > m) {
break;
}
for (register unsigned short j = 0; j < n; j++) {
if (k[i] + k[j] > m) {
break;
}
for (register unsigned short l = 0; l < n; l++) {
if (k[i] + k[j] + k[l] > m) {
break;
}
for (register unsigned short o = 0; o < n; o++) {
if (k[i] + k[j] + k[l] + k[o] > m) {
break;
}
else if (k[i] + k[j] + k[l] + k[o] == m) {
std::cout << "Yes";
return 0;
}
}
}
}
}
std::cout << "No";
return 0;
}
可能有人怀疑,在C++23标准发布的今天,register储存类是否还有作用,确实,现在编译器越来越聪明,register按照目前的趋势在C++中已被弃用,但一些较老编译器或没有优化的在线批阅系统中,还是有用处的。
register 优化前 vs 优化后 对比如下: