杭电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>