欢迎关注更多精彩
关注我,学习常用算法与数据结构,一题多解,降维打击。
往期相关背景
点击前往
题目大意
题目链接
https://www.luogu.com.cn/problem/P3194
有一场赛车比赛,每辆车有自己的起位置和速度,比赛时间无限长。
问题是求有哪些车辆跑在最前面。
解析
在直角坐标系中画出直线代表每辆车的状态。
每个时刻,处在最高处的车辆就是最前在的车辆。
所有时刻组成的图形是一个半平在交,所有半平面交的边就是领先过的车辆。
由于比赛时刻是从0开始,需要加一条负Y轴来限制。
注意:有些车辆的起位置和速度是相同的,需要去重。
代码
#include<stdio.h>
#include<cmath>
#include <algorithm>
#include <vector>
#include <list>
#include <cstring>
#include <set>
#include <map>
using namespace std;
const double EPS = 1e-6;
const double EPS2 = 0;
const int N = 1e6 + 10;
const int M = 1e6 + 10;
int cmp(double d) {
if (abs(d) < EPS)return 0;
if (d > 0)return 1;
return -1;
}
class Point {
public:
double x, y;
int id;
Point() {}
Point(double a, double b) :x(a), y(b) {}
Point(const Point& p) :x(p.x), y(p.y), id(p.id) {}
void in() {
scanf("%lf %lf", &x, &y);
}
void out() {
printf("%f %f\n", x, y);
}
double dis() {
return sqrt(x * x + y * y);
}
double dis2() {
return x * x + y * y;
}
Point operator -() const {
return Point(-x, -y);
}
Point operator -(const Point& p) const {
return Point(x - p.x, y - p.y);
}
Point operator +(const Point& p) const {
return Point(x + p.x, y + p.y);
}
Point operator *(double d)const {
return Point(x * d, y * d);
}
Point operator /(double d)const {
return Point(x / d, y / d);
}
void operator -=(Point& p) {
x -= p.x;
y -= p.y;
}
void operator +=(Point& p) {
x += p.x;
y += p.y;
}
void operator *=(double d) {
x *= d;
y *= d;
}
void operator /=(double d) {
this ->operator*= (1 / d);
}
bool operator<(const Point& a) const {
return x < a.x || (abs(x - a.x) < EPS && y < a.y);
}
bool operator==(const Point& a) const {
return abs(x - a.x) < EPS && abs(y - a.y) < EPS;
}
};
// 向量操作
double cross(const Point& a, const Point& b) {
return a.x * b.y - a.y * b.x;
}
double dot(const Point& a, const Point& b) {
return a.x * b.x + a.y * b.y;
}
class Line {
public:
Point front, tail;
int ind;
Line() {}
Line(const Point &a, const Point &b) :front(a), tail(b) {}
Line(const Point &a, const Point &b, int i) :front(a), tail(b), ind(i) {}
};
int cmp(const Line& a, const Line& b) {
//auto ta = atan2(a.front.y - a.tail.y, a.front.x - a.tail.x);
//auto tb = atan2(b.front.y - b.tail.y, b.front.x - b.tail.x);
// return cmp(ta - tb);
return cmp(a.front.y - a.tail.y - (b.front.y - b.tail.y));
}
// 点在直线哪一边>0 左边,<0边
int SideJudge(const Line& a, const Point& b) {
if (cross(a.front - a.tail, b - a.tail) < 0)return -1;
return 1;
}
int LineSort(const Line& a, const Line& b) {
int c = cmp(a, b);
if (c)return c < 0;
return a.tail.y > b.tail.y;
}
/*
点p 到 p+r 表示线段1
点q 到 q+s 表示线段2
线段1 上1点用 p' = p+t*r (0<=t<=1)
线段2 上1点用 q' = q+u*s (0<=u<=1)
让两式相等求交点 p+t*r = q+u*s
两边都叉乘s
(p+t*r)Xs = (q+u*s)Xs
pXs + t*rXs = qXs
t = (q-p)Xs/(rXs)
同理,
u = (p-q)Xr/(sXr) -> u = (q-p)Xr/(rXs)
以下分4种情况:
1. 共线,sXr==0 && (q-p)Xr==0, 计算 (q-p)在r上的投影在r长度上的占比t0,
计算(q+s-p)在r上的投影在r长度上的占比t1,查看[t0, t1]是否与范围[0,1]有交集。
如果t0>t1, 则比较[t1, t0]是否与范围[0,1]有交集。
t0 = (q-p)*r/(r*r)
t1 = (q+s-p)*r/(r*r) = t0 + s · r / (r · r)
2. 平行sXr==0 && (q-p)Xr!=0
3. 0<=u<=1 && 0<=t<=1 有交点
4. 其他u, t不在0到范围内,没有交点。
*/
pair<double, double> intersection(const Point& q, const Point& s, const Point& p, const Point& r) {
// 计算 (q-p)Xr
auto qpr = cross(q - p, r);
auto qps = cross(q - p, s);
auto rXs = cross(r, s);
if (cmp(rXs) == 0)return { -1, -1 }; // 平行或共线
// 求解t, u
// t = (q-p)Xs/(rXs)
auto t = qps / rXs;
// u = (q-p)Xr/(rXs)
auto u = qpr / rXs;
return { u, t };
}
Point LineCross(const Line &a, const Line &b) {
Point dira = a.front - a.tail;
Point dirb = b.front - b.tail;
auto p = intersection(a.tail, dira, b.tail, dirb);
return a.tail + dira*p.first;
}
class HalfPlane {
public:
vector<Line> lines;
void addLine(const Line &a) {
lines.push_back(a);
}
set<int> run() {
sort(lines.begin(), lines.end(), LineSort);
vector<int> q(lines.size()+10);
vector<Point> t(lines.size()+10);
int l = -1, r = 0;
q[0] = 0;
for (int i = 1; i < lines.size(); ++i) {
if (cmp(lines[i], lines[i - 1]) == 0)continue;
while (r - l > 1 && SideJudge(lines[i], t[r]) < 0)r--;
while (r - l > 1 && SideJudge(lines[i], t[l+2]) < 0)l++;
q[++r] = i;
t[r] = LineCross(lines[q[r]], lines[q[r - 1]]);
}
/*while (r - l > 1 && SideJudge(lines[q[l + 1]], t[r]) < 0)r--;
t[r+1] = LineCross(lines[q[l+1]], lines[q[r]]);
r++;*/
// 统计交点
/*
l++;
vector<Point> ans(r-l);
for (int i = 0; i < ans.size(); ++i) {
ans[i] = t[i + l + 1];
}*/
set<int> ans;
for (int i = l + 1; i <= r; ++i) ans.insert(lines[q[i]].ind);
return ans;
}
};
Point oiPs[N];
int position[N], volocity[N];
int carBelong[N];
void solve() {
int n;
scanf("%d", &n);
HalfPlane hp;
for (int i = 0; i < n; ++i) scanf("%d", position+i);
for (int i = 0; i < n; ++i) scanf("%d", volocity+i);
hp.addLine({ Point(0, -1), Point(0, 0), 0});
map<pair<int, int>, int> numInd;
for (int i = 0; i < n; ++i) {
pair<int, int> ele(position[i], volocity[i]);
if (numInd.count(ele) > 0) {
carBelong[i] = numInd[ele];
continue;
}
carBelong[i] = numInd.size() + 1;
numInd[ele] = numInd.size() + 1;
Line l(Point( 1, position[i]+volocity[i]), Point(0, position[i]), carBelong[i]);
hp.addLine(l);
}
auto seeLine = hp.run();
vector<int> ans;
for (int i = 0; i < n; ++i) {
if (seeLine.count(carBelong[i]))ans.push_back(i + 1);
}
printf("%d\n", ans.size());
for (int i = 0; i < ans.size(); ++i) {
if (i)putchar(' ');
printf("%d", ans[i]);
}
puts("");
}
int main() {
solve();
return 0;
}
/*
*/
本人码农,希望通过自己的分享,让大家更容易学懂计算机知识。创作不易,帮忙点击公众号的链接。