hdu - 4631 - Sad Love Story

题意：一个空平面，每次增加一个点，其坐标根据上一个点算出：(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;
}