2019 ICPC 上海站网络赛 K.Peekaboo (圆上整点)

https://nanti.jisuanke.com/t/41421
题意：给定平面上两个点到原点之间的距离$a$和$b$以及这两点之间的距离$c$，且这两点的坐标均是整数，求该两点可能的坐标，按字典序从小到大输出。

给定距离$a$和距离$b$其实就是画了两个以原点为圆心半径分别为$a$和$b$的圆，然后分别找到这两个圆上的坐标为整数的点$p_1$和$p_2$，使得$p_1$和$p_2$之间的距离为$c$。

圆上的整点可以参考[HAOI2008] 圆上的整点(BZOJ 1041或洛谷2508)的题解，这里参考的是Taday_Bule_Rainbow的题解，网上的题解大多提到了这个神仙视频可以有助于理解。

找到圆上的整点剩下的就是暴力两两匹配算距离等不等于$c$了，比赛的时候因为不知道哪里没用long long爆掉了贡献了一发罚时。

#include<bits/stdc++.h>
using namespace std;
struct node{
    long long x,y;
}a[2][100010];
struct resnode{
    long long x1,x2,y1,y2;
}res[100010];
inline long long read()
{
    long long x=0,f=1;char c=getchar();
    while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') x=x*10+c-48,c=getchar();
    return x*f;
}
inline long long gcd(long long a,long long b)
{
    if(!b) return a;
    return gcd(b,a%b);
}
long long r,T,cnt,ans[2];
int dx[4]={1,-1,1,-1};
int dy[4]={1,1,-1,-1};
void work(long long d,int p)
{
    for(long long s=1;s*s<=r/d;++s)
    {
        long long t=sqrt(r/d-s*s);
        if(gcd(s,t)==1&&s*s+t*t==r/d)
        {
            long long x=(s*s-t*t)/2*d;
            long long y=d*s*t;
            if(x>0&&y>0&&x*x+y*y==(r/2)*(r/2))
            {
                for(int i=0;i<4;i++)
                {
                    a[p][++ans[p]].x=x*dx[i];a[p][ans[p]].y=y*dy[i];
                    a[p][++ans[p]].x=y*dx[i];a[p][ans[p]].y=x*dy[i];
                }
            }
        }
    }
}
long long dist(node p,node q)
{
    return (p.x-q.x)*(p.x-q.x)+(p.y-q.y)*(p.y-q.y);
}
bool cmp(resnode p,resnode q)
{
    if(p.x1==q.x1&&p.y1==q.y1&&p.x2==q.x2) return p.y2<q.y2;
    if(p.x1==q.x1&&p.y1==q.y1) return p.x2<q.x2;
    if(p.x1==q.x1) return p.y1<q.y1;
    return p.x1<q.x1;
}
int main()
{
    T=read();
    while(T--)
    {
        ans[0]=0;
        ans[1]=0;
        cnt=0;
        r=read()*2;
        for(long long i=1;i*i<=r;++i)
        {
            if(r%i==0)
            {
                work(i,0);
                if(i*i!=r) work(r/i,0);
            }
        }
        a[0][++ans[0]].x=0;a[0][ans[0]].y=r/2;
        a[0][++ans[0]].x=r/2;a[0][ans[0]].y=0;
        a[0][++ans[0]].x=0;a[0][ans[0]].y=-r/2;
        a[0][++ans[0]].x=-r/2;a[0][ans[0]].y=0;
        r=read()*2;
        for(long long i=1;i*i<=r;++i)
        {
            if(r%i==0)
            {
                work(i,1);
                if(i*i!=r) work(r/i,1);
            }
        }
        a[1][++ans[1]].x=0;a[1][ans[1]].y=r/2;
        a[1][++ans[1]].x=r/2;a[1][ans[1]].y=0;
        a[1][++ans[1]].x=0;a[1][ans[1]].y=-r/2;
        a[1][++ans[1]].x=-r/2;a[1][ans[1]].y=0;
        r=read();
        for(int i=1;i<=ans[0];i++)
        {
            for(int j=1;j<=ans[1];j++)
            {
                if(dist(a[0][i],a[1][j])==r*r)
                {
                    res[++cnt].x1=a[0][i].x;res[cnt].y1=a[0][i].y;
                    res[cnt].x2=a[1][j].x;res[cnt].y2=a[1][j].y;
                }
            }
        }
        /*for(int i=0;i<=1;i++)
        {
            printf("%d\n",ans[i]);
            for(int j=1;j<=ans[i];j++)
                printf("%d,%d\n",a[i][j].x,a[i][j].y);
        }*/
        sort(res+1,res+1+cnt,cmp);
        printf("%lld\n",cnt);
        for(int i=1;i<=cnt;i++)
            printf("%lld %lld %lld %lld\n",res[i].x1,res[i].y1,res[i].x2,res[i].y2);
    }
    return 0;
}