等差数列 USACO

题目描述

一个等差数列是一个能表示成a, a+b, a+2b,..., a+nb (n=0,1,2,3,...)的数列。

在这个问题中a是一个非负的整数，b是正整数。写一个程序来找出在双平方数集合S中长度为n的等差数列。双平方数集合是所有能表示成p^2＋q^2的数的集合。

输入格式

第一行: N(3<= N<=25),要找的等差数列的长度。

第二行: M(1<= M<=250),搜索双平方数的上界0 <= p,q <= M。

输出格式

如果没有找到数列,输出`NONE'。

如果找到了，输出一行或多行, 每行由二个整数组成a,b。

这些行应该先按b排序再按a排序。

所求的等差数列将不会多于10,000个。

输入样例 复制

5
7

输出样例 复制

1 4
37 4
2 8
29 8
1 12
5 12
13 12
17 12
5 20
2 24

算法解析

可以枚举双平方数，做好标记，再枚举b，判断是否存在公差为b的等差数列。

代码如下

#include<bits/stdc++.h>
using namespace std;
int n,m,maxn=0;
bool f[1000000];
int main()
{
    cin>>n>>m;
    n--;
    for(int i=0;i<=m;i++)
        for(int j=0;j<=m;j++)
        {
            f[i*i+j*j]=1;
            maxn=max(maxn,i*i+j*j);
        }
    bool w=0;
    for(int b=1;b<=maxn;b++)
        for(int a=0;a<=maxn-n*b;a++)
            if(f[a]==1&&f[a+n*b]==1)
            {
                bool flag=0;
                for(int k=1;k<n;k++)
                    if(!f[a+k*b])
                    {
                        flag=1;
                        break;
                    }
                if(!flag)
                {
 
                    cout<<a<<' '<<b<<endl;
                    w=1;
                }
            }
    if(w==0)
        cout<<"NONE"<<endl;
 
    return 0;
}