Arithmetic Progressions

链接

分析：首先预处理数据集，然后将数据集排序，我们对于当前位置，去搜索是否存在满足条件的等差数列即可，之前一直TLE，完全是map的锅，太慢了

#include "iostream"
#include "cstdio"
#include "cstring"
#include "map"
#include "algorithm"
#include "vector"
using namespace std;
const int maxn=2e5;
struct Node{
    int x,y;
};
int mp[maxn];
vector<int> h;
vector<Node> p;
int n,m,len;
bool cmp(Node a,Node b){
    if(a.y==b.y){
        return a.x<b.x;
    }
    return a.y<b.y;
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<=m;i++){
        for(int j=i;j<=m;j++){
            int num=i*i+j*j;
            if(!mp[num]){
                mp[num]=1;
                h.push_back(num);
            }
        }
    }
    sort(h.begin(),h.end());
    len=h.size();
    int cnt=0;
    for(int i=0;i<len-n+1;i++){
        for(int j=i+1;j<len-n+2;j++){
            int cha=h[j]-h[i];
            int flag=0;
            if(h[i]+(n-1)*cha>h[len-1])  break;
            for(int k=1;k<=n-2;k++){
                if(h[i]+(k+1)*cha>h[len-1]){
                    flag=2; break;
                }
                if(!mp[h[j]+k*cha]){
                    flag=1; break;
                }
            }
            if(flag==2)  break;
            if(!flag){
                cnt++;
                Node tt;
                tt.x=h[i],tt.y=cha;
                p.push_back(tt);
            }
        }
    }
    if(cnt==0){
        cout<<"NONE"<<endl;
    }else{
        sort(p.begin(),p.end(),cmp);
        //cout<<cnt<<endl;
        for(int i=0;i<p.size();i++)
            cout<<p[i].x<<" "<<p[i].y<<endl;
    }
    return 0;
}