题目有n个点 初始时先加入一个点到集合中,然后依次加入点,直到集合中的点的个数为n为止,每加一个点,算一次集合内所有点之间的最短距离,输出这个距离;为了避免输出过多,把每次要输出的结果相加后,每组只输出一个值;
细节见代码注释
#include<cstdio>
#include<algorithm>
#include<set>
#define INF 1e18
using namespace std;
long long n,Ax,Ay,Az,Bx,By,Bz;
struct point
{
long long x,y;
bool operator <(const point& b)const//递增排序
{
if(x==b.x) return y<b.y;//二级排序
return x<b.x;
}
};
typedef set<point>::iterator SetIt;
inline point &xy(point &x)//迭代计算,出下一个点
{
x.x=(x.x*Ax+Ay)%Az;
x.y=(x.y*Bx+By)%Bz;
return x;
}
void deal()
{
set<point> se;
long long ans=0,milen=INF,temp,i;
point now;now.x=now.y=0;
se.insert(xy(now));//插入第一个点
for(i=2;i<=n;i++)
{
SetIt it=se.lower_bound(xy(now));//二分查找,查找当前的点需要进位子
SetIt it1=it,it2=it;
while(it1!=se.begin())//向左扩张
{
--it1;
temp=(it1->x-now.x)*(it1->x-now.x);
if(temp>=milen) break;
temp+=(it1->y-now.y)*(it1->y-now.y);
milen=min(milen,temp);
}
while(it2!=se.end())//向右扩张
{
temp=(it2->x-now.x)*(it2->x-now.x);
if(temp>=milen) break;
temp+=(it2->y-now.y)*(it2->y-now.y);
milen=min(milen,temp);
++it2;
}
se.insert(now);// 插入
ans+=milen;
}
printf("%I64d\n",ans);
}
int main()
{
freopen("in.txt","r",stdin);
int T;scanf("%d",&T );
while(T--)
{
scanf("%I64d%I64d%I64d%I64d%I64d%I64d%I64d",&n,&Ax,&Ay,&Az,&Bx,&By,&Bz);
deal();
}
return 0;
}