题意:一个空平面,每次增加一个点,其坐标根据上一个点算出:(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 ", solve()); } return 0; }