打卡信奥刷题（2404）用C++实现信奥 P1214 [USACO1.4] 等差数列 Arithmetic Progressions

最新推荐文章于 2026-07-03 19:39:28 发布

原创最新推荐文章于 2026-07-03 19:39:28 发布 · 336 阅读

10 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

C++

P1214 [USACO1.4] 等差数列 Arithmetic Progressions

题目描述

一个等差数列是一个能表示成 $\dots ,a+nb\space (n \in \mathbb N)$ 的数列。

在这个问题中 $a$ 是一个非负的整数， $b$ 是正整数。
写一个程序来找出在双平方数集合：
${x∣x=p2+q2∧p,q∈N∩[0,m]}\{ x | x = p^2 + q^2 \wedge p,q \in \mathbb N \cap [0,m]\}$
中长度为 $n$ 的等差数列。

输入格式

第一行一个正整数 $n$ ，表示要找的数列长度。
第二行一个非负整数 $m$ ，表示 $p, q$ 的上界。

输出格式

如果没有找到数列，输出 NONE。

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

这些行应该以 $b$ 为第一关键字， $a$ 为第二关键字升序排序。

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

输入输出样例 #1

输入 #1

5
7

输出 #1

说明/提示

【数据范围】
对于 $100%100\%$ 的数据， $\le n \le 25$ ， $\le m \le 250$ 。

题目翻译来自NOCOW。

USACO Training Section 1.4

C++实现

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std;
int n,m,p,q,s[125000];
bool num[125000];
int main()
{
    scanf("%d%d",&n,&m);
    memset(num,false,sizeof(num));  //初始化 
    int i,j,a,b,l,cnt=0;
    for(i=0;i<=m;i++)
    for(j=0;j<=m;j++)
    num[i*i+j*j]=1;        //存所有双平方数 
    l=0;
    for(i=0;i<=125000;i++)
    if(num[i]) s[++l]=i;   //把所有数集中
    for(b=1;b<s[l];b++)   //从小到大搜，就不用排序了 
    for(i=1;i<l-1;i++)    //因为n>=3，所以运行到倒数第三个就可以了
    {
    if(s[i]+(n-1)*b>s[l]) break;    //判断优化 
    for(a=2;a<=n;a++)
    if(!num[s[i]+(a-1)*b]) break;   //搜素等差数列 
    if(a==n+1&&num[s[i]+(n-1)*b])   //判断是否满足要求 
    printf("%d %d\n",s[i],b),cnt++; 
    }
    if(!cnt) printf("NONE");    //无答案则输出NONE
    return 0;
}