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

最新推荐文章于 2023-02-13 13:09:50 发布

ETwinner

最新推荐文章于 2023-02-13 13:09:50 发布

阅读量168

点赞数

本文链接：https://blog.csdn.net/qq_42920122/article/details/88777493

版权

该博客介绍了如何解决洛谷P1214等差数列问题，避免直接枚举a和b导致的时间超限。通过枚举数列的前两项并排序双重平方数，计算公差b，再检查数列项是否为双重平方数，最后进行排序以符合题目要求。

摘要由CSDN通过智能技术生成

题目描述 https://www.luogu.org/problemnew/show/P1214
如果直接枚举a,b会超时，可以枚举数列的前两项，将双重平方数【注意：小心重复】从大到小排序，枚举前两项，可得到公差b,在依次-b，判断这个数是否是双重平方数，最后按题目要求排序。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int a[200010],tot,num,n,m;
bool fl[200010];
struct node
{
    int a,b;
}ans[200010];
bool cmp(int x,int y)
{
    return x>y;
}
bool tmp(node x,node y)
{
    if(x.a!=y.a) return x.a<y.a;
    return x.b<y.b;
}
int main( )
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<=m;i++)
      for(int j=i;j<=m;j++)
	    if(!fl[i*i+j*j]) a[++tot]=i*i+j*j,fl[i*i+j*j]=1;
    sort(a+1,a+tot+1,cmp);
    for(int i=1;i<=tot-n+1;i++)
      for(int j=i+1;j<