洛谷 P1214 等差数列

https://www.luogu.org/problemnew/show/P1214

首先暴力枚举可以凑出来的数，对于每个数进行标记。

对于每一个等差数列，当我们知道前两个数后即可以得出整个序列，那么我们就要判断序列中的每一个数是否被标记过，合法记录答案。

最后对于答案进行排序，输出。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
using namespace std;
#define LL long long
int a[],tot,cnt,n,m;
bool vis[];
struct ahah
{
    int s,k;
} ans[];
bool cmp(ahah a,ahah b)
{
    return a.k<b.k;
}
void prepare(int x)
{
    for(int i=; i<=x; i++)
    {
        for(int j=; j<=x; j++)
        {
            a[++tot]=i*i+j*j;
        }
    }
    sort(a+,a++tot);
    tot=unique(a+,a++tot)-a-;
    for(int i=; i<=tot; i++)vis[a[i]]=;
}
int main()
{
 
    scanf("%d%d",&n,&m);
    prepare(m);
    for(int i=; i<=tot; i++)
    {
        if(a[i]>m*m*)break;
        for(int j=i+; j<=tot; j++)
        {
            int d=a[j]-a[i];
            if(a[i]+(n-)*d > m*m*)break;
            for(int k=; k<=n; k++)
            {
                if(!vis[a[i]+d*(k-)])break;
                if(k==n)ans[++cnt].s=a[i],ans[cnt].k=d;
            }
        }
    }
    if(cnt==)printf("NONE");
    else
    {
        sort(ans+,ans++cnt,cmp);
        for(int i=; i<=cnt; i++)printf("%d %d\n",ans[i].s,ans[i].k);
    }
}

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
using namespace std;
#define LL long long
int a[],tot,cnt,n,m;
bool vis[];
struct ahah
{
    int s,k;
} ans[];
bool cmp(ahah a,ahah b)
{
    return a.k<b.k;
}
void prepare(int x)
{
    for(int i=; i<=x; i++)
    {
        for(int j=; j<=x; j++)
        {
            a[++tot]=i*i+j*j;
        }
    }
    sort(a+,a++tot);
    tot=unique(a+,a++tot)-a-;
    for(int i=; i<=tot; i++)vis[a[i]]=;
}
int main()
{
 
    scanf("%d%d",&n,&m);
    prepare(m);
    for(int i=; i<=tot; i++)
    {
        if(a[i]>m*m*)break;
        for(int j=i+; j<=tot; j++)
        {
            int d=a[j]-a[i];
            if(a[i]+(n-)*d > m*m*)break;
            for(int k=; k<=n; k++)
            {
                if(!vis[a[i]+d*(k-)])break;
                if(k==n)ans[++cnt].s=a[i],ans[cnt].k=d;
            }
        }
    }
    if(cnt==)printf("NONE");
    else
    {
        sort(ans+,ans++cnt,cmp);
        for(int i=; i<=cnt; i++)printf("%d %d\n",ans[i].s,ans[i].k);
    }
}