Skip to content

minkowski_sum.hpp

SECTIONGeometry INCLUDEnoya/minkowski_sum.hpp

Minkowski sum of two point sets as a strict counter-clockwise convex polygon in O((n+m) log(n+m)), including degenerate point and segment cases.

求两个点集凸包的 Minkowski 和,并正确处理点、线段等退化输入。

Implementation

View on GitHub

#ifndef NOYA_MINKOWSKI_SUM_HPP
#define NOYA_MINKOWSKI_SUM_HPP 1

/// @complexity Time: O(n log n + m log m) including hull normalization.
/// Space: O(n + m).

#include "noya/geometry_base.hpp"

#include <algorithm>
#include <vector>

namespace noya {

/// @brief Minkowski sum of two point sets as a strict counter-clockwise convex
/// polygon in O((n+m) log(n+m)), including degenerate point and segment cases.
template <class T>
std::vector<point<T>> minkowski_sum(std::vector<point<T>> first,
                                    std::vector<point<T>> second) {
  first = convex_hull(std::move(first));
  second = convex_hull(std::move(second));
  if (first.empty() || second.empty()) {
    return {};
  }
  if (first.size() == 1) {
    for (point<T> &value : second) {
      value += first[0];
    }
    return second;
  }
  if (second.size() == 1) {
    for (point<T> &value : first) {
      value += second[0];
    }
    return first;
  }

  auto rotate_lowest = [](std::vector<point<T>> &polygon) {
    auto iterator =
        std::min_element(polygon.begin(), polygon.end(),
                         [](const point<T> &a, const point<T> &b) {
                           return a.y < b.y || (a.y == b.y && a.x < b.x);
                         });
    std::rotate(polygon.begin(), iterator, polygon.end());
  };
  rotate_lowest(first);
  rotate_lowest(second);

  std::vector<point<T>> first_edges(first.size());
  std::vector<point<T>> second_edges(second.size());
  for (int index = 0; index < int(first.size()); index++) {
    first_edges[index] = first[(index + 1) % first.size()] - first[index];
  }
  for (int index = 0; index < int(second.size()); index++) {
    second_edges[index] = second[(index + 1) % second.size()] - second[index];
  }

  point<T> current = first[0] + second[0];
  std::vector<point<T>> result{current};
  int first_index = 0;
  int second_index = 0;
  while (first_index < int(first_edges.size()) ||
         second_index < int(second_edges.size())) {
    if (second_index == int(second_edges.size())) {
      current += first_edges[first_index++];
    } else if (first_index == int(first_edges.size())) {
      current += second_edges[second_index++];
    } else {
      T turn = cross(first_edges[first_index], second_edges[second_index]);
      if (turn >= T{}) {
        current += first_edges[first_index++];
      }
      if (turn <= T{}) {
        current += second_edges[second_index++];
      }
    }
    result.push_back(current);
  }
  if (result.size() > 1 && result.front() == result.back()) {
    result.pop_back();
  }
  return convex_hull(std::move(result));
}

} // namespace noya

#endif // NOYA_MINKOWSKI_SUM_HPP
#include <algorithm>
#include <cmath>
#include <optional>
#include <vector>

/// @complexity Time: O(n log n + m log m) including hull normalization.
/// Space: O(n + m).

/// @complexity Time: O(1) per primitive; O(n log n) for convex hull.
/// Space: O(1) per primitive and O(n) for hull construction.

