54 等差数列

问题描述 :

一个等差数列是一个能表示成a, a+b, a+2b,..., a+nb (n=0,1,2,3,...) 在这个问题中a是一个非负的整数，b是正整数。

写一个程序来找出在双平方数集合S中长度为n的等差数列。双平方数集合是所有能表示成p²＋q²的数的集合。

输入说明 :

第一行: 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

 

思想:用一个数组来保存所有平方和的结果，然后遍历a，b，如果符合，那么输出。

#include <stdio.h>
#include <stdlib.h>




int main()
{
    int num,m;
    int count;
    int flag = 0;
    int i,j;
    int cur =0;
    int a,b;
    scanf("%d %d",&num,&m);
    int nums[20000];
    for(i =0;i<=m;i++){
        for(j =0;j<=m;j++){
            cur = i*i+j*j;
            nums[cur] = 1;
        }
    }
    for(b=1;b<=(2*m*m)/(num-1);b++){ //b的范围是在1-最大的值的平方相加/（需求的等差数列每组的个数-1）
        for(a=0;a<m*m;a++){
            for(j=0;j<num;j++){
                if(nums[a+b*j]!=1){
                    break;
                }
            }
            if(j==num){
                flag=1;
                printf("%d %d\n",a,b);
            }
        }
    }
    if(flag==0){
        printf("NONE\n");
    }

   
    return 0;
}