题意:一个空平面,每次增加一个点,其坐标根据上一个点算出:(x[i-1] * Ax + Bx ) mod Cx,(y[i-1] * Ay + By ) mod Cy求出现有点集中的最近点对的距离的平方,共增加n个点,求每次求得的平方的和( n <= 5*100000, T(T <= 10)组测试数据)。
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4631
——>>时间给得挺长的,20s,不过暴力求还是会TLE……这里进行贪心剪枝。
在已有点的集合中点按x坐标排序,每增一个点,就从这个点开始往右扫,扫到单单横坐标的差的平方 >= Min就跳出,因为加上纵坐标的差的平方肯定 >= Min,且越往右越大;同理,从这个点的左边第一个点开始往左扫,扫完后统计结果。
#include <cstdio>
#include <set>using namespace std;const int maxn = 500000 + 10;
const long long INF = (1LL<<60);
int n, x[maxn], y[maxn];struct Point{int x;int y;bool operator < (const Point& e) const{return x < e.x;}
};void read(int *a){int A, B, C, i;a[0] = 0;scanf("%d%d%d", &A, &B, &C);for(i = 1; i <= n; i++){a[i] = ((long long)a[i-1] * A + B) % C;}
}long long solve(){int i;long long Min = INF, ret = 0;multiset<Point> se;se.clear();Point v;v.x = x[1];v.y = y[1];se.insert(v);for(i = 2; i <= n; i++){v.x = x[i];v.y = y[i];multiset<Point>::iterator p = se.lower_bound(v), iter;for(iter = p; iter != se.end(); iter++){ //从p开始,一直到se.end()前一个位置long long dx = v.x - iter->x;dx *= dx;if(dx >= Min) break; //剪枝long long dy = v.y - iter->y;dy *= dy;Min = min(Min, dx + dy);}for(iter = p; iter != se.begin();){ //从p的上一个位置开始,一直算完se.begin()iter--;long long dx = v.x - iter->x;dx *= dx;if(dx >= Min) break; //剪枝long long dy = v.y - iter->y;dy *= dy;Min = min(Min, dx + dy);}ret += Min;se.insert(v);}return ret;
}int main()
{int T;scanf("%d", &T);while(T--){scanf("%d", &n);read(x);read(y);printf("%I64d\n", solve());}return 0;
}