namespace noya {

/// @brief Two-dimensional point with vector arithmetic and lexicographic order.
template <class T> struct point {
  T x{};
  T y{};

  point() = default;
  point(T x_, T y_) : x(x_), y(y_) {}

  point &operator+=(const point &other) {
    x += other.x;
    y += other.y;
    return *this;
  }
  point &operator-=(const point &other) {
    x -= other.x;
    y -= other.y;
    return *this;
  }
  point &operator*=(const T &scale) {
    x *= scale;
    y *= scale;
    return *this;
  }
  point &operator/=(const T &scale) {
    x /= scale;
    y /= scale;
    return *this;
  }

  friend point operator+(point left, const point &right) {
    return left += right;
  }
  friend point operator-(point left, const point &right) {
    return left -= right;
  }
  friend point operator*(point value, const T &scale) { return value *= scale; }
  friend point operator*(const T &scale, point value) { return value *= scale; }
  friend point operator/(point value, const T &scale) { return value /= scale; }
  friend bool operator==(const point &, const point &) = default;
  friend bool operator<(const point &left, const point &right) {
    return left.x < right.x || (left.x == right.x && left.y < right.y);
  }
};

/// @brief Circle represented by a center and a nonnegative radius.
template <class Real> struct circle {
  point<Real> center;
  Real radius{};
};

/// @brief Return the dot product of two vectors.
template <class T> T dot(const point<T> &a, const point<T> &b) {
  return a.x * b.x + a.y * b.y;
}

/// @brief Return the signed cross product of two vectors.
template <class T> T cross(const point<T> &a, const point<T> &b) {
  return a.x * b.y - a.y * b.x;
}

/// @brief Return cross(a - origin, b - origin).
template <class T>
T cross(const point<T> &origin, const point<T> &a, const point<T> &b) {
  return cross(a - origin, b - origin);
}

/// @brief Return the squared Euclidean norm.
template <class T> T norm2(const point<T> &value) { return dot(value, value); }

/// @brief Compare a value with zero using an optional absolute tolerance.
template <class T> int sign(const T &value, const T &epsilon = T{}) {
  return (value > epsilon) - (value < -epsilon);
}

/// @brief Return -1, 0, or 1 for a clockwise, collinear, or counter-clockwise
/// turn.
template <class T>
int orientation(const point<T> &a, const point<T> &b, const point<T> &c,
                const T &epsilon = T{}) {
  return sign(cross(a, b, c), epsilon);
}

/// @brief Test whether p lies on the closed segment [a, b].
template <class T>
bool on_segment(const point<T> &p, const point<T> &a, const point<T> &b,
                const T &epsilon = T{}) {
  if (orientation(a, b, p, epsilon) != 0) {
    return false;
  }
  return std::min(a.x, b.x) - epsilon <= p.x &&
         p.x <= std::max(a.x, b.x) + epsilon &&
         std::min(a.y, b.y) - epsilon <= p.y &&
         p.y <= std::max(a.y, b.y) + epsilon;
}

/// @brief Test whether the closed segments [a, b] and [c, d] intersect.
template <class T>
bool segments_intersect(const point<T> &a, const point<T> &b, const point<T> &c,
                        const point<T> &d, const T &epsilon = T{}) {
  int ab_c = orientation(a, b, c, epsilon);
  int ab_d = orientation(a, b, d, epsilon);
  int cd_a = orientation(c, d, a, epsilon);
  int cd_b = orientation(c, d, b, epsilon);
  if (ab_c == 0 && on_segment(c, a, b, epsilon)) {
    return true;
  }
  if (ab_d == 0 && on_segment(d, a, b, epsilon)) {
    return true;
  }
  if (cd_a == 0 && on_segment(a, c, d, epsilon)) {
    return true;
  }
  if (cd_b == 0 && on_segment(b, c, d, epsilon)) {
    return true;
  }
  return ab_c * ab_d < 0 && cd_a * cd_b < 0;
}

/// @brief Intersect the infinite lines through (a, b) and (c, d), returning
/// nullopt when they are parallel or coincident.
template <class T>
std::optional<point<long double>>
line_intersection(const point<T> &a, const point<T> &b, const point<T> &c,
                  const point<T> &d, long double epsilon = 0) {
  point<long double> first{static_cast<long double>(a.x),
                           static_cast<long double>(a.y)};
  point<long double> second{static_cast<long double>(b.x),
                            static_cast<long double>(b.y)};
  point<long double> third{static_cast<long double>(c.x),
                           static_cast<long double>(c.y)};
  point<long double> fourth{static_cast<long double>(d.x),
                            static_cast<long double>(d.y)};
  point<long double> direction_a = second - first;
  point<long double> direction_b = fourth - third;
  long double denominator = cross(direction_a, direction_b);
  if (std::abs(denominator) <= epsilon) {
    return std::nullopt;
  }
  long double ratio = cross(third - first, direction_b) / denominator;
  return first + direction_a * ratio;
}

/// @brief Return the convex hull in counter-clockwise order without repetition.
template <class T>
std::vector<point<T>> convex_hull(std::vector<point<T>> points,
                                  bool keep_collinear = false) {
  std::sort(points.begin(), points.end());
  points.erase(std::unique(points.begin(), points.end()), points.end());
  if (points.size() <= 1) {
    return points;
  }
  bool all_collinear = true;
  for (int i = 2; i < int(points.size()); i++) {
    all_collinear &= orientation(points[0], points[1], points[i]) == 0;
  }
  if (keep_collinear && all_collinear) {
    return points;
  }
  std::vector<point<T>> lower, upper;
  for (const point<T> &p : points) {
    while (lower.size() >= 2) {
      int turn = orientation(lower[lower.size() - 2], lower.back(), p);
      if (turn > 0 || (keep_collinear && turn == 0)) {
        break;
      }
      lower.pop_back();
    }
    lower.push_back(p);
  }
  for (auto it = points.rbegin(); it != points.rend(); ++it) {
    while (upper.size() >= 2) {
      int turn = orientation(upper[upper.size() - 2], upper.back(), *it);
      if (turn > 0 || (keep_collinear && turn == 0)) {
        break;
      }
      upper.pop_back();
    }
    upper.push_back(*it);
  }
  lower.pop_back();
  upper.pop_back();
  lower.insert(lower.end(), upper.begin(), upper.end());
  return lower;
}

/// @brief Return twice the signed area of a polygon.
template <class T> T polygon_area2(const std::vector<point<T>> &polygon) {
  T result{};
  for (int i = 0; i < int(polygon.size()); i++) {
    result += cross(polygon[i], polygon[(i + 1) % polygon.size()]);
  }
  return result;
}

/// @brief Classify a point relative to a polygon: -1 outside, 0 boundary, 1
/// inside.
template <class T>
int point_in_polygon(const point<T> &p, const std::vector<point<T>> &polygon) {
  bool inside = false;
  for (int i = 0; i < int(polygon.size()); i++) {
    point<T> a = polygon[i];
    point<T> b = polygon[(i + 1) % polygon.size()];
    if (on_segment(p, a, b)) {
      return 0;
    }
    if (a.y <= p.y && p.y < b.y && orientation(a, b, p) > 0) {
      inside = !inside;
    }
    if (b.y <= p.y && p.y < a.y && orientation(a, b, p) < 0) {
      inside = !inside;
    }
  }
  return inside ? 1 : -1;
}

/// @brief Return the Euclidean distance between two points.
template <class T> long double distance(const point<T> &a, const point<T> &b) {
  return std::hypot(
      static_cast<long double>(a.x) - static_cast<long double>(b.x),
      static_cast<long double>(a.y) - static_cast<long double>(b.y));
}

} // namespace noya

