题目描述:
你的朋友提议玩一个游戏:将写有数字的n(1<= n <= 1000)个纸片放入口袋中,你可以从口袋中抽取4次纸片,每次记下纸片上的数字后都将其放回口袋中。如果这4个数字的和是m(1<= m <= 10^8),就是你赢,否则就是你的朋友赢。你挑战了好几回,结果一次也没赢过,于是怒而撕破口袋,取出所有纸片,检查自己是否真的有赢的可能性。请你编写一个程序,判断当纸片上所写的数字是k1,k2,...,kn(1<= ki <= 10^8)时,是否存在抽取4次和为m的方案。如果存在,输入Yes;否则,输出No。
输入:
多组测试数据
第一行为n,接下来是n个整数。
最后一行一个m。
输出:
每个数据输出一行,对于每个m,如果存在赢的可能性,输出Yes;否则,输出No。
样例输入:
3
1 3 5
10
样例输出:
Yes
思想:
这道题目不难,单纯的模拟就可以过,但还要附加上一些条件,否则所花时间超过1s,循环中用剪枝是可以想到的,但我们还可以去重,这样就可以保证时间复杂度低,且代码编写难度不高了。
代码示例:
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int a[1000];
int main()
{
int n, m;
while (~scanf("%d", &n)) //使用scanf()是因为题目有大量输入,用cin会慢一些
{
for (int i = 0; i < n; i++) scanf("%d", a + i);
sort(a, a + n);
n = unique(a, a + n) - a; //去重
scanf("%d", &m);
bool flage = false;
int temp = 0;
for (int i = 0; i < n; i++)
{
temp += *(a + i); //这样写是为了不必下面每次比较时,几个数都要相加一遍。省时间
if (temp>m) break; //剪枝
for (int j = i; j < n; j++)
{
temp += *(a + j);
if (temp>m) { temp -= *(a + j); break; } //注意break时,也要去掉当前循环的值,以使得上一循环没有这个循环的值,否则程序出错
for (int k = j; k < n; k++)
{
temp += *(a + k);
if (temp>m) { temp -= *(a + k); break; }
for (int r = k; r < n; r++)
{
temp += *(a + r);
if (temp>m) { temp -= *(a + r); break; }
if (temp == m) { flage = true; goto qqq; } goto使程序没那么多判断
temp -= *(a + r);
}
temp -= *(a + k);
}
temp -= *(a + j);
}
temp -= *(a + i);
}
qqq:
if (flage) printf("Yes\n");
else printf("No\n");
}
}