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;

}