杭电ACM2058--The sum problem

http://acm.hdu.edu.cn/showproblem.php?pid=2058

 

以为简单的穷举就完了，结果是一直Time Limit Exceeded。。

这是代码：

#include <stdio.h>#include <stdlib.h>#include <math.h>int main(){int a,b;int i,j,k;while (scanf("%d%d",&a,&b)!=EOF&&(a!=0||b!=0)){for (i=1;i<=a;i++){k = 0;for (j=i;j<=i+(int)sqrt((double)(2*b))+1;j++){k = k+j;if (k == b)printf("[%d,%d]\n",i,j); }}printf("\n");}return 0;} 

 

穷举是穷举，还不能任意穷举，要不超时！

设初始值i，个数为j的数列满足要求，则（i+i+j-1)*j/2=m

==>(2*i-1+j)*j=2m

所以j肯定小于等于sqrt(2*m)，穷举。

 

之后才发现数据量之大，然后找到了解决办法。

 

<span style="font-size:24px;">#include <stdio.h>#include <stdlib.h>#include <math.h>int main(){int a,b;int i,j,k;while (scanf("%d%d",&a,&b)!=EOF&&a&&b){for (j=(int)sqrt((double)(2*b));j>=1;j--){i = (2*b/j+1-j)/2;if (b == (2*i+j-1)*j/2)printf("[%d,%d]\n",i,i+j-1);}printf("\n");}return 0;}</span>