geometry/TrianglePointCount.h

• View this file on GitHub
• Last update: 2025-11-22 00:26:56+07:00

Depends on

• geometry/Point.h

Verified with

• tests/Count_Points_in_Triangle.test.cpp

Code

#include "Point.h"

template <class P, int MAXN, int MAXM>
struct TrianglePointCount {
  // Bảng lưu trạng thái: side[i][j] = bitset các điểm B nằm bên trái vector A[i]->A[j]
  bitset<MAXM> side[MAXN][MAXN];
  const vector<P>& A;  // Tham chiếu tới mảng A để kiểm tra hướng khi truy vấn
  // Constructor: Thực hiện Precomputation O(N^2 * M)
  TrianglePointCount(const vector<P>& A, const vector<P>& B)
      : A(A) {
    int n = sz(A), m = sz(B);
    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < n; ++j) {
        if (i == j) continue;
        P vecIJ = A[j] - A[i]; // Vector A[i] -> A[j]
        for (int k = 0; k < m; ++k) {
          P vecIK = B[k] - A[i]; // Vector A[i] -> B[k]
          // Nếu B[k] nằm thực sự bên trái A[i]->A[j] (cross product > 0)
          if (vecIJ.cross(vecIK) > 0) side[i][j][k] = 1;
        }
      }
    }
  }
  // Truy vấn: Đếm số điểm B nằm trong tam giác A[a], A[b], A[c]
  // Độ phức tạp: O(M/64) ~ O(1)
  int query(int a, int b, int c) {
    // Kiểm tra hướng của tam giác
    auto area = A[a].cross(A[b], A[c]);
    if (area == 0) return 0;  // Tam giác suy biến (thẳng hàng)
    if (area > 0) {
      // Ngược chiều kim đồng hồ (CCW): A->B->C
      // Điểm trong tam giác phải nằm trái AB, trái BC, VÀ trái CA
      return (side[a][b] & side[b][c] & side[c][a]).count();
    } else {
      // Cùng chiều kim đồng hồ (CW): A->C->B là CCW
      return (side[a][c] & side[c][b] & side[b][a]).count();
    }
  }
};

#line 2 "geometry/Point.h"

template <class T>
int sgn(T x) { return (x > 0) - (x < 0); }
template <class T>
struct Point {
  typedef Point P;
  T x, y;
  explicit Point(T x = 0, T y = 0) : x(x), y(y) {}
  bool operator<(P p) const { return tie(x, y) < tie(p.x, p.y); }
  bool operator==(P p) const { return tie(x, y) == tie(p.x, p.y); }
  P operator+(P p) const { return P(x + p.x, y + p.y); }
  P operator-(P p) const { return P(x - p.x, y - p.y); }
  P operator*(T d) const { return P(x * d, y * d); }
  P operator/(T d) const { return P(x / d, y / d); }
  T dot(P p) const { return x * p.x + y * p.y; }
  T cross(P p) const { return x * p.y - y * p.x; }
  T cross(P a, P b) const { return (a - *this).cross(b - *this); }
  T dist2() const { return x * x + y * y; }
  T dist() const { return sqrt(dist2()); }
  // angle to x-axis in interval [-pi, pi]
  T angle() const { return atan2l(y, x); }
  P unit() const { return *this / dist(); }  // makes dist()=1
  P perp() const { return P(-y, x); }        // rotates +90 degrees
  P normal() const { return perp().unit(); }
  // returns point rotated 'a' radians ccw around the origin
  P rotate(ld a) const {
    return P(x * cos(a) - y * sin(a), x * sin(a) + y * cos(a));
  }
  friend ostream& operator<<(ostream& os, P p) {
    return os << "(" << p.x << "," << p.y << ")";
  }
};
#line 2 "geometry/TrianglePointCount.h"

template <class P, int MAXN, int MAXM>
struct TrianglePointCount {
  // Bảng lưu trạng thái: side[i][j] = bitset các điểm B nằm bên trái vector A[i]->A[j]
  bitset<MAXM> side[MAXN][MAXN];
  const vector<P>& A;  // Tham chiếu tới mảng A để kiểm tra hướng khi truy vấn
  // Constructor: Thực hiện Precomputation O(N^2 * M)
  TrianglePointCount(const vector<P>& A, const vector<P>& B)
      : A(A) {
    int n = sz(A), m = sz(B);
    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < n; ++j) {
        if (i == j) continue;
        P vecIJ = A[j] - A[i]; // Vector A[i] -> A[j]
        for (int k = 0; k < m; ++k) {
          P vecIK = B[k] - A[i]; // Vector A[i] -> B[k]
          // Nếu B[k] nằm thực sự bên trái A[i]->A[j] (cross product > 0)
          if (vecIJ.cross(vecIK) > 0) side[i][j][k] = 1;
        }
      }
    }
  }
  // Truy vấn: Đếm số điểm B nằm trong tam giác A[a], A[b], A[c]
  // Độ phức tạp: O(M/64) ~ O(1)
  int query(int a, int b, int c) {
    // Kiểm tra hướng của tam giác
    auto area = A[a].cross(A[b], A[c]);
    if (area == 0) return 0;  // Tam giác suy biến (thẳng hàng)
    if (area > 0) {
      // Ngược chiều kim đồng hồ (CCW): A->B->C
      // Điểm trong tam giác phải nằm trái AB, trái BC, VÀ trái CA
      return (side[a][b] & side[b][c] & side[c][a]).count();
    } else {
      // Cùng chiều kim đồng hồ (CW): A->C->B là CCW
      return (side[a][c] & side[c][b] & side[b][a]).count();
    }
  }
};

Back to top page