题意:
你的朋友提议玩一个游戏:将写有数字的n个纸片放入口袋中,你可以从口袋中抽取4次纸片,每次记下纸片上的数字后都将其放回口袋。如果这4个数字的和是m,就是你赢,否则就是你的朋友赢。请编写一个程序,纸片所写数据为:K1,K2,K3,K4.....Kn,判断你是否有赢的可能性。(限制条件:1<=n<=50,1<=m<=108,1<=K1...Kn<=108)
直接暴力循环:
代码:
int n,m,data[55];
while(~scanf("%d%d",&n,&m))
{
for(int i=0;i<n;i++)
scanf("%d",&data[i]);
bool Judge = false;
for(int a=0;a<n;a++)
for(int b=0;b<n;b++)
for(int c=0;c<n;c++)
for(int d=0;d<n;d++)
if(data[a] + data[b] + data[c] + data[d] == m)
Judge = true;
if(Judge)
printf("Yes\n");
else
printf("No\n");
}
结果:
(对于第一组数据来说:1+3+3+3 = 10,第二组数据中不存在和为9的方案)
一般竞赛的时间限制为1s。对于循环来说:10万次绰绰有余,100万次勉强,1000万就很危险了。
题目要求是n<=50。上面算法的复杂度为n4。循环次数为50*50*50*50 = 625 0000。情况不是很好。
那么问题来了,怎么优化呢?
先想一下:一次一次加的话,还不如最后直接匹配。例如,抽出了3个数,那么第四次抽出的如果不可能是m-k1-k2-k3,那么就不能赢。时间复杂度为:n3log(n)。
这里查询建议用二分法查询,比较快,C++有相关查询函数binary_search(),这样就不能用数组了,需要用vector。
代码:
int n,m,data;
vector<int> v;
while(~scanf("%d%d",&n,&m))
{
for(int i=0;i<n;i++)
{
scanf("%d",&data);
v.push_back(data);
}
bool Judge = false;
for(int a=0;a<n;a++)
for(int b=0;b<n;b++)
for(int c=0;c<n;c++)
{
int temp = m - v[a] - v[b] - v[c];
Judge = binary_search(v.begin(),v.end(),temp);
}
if(Judge)
printf("Yes\n");
else
printf("No\n");
}
结果:
这样的话,那能否计算任意两个数的值,新建一个序列,再去查找新序列中是否存在两个数的和等于m呢?这样时间复杂度为:n2log(n)。
优化最终代码:
int n,m,data;
vector<int> v;
while(~scanf("%d%d",&n,&m))
{
v.clear();
for(int i=0;i<n;i++)
{
scanf("%d",&data);
v.push_back(data);
}
// 创建新序列
vector<int>vTemp;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
vTemp.push_back(v[i]*v[j]);
bool Judge = false;
for(int i=0;i<n*n;i++)
{
int temp = m - vTemp[i];
if(binary_search(v.begin(),v.end(),temp))
Judge = true;
}
if(Judge)
printf("Yes\n");
else
printf("No\n");
}
结果:
尽量减少循环的嵌套次数,然后是循环次数。
PS:最近看的竞赛书籍是《挑战程序设计竞赛》,都是高手的经验总结。下载网址:http://www.gaofi.cn/book/detail/402