hdu 4631 Sad Love Story(最近点对+暴力)
这题第一次做的时候觉得太暴力了。。不敢搞。。今天看了题解,一共有三种方法可以搞:sol 1是用STL暴力的。。简单粗暴,但是考虑到自己的时间复杂度分析以及对数据的理解不够深,比赛的时候还是慎用STL的好。sol 2是用KD树,那是什么。。。无奈只能尝试一下sol 3了。。先求出所有点的最近点对,然后,对于这对点之后出现的点,累加就行了,然后不断减少上界,直到大于1为止(一个点的时候无距离)。要改一下Closest_Pair的模板,使其返回最近点对的下标然后才能减少上界。。。10+Y,给hdu的long long输入输出跪了,最后怒换cin cout。。。。#include<algorithm> #include<iostream> #include<cstring> #include<cstdlib> #include<fstream> #include<sstream> #include<bitset> #include<vector> #include<string> #include<cstdio> #include<cmath> #include<stack> #include<queue> #include<stack> #include<map> #include<set> #define FF(i, a, b) for(int i=a; i<b; i++) #define FD(i, a, b) for(int i=a; i>=b; i--) #define REP(i, n) for(int i=0; i<n; i++) #define CLR(a, b) memset(a, b, sizeof(a)) #define debug puts("**debug**") #define LL long long #define PB push_back #define eps 1e-10 using namespace std; const LL INF = 1LL << 50; const int N = 500005; template <class T> T sqr(T x) { return x*x; } struct Point { LL x, y; //id1用于求Closest_Pair id2用于求每次找到的上界 int id1, id2; Point(){} Point(LL x, LL y, int id2):x(x), y(y), id2(id2) {} }p[N], point[N], tmpt[N]; int n; bool cmpxy(const Point& a, const Point& b) { return a.x < b.x || (a.x == b.x && a.y < b.y); } bool cmpy(const Point& a, const Point& b) { return a.y < b.y; } LL dist(Point a, Point b) //dist函数注意不要sqrt { return sqr(a.x-b.x) + sqr(a.y-b.y); } struct Ans { int l, r; Ans(){} Ans(int l, int r):l(l), r(r){} }; Ans Closest_Pair(int left, int right) { LL d = INF; //由于函数返回结构体 原先模板中只有1或2个点时统一考虑 if(left==right || left+1 == right) return Ans(left, right); int mid = (left+right)>>1; Ans ans1 = Closest_Pair(left, mid); double d1; if(ans1.l == ans1.r) d1 = INF; //只有一个点的时候无距离 else d1 = dist(point[ans1.l],point[ans1.r]); Ans ans2 = Closest_Pair(mid+1, right); double d2; if(ans2.l == ans2.r) d2 = INF; else d2 = dist(point[ans2.l], point[ans2.r]); Ans ret; if(d1 < d2) ret = ans1; else ret = ans2; d = min(d1,d2); int i,j,k=0; for(i = left; i <= right; i++) { if(sqr(point[mid].x-point[i].x) <= d) //这里也需要改 因为dist函数变了 tmpt[k++] = point[i]; } sort(tmpt,tmpt+k,cmpy); for(i = 0; i < k; i++) { for(j = i+1; j < k && sqr(tmpt[i].y-tmpt[j].y) < d; j++)//这里也是。。。 { double d3 = dist(tmpt[i],tmpt[j]); if(d > d3) { d = d3; ret = Ans(tmpt[i].id1, tmpt[j].id1); } } } return ret; } void read() { LL a1, b1, c1, a2, b2, c2; cin>>a1>>b1>>c1>>a2>>b2>>c2; p[0] = Point(0, 0, 0); FF(i, 1, n+1) p[i] = Point((p[i-1].x*a1+b1)%c1, (p[i-1].y*a2+b2)%c2, i); } int main() { int T; scanf("%d", &T); while(T--) { scanf("%d", &n); read(); int end = n; LL ans = 0; while(end > 1) { REP(i, end) point[i] = p[i+1]; sort(point, point+end, cmpxy); REP(i, end) point[i].id1 = i; Ans t = Closest_Pair(0, end-1); int m = max(point[t.l].id2, point[t.r].id2); ans += (end-m+1)*dist(point[t.l], point[t.r]); end = m - 1; } cout<<ans<<endl; } return 0; }
补充:软件开发 , C++ ,