# 用途

x軸あるいはy軸に平行な線分（それぞれ水平線分、垂直線分）が複数与えられてその交点を求める.

# アルゴリズム

すべての点に以下の情報を加える
- 垂直線分の上下（UP, DOWN）、水平線分の左右（LEFT, RIGHT）のどれに属するか type
- 自身が作る線分の長さ length （水平線分のみで使用）
すべての点を以下に基づいてソートする
- y昇順
- yが同点の場合はtypeで次の順: DOWN -> LEFT -> UP -> RIGHT
- typeが同点の場合はx昇順
カウンタcounterを0で初期化する
先頭からすべての点について以下を実行する
- typeがUPの場合) 二分探索木BTにx座標を追加
- typeがLEFTの場合) x座標とx座標+lengthの範囲に収まるBTに収まる要素数をcounterに加える
- typeがDOWNの場合) BTからx座標を取り除く

# 計算量

点のソートに $O(N \log N)$ .

その後、setに対する挿入、削除、検索すべてが $O(\log N)$ .
それを点の数 $N$ だけ繰り返す.

$O(N \log N)$

# コード

// C++ 14
#include <iostream>
#include <string>
#include <vector>
#include <list>
#include <algorithm>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <unordered_map>
#include <math.h>

#define ll long long
#define Int int
#define loop(x, start, end) for(Int x = start; x < end; x++)
#define loopdown(x, start, end) for(int x = start; x > end; x--)
#define rep(n) for(int x = 0; x < n; x++)
#define span(a,x,y) a.begin()+x,a.begin()+y
#define span_all(a) a.begin(),a.end()
#define len(x) (x.size())
#define last(x) (*(x.end()-1))

using namespace std;

#define EPS 0.0000000001
#define fequals(a,b) (fabs((a) - (b)) < EPS)

class Vector2 {
public:
  double x, y;
  
  Vector2(double x = 0, double y = 0): x(x), y(y) {}
  
  Vector2 operator + (const Vector2 v) const { return Vector2(x + v.x, y + v.y); }
  Vector2 operator - (const Vector2 v) const { return Vector2(x - v.x, y - v.y); }
  Vector2 operator * (const double k) const { return Vector2(x * k, y * k); }
  Vector2 operator / (const double k) const { return Vector2(x / k, y / k); }
  
  double length() { return sqrt(norm()); }
  double norm() { return x * x + y * y; }
  double dot (Vector2 const v) { return x * v.x + y * v.y; }
  double cross (Vector2 const v) { return x * v.y - y * v.x; }
  
  bool parallel(Vector2 &other) {
    return fequals(0, cross(other));
  }
  
  bool orthogonal(Vector2 &other) {
    return fequals(0, dot(other));
  }
  
  bool operator < (const Vector2 &v) {
    return x != v.x ? x < v.x : y < v.y;
  }
  
  bool operator == (const Vector2 &v) {
    return fabs(x - v.x) < EPS && fabs(y - v.y) < EPS;
  }
};

ostream & operator << (ostream & out, Vector2 const & v) { 
  out<< "Vector2(" << v.x << ", " << v.y << ')';
  return out;
}

istream & operator >> (istream & in, Vector2 & v) { 
  double x, y;
  in >> x;
  in >> y;
  v.x = x;
  v.y = y;
  return in;
}

enum PointType {
  DOWN, LEFT, UP, RIGHT // The order matters.
};

class Point {
public:
  Vector2 coord;
  PointType type;
  Int length;
  
  bool operator < (const Point &p) {
    // if y is a tie, use type.
    return (p.coord.y == coord.y && type != p.type)
      ? type < p.type
      : coord < p.coord;
  }
  
  bool operator == (const Point &p) {
    return coord == p.coord && type == p.type;
  }
};

Int line_sweep() {
  set<Int> BT;
  Int count = 0;
  sort(span_all(points));
  
  for (auto p: points) {
    switch (p.type) {
      case DOWN:
        BT.insert(p.coord.x);
        break;
      case LEFT:
        count += distance(
          lower_bound(span_all(BT), p.coord.x),
          upper_bound(span_all(BT), p.coord.x + p.length)
        );
        break;
      case UP:
        BT.erase(p.coord.x);
        break;
      case RIGHT:
        // do nothing
        break;
    }
  }
  return count;
}

線分交差 | ラインスウィープアルゴリズム

# 用途

# アルゴリズム

# 計算量

# コード

コメント