题目链接:https://cn.vjudge.net/contest/279225#problem/K、
题目大意:输入一个整数n,找到一个连续的序列使这个序列的每个元素的平方加起来等于n,输出满足条件的序列数,按元素的多少降序输出元素数和这些元素。
思路:这道题开始的时候是每次都把这个序列的平方和加一遍,然后超时了。后来改,改成sum是在移动尺子的时候加right^ 2,或减left^ 2,这个尺取就一直到左右端点到达n开根为止。
#include <stdio.h>
#include <math.h>
#include <algorithm>
#define INF 10000005
using namespace std;
typedef long long ll;
struct node
{
ll answer,l,r;//这里的answer是区间长度,l是左端点,r是右端点
} ans[INF];
bool cmp(node n1,node n2)
{
return n1.answer>n2.answer;
}
int main()
{
ll n;
while(~scanf("%lld",&n))
{
ll left=1,right=1,sum=0,x=0;
ll s=sqrt(n);
while(left<=s)//左端点在s范围内
{
while(sum<n&&right-1<=s)//这里是右端点在s范围内
{
sum+=right*right;
right++;//因为这里每次都要跑一次right++,所以实际sum的区间长度是[left,right-1]
}
if(sum==n)
{
ans[x].answer=right-1-left+1;//右端点-左端点-1是区间长度
ans[x].l=left;//左端点
ans[x].r=right-1;//右端点
x++;
sum+=right*right;//这儿呢必须要有这个,就是说sum=n之后,判断之后的还有没有序列满足条件,就直接再加下一个数,然后sum就会大于n了,再接着跑下边那一个if(再说一下就是这儿的right已经是区间右端点的下一个数了,所以直接加right就ok)
right++;//保证right一直是区间右端点的右边一个数
}
if(sum>n)
{
sum-=left*left;
left++;
}
}
sort(ans,ans+x,cmp);
printf("%lld\n",x);
for(ll i=0; i<x; i++)
{
printf("%lld ",ans[i].answer);
for(ll j=ans[i].l; j<=ans[i].r; j++)
printf("%lld%c",j," \n"[j==ans[i].r]);
}
}
return 0;
}