math/rational.hpp

• View this file on GitHub
• Last update: 2025-04-05 10:48:22+09:00
• Include: #include "math/rational.hpp"

Depends on

• type_traits/io.hpp

Required by

• geometry/cross_point_line_line.hpp

Code

#ifndef KK2_MATH_RATIONAL_HPP
#define KK2_MATH_RATIONAL_HPP 1

#include <numeric>

#include "../type_traits/io.hpp"

namespace kk2 {

namespace rational {

template <typename T, typename Compare> struct RationalBase {
    T a, b;

    // a / b

    RationalBase() : a(0), b(1) {}

    RationalBase(T a_) : a(a_), b(1) {}

    RationalBase(T a_, T b_) : a(a_), b(b_) {
        assert(b);
        if (b < 0) a = -a, b = -b;
        T g = std::gcd(a, b);
        a /= g, b /= g;
    }

    static RationalBase raw(T a_, T b_) {
        T res;
        res.a = a_, res.b = b_;
        return res;
    }

    operator bool() const { return a; }

    double to_double() const { return (double)a / b; }

    long double to_ldouble() const { return (long double)a / b; }

    friend RationalBase operator+(const RationalBase &lhs, const RationalBase &rhs) {
        if (lhs.b == rhs.b) return RationalBase(lhs.a + rhs.a, lhs.b);
        return RationalBase(lhs.a * rhs.b + rhs.a * lhs.b, lhs.b * rhs.b);
    }

    friend RationalBase operator-(const RationalBase &lhs, const RationalBase &rhs) {
        if (lhs.b == rhs.b) return RationalBase(lhs.a - rhs.a, lhs.b);
        return RationalBase(lhs.a * rhs.b - rhs.a * lhs.b, lhs.b * rhs.b);
    }

    friend RationalBase operator*(const RationalBase &lhs, const RationalBase &rhs) {
        return RationalBase(lhs.a * rhs.a, lhs.b * rhs.b);
    }

    friend RationalBase operator/(const RationalBase &lhs, const RationalBase &rhs) {
        return RationalBase(lhs.a * rhs.b, lhs.b * rhs.a);
    }

    RationalBase &operator+=(const RationalBase &rhs) { return (*this) = (*this) + rhs; }

    RationalBase &operator-=(const RationalBase &rhs) { return (*this) = (*this) - rhs; }

    RationalBase &operator*=(const RationalBase &rhs) { return (*this) = (*this) * rhs; }

    RationalBase &operator/=(const RationalBase &rhs) { return (*this) = (*this) / rhs; }

    RationalBase operator-() const { return raw(-a, b); }

    RationalBase operator+() const { return *this; }

    RationalBase inv() const { return a < 0 ? raw(-b, -a) : raw(b, a); }

    RationalBase abs() const { return a < 0 ? raw(-a, b) : *this; }

    RationalBase pow(long long k) const {
        RationalBase res(1), base = *this;
        while (k) {
            if (k & 1) res *= base;
            base *= base;
            k >>= 1;
        }
        return res;
    }

    friend bool operator==(const RationalBase &lhs, const RationalBase &rhs) {
        return lhs.a == rhs.a and lhs.b == rhs.b;
    }

    friend bool operator!=(const RationalBase &lhs, const RationalBase &rhs) {
        return !(lhs == rhs);
    }

    friend bool operator<(const RationalBase &lhs, const RationalBase &rhs) {
        return Compare{lhs.a} * rhs.b < Compare{rhs.a} * lhs.b;
    }

    friend bool operator>(const RationalBase &lhs, const RationalBase &rhs) {
        return Compare{lhs.a} * rhs.b > Compare{rhs.a} * lhs.b;
    }

    friend bool operator<=(const RationalBase &lhs, const RationalBase &rhs) {
        return lhs == rhs or lhs < rhs;
    }

    friend bool operator>=(const RationalBase &lhs, const RationalBase &rhs) {
        return lhs == rhs or lhs > rhs;
    }

    template <class OStream, is_ostream_t<OStream> * = nullptr>
    friend OStream &operator<<(OStream &os, const RationalBase &rhs) {
        return os << rhs.a << " / " << rhs.b;
    }
};

} // namespace rational

template <typename T> using Rational = rational::RationalBase<T, T>;
using RationalInt = rational::RationalBase<int, long long>;
using RationalI64 = rational::RationalBase<long long, __int128>;

} // namespace kk2

#endif // KK2_MATH_RATIONAL_HPP

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 401, in update
    self.update(self._resolve(pathlib.Path(included), included_from=path))
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 312, in update
    raise BundleErrorAt(path, i + 1, "#pragma once found in a non-first line")
onlinejudge_verify.languages.cplusplus_bundle.BundleErrorAt: type_traits/io.hpp: line 4: #pragma once found in a non-first line

Back to top page