USACO1.4 等差数列 Arithmetic Progressions

等差数列

题目描述

一个等差数列是一个能表示成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.