c语言程序循环运行速度优化方案
问题背景
题目如下:
中国人民大学2018年秋季学期程序设计期末考试第3题
用下述直接循环n次的方法,虽然能够得到结果,但是运行速度太慢,例如输入111111要38.5s,输入1111111要四分钟以上,无法通过考试验收系统的时间限制。所以考虑减少循环次数和不必要的循环。
#include<stdio.h>
int main()
{
long long n,h,x,y;
scanf("%lld",&n);
long long a,b;
for(a=1;a<=n/2;a++)
{
for(b=2;b<n;b++)
{
x=a+b,y=b-a+1;
h=x*y/2;
if(h==n)
{
printf("[%lld, %lld]\n",a,b);
}
}
}
return 0;
}
优化思路
在本站查找到这篇文章:
原文链接
该文的思路为通过求出最大枚举范围来缩小循环次数,适用于欲减少循环次数的程序中。
设等差数列:
an=a+(n-1)*d (这里首项为a,公差d=1,第n项为an,前n项和为sn)
a1=a
an=a+n-1
sn=(a1+an)n/2=(2a-1+n)*n/2
再回到这个编程上来:
我们的输入数据其实就是sn,需要找到以a开始的n个连续的递增数列使得和为sn。
这里我们可以用循环来判定,给定一个n,sn已知,就可以求出a,如果a为正整数那么就可以找到等差数列的首项,加上n给定,d=1,那么就可以写出这个和式子。
进一步优化提高程序效率:这里的n无须一直从2开始枚举下去,可以由sn=(2a-1+n)*n/2,所以a=sn/n-n/2+1/2,该式子为递减函数,n越大,a越小,而a最小为1,故另a=1时可确定n的最大范围。
令a=1,得二元一次方程(1/2+n/2)n=sn,即n^2+n-2sn=0,可得方程两个根中取较大的根n=0.5*(-1+sqrt(1+8*sn)),从而确定n的最大枚举范围。
优化代码及结果
原题要求区间从小到大输出,参考文献的输出与题目不符,而且输出次序也与题目不符。作者在参考文献代码的基础上做出了逆序输出的修改,并且修改成题目要求的输出格式。
#include<stdio.h>
#include<iostream>
#include<math.h>
using namespace std;
int main()
{
int n,s,an,i,ans,cnt,fg,c[100000]={0},d[100000]={0};
double a1;
while(scanf("%d",&n)!=EOF)
{
ans=0;
int mm=0.5*(-1+sqrt(1+8*n)); //枚举的最大范围
for(cnt=mm;cnt>=2;cnt--) //逆序循环
{
a1=(2*n+cnt-cnt*cnt)*1.0/(2.0*cnt);
if(int(a1)==a1&&a1>0)
{
printf("[%d",int(a1));
for(i=1;i<=cnt-1;i++)
{
if(i==cnt-1)
{
printf(",%d]",int(a1+i));
}
}
printf("\n");
ans++;
}
}
break;
}
return 0;
}
结果:
111111只需3.952s
1111111只需要2.731s
可见优化效果良好
鸣谢
感谢妹妹小贾同学提出的问题