等差数列
题目描述
一个等差数列是一个能表示成a, a+b, a+2b,..., a+nb (n=0,1,2,3,...)的数列。
在这个问题中a是一个非负的整数,b是正整数。写一个程序来找出在双平方数集合(双平方数集合是所有能表示成p的平方 + q的平方的数的集合,其中p和q为非负整数)S中长度为n的等差数列。
分析:把所有双平方数集合求出,bool数组记录这个数是否在集合里,然后把具体值存入数组x(排序),由于题目要求输出先按b排序,所以先枚举b再枚举a,范围自己想,只要满足a,a+b,a+2b,……,a+nb,就可以直接输出,不过还要加个优化,如果当前枚举的a加上nb后大于集合中的最大双平方数就break。
时间复杂度:O(n ^2logn)
代码
const
maxn=10000000;
var
a:array[0..maxn] of boolean;
b:array[0..maxn] of longint;
i,j,k,n,p,m,sum,x,l:longint;
procedure qsort(l,r:longint);
var
i,j,k:longint;
begin
if l>=r then exit;
i:=l;j:=r;
k:=b[(l+r) div 2];
repeat
while b[i]<k do inc(i);
while b[j]>k do dec(j);
if i<=j then
begin
b[0]:=b[i];b[i]:=b[j];b[j]:=b[0];
inc(i);dec(j);
end;
until i>j;
qsort(l,j);qsort(i,r);
end;
begin
readln(n);
readln(m);
for i:=0 to m do
for j:=i to m do
begin
x:=sqr(i)+sqr(j);
if not a[x] then
begin
a[x]:=true;
inc(l);
b[l]:=x;
end;
end;
qsort(1,l);
for i:=1 to b[l] div (n-1) do
for j:=1 to l-1 do
begin
p:=1;
if b[j]+i*(n-1)>b[l] then break;
for k:=1 to n-1 do
if not a[b[j]+k*i] then break else inc(p);
if p=n then
begin
x:=1;
writeln(b[j],' ',i);
end;
end;
if x<>1 then writeln('NONE');
end.