namespace noya {

/// @brief Minkowski sum of two point sets as a strict counter-clockwise convex
/// polygon in O((n+m) log(n+m)), including degenerate point and segment cases.
template <class T>
std::vector<point<T>> minkowski_sum(std::vector<point<T>> first,
                                    std::vector<point<T>> second) {
  first = convex_hull(std::move(first));
  second = convex_hull(std::move(second));
  if (first.empty() || second.empty()) {
    return {};
  }
  if (first.size() == 1) {
    for (point<T> &value : second) {
      value += first[0];
    }
    return second;
  }
  if (second.size() == 1) {
    for (point<T> &value : first) {
      value += second[0];
    }
    return first;
  }

  auto rotate_lowest = [](std::vector<point<T>> &polygon) {
    auto iterator =
        std::min_element(polygon.begin(), polygon.end(),
                         [](const point<T> &a, const point<T> &b) {
                           return a.y < b.y || (a.y == b.y && a.x < b.x);
                         });
    std::rotate(polygon.begin(), iterator, polygon.end());
  };
  rotate_lowest(first);
  rotate_lowest(second);

  std::vector<point<T>> first_edges(first.size());
  std::vector<point<T>> second_edges(second.size());
  for (int index = 0; index < int(first.size()); index++) {
    first_edges[index] = first[(index + 1) % first.size()] - first[index];
  }
  for (int index = 0; index < int(second.size()); index++) {
    second_edges[index] = second[(index + 1) % second.size()] - second[index];
  }

  point<T> current = first[0] + second[0];
  std::vector<point<T>> result{current};
  int first_index = 0;
  int second_index = 0;
  while (first_index < int(first_edges.size()) ||
         second_index < int(second_edges.size())) {
    if (second_index == int(second_edges.size())) {
      current += first_edges[first_index++];
    } else if (first_index == int(first_edges.size())) {
      current += second_edges[second_index++];
    } else {
      T turn = cross(first_edges[first_index], second_edges[second_index]);
      if (turn >= T{}) {
        current += first_edges[first_index++];
      }
      if (turn <= T{}) {
        current += second_edges[second_index++];
      }
    }
    result.push_back(current);
  }
  if (result.size() > 1 && result.front() == result.back()) {
    result.pop_back();
  }
  return convex_hull(std::move(result));
}

} // namespace noya