[USACO1.4]等差数列 Arithmetic Progressions

最新推荐文章于 2021-12-04 22:05:18 发布

AAA_Ljw

最新推荐文章于 2021-12-04 22:05:18 发布

阅读量234

点赞数

本文链接：https://blog.csdn.net/weixin_43909855/article/details/84843916

版权

[USACO1.4]Arithmetic Progressions

题目描述

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

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

输入输出格式

输入格式：

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

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

输出格式：

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

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

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

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

输入输出样例

输入样例#1：

5
7

输出样例#1：

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

题目解析

枚举双平方数集合
再枚举该等差数列的前两位（a,a+b）即可推出公差
从而推出整个等差数列
判断等差数列是否都在双平方数集合
ans排序后输出即可

代码

#include<bits/stdc++.h>
using namespace std;
struct A
{
	int a,b;
}ans[10005];
int n,m,ai,lmt,cha,last;
bool a[200000];
bool cmp(A a,A b)
{
	return a.b==b.b?a.a<b.a:a.b<b.b;
}
bool check(int x,int y)
{
	for(int i=0;i<n;i++)
	 if(!a[x+y*i])
	  return false;
	return true;
}
int main()
{
	cin>>n>>m;
	for(int i=0;i<=m;i++)
	 for(int j=i;j<=m;j++)
	  a[i*i+j*j]=1;//枚举双平方数集合
	lmt=m*m*2;//枚举的上限
	for(int i=0;i<=lmt;i++)
	{
	  if(a[i])
	   for(int j=i+1;j<=lmt;j++)
	    if(a[j])
	    {
	      cha=j-i;//公差
	      last=i+cha*(n-1);//等差数列的最后一个数
	      if(last>lmt) break;//如果最后一个数大于上限，则跳出 
	      if(check(i,cha))//判断是否合法
	       ans[++ai].a=i,ans[ai].b=cha;//记录答案
		}
	}
	sort(ans+1,ans+1+ai,cmp);//排序
	for(int i=1;i<=ai;i++)
	 cout<<ans[i].a<<" "<<ans[i].b<<endl;
	if(ai==0)
	 cout<<"NONE";
}