条件を満たす整数の組の計上
(Summation/Counting.hpp)

• View this file on GitHub
• Last update: 2026-03-12 00:53:37+09:00
• Include: #include "Summation/Counting.hpp"

Outline

条件を満たす整数の組 $(x, y)$ の数を求める.

Theory

count_between_lines (1 つ)

以下を満たす整数の組 $(x, y)$ の数を求める.

• $ax+b \leq y \leq cx+d$.
• $l \leq x \leq r$.

$l \leq x \leq r$ である $x$ を固定する. このとき, 条件を満たす $y$ の数は

\[\begin{cases} (cx+d)-(ax+b)+1 & (ax+b \leq cx+d) \\ 0 & (ax+b \gt cx+d) \end{cases} = \operatorname{relu}((c-a)x+(d-b+1))\]

となる. 従って, $x$ を走らせることにより, 求めるべき個数は

\[\sum_{x=l}^r \operatorname{relu}((c-a)x+(d-b+1))\]

である. よって, relu_linear を利用して求められる.

count_between_lines (2 つ)

以下を満たす整数の組 $(x, y)$ の数を求める.

• $ax+b \leq y \leq cx+d$.
• $ex+f \leq y \leq gx+h$.
• $l \leq x \leq r$.

各 $x$ に対して, 条件を満たす $y$ の個数は

\[\begin{align*} \max(0, \min(cx + d, gx + h) - \max(ax + b, ex + f) + 1) &= \max(0, \min( (c - a)x + (d - b + 1), (c - e)x + (d - f + 1), (g - a)x + (h - b + 1), (g - e)x + (h - f + 1))) \end{align*}\]

と分割できる.

よって, これら 4 つの直線の最小値 (ただし負なら0) の和を求める問題に帰着される.

従って, それぞれの直線について最小値を担う区間を計算し, その区間内における $\operatorname{relu}$ に関する総和を求めれば良い.

count_sum_eq, count_sum_range

以下を満たす整数の組 $(x, y)$ の数を求める.

• $p \leq x \leq q$.
• $r \leq y \leq s$.
• $a \leq x + y \leq b$.

この条件式を変形すると,

• $-x+a \leq y \leq -x+b$.
• $r \leq y \leq s$.
• $p \leq x \leq q$.

と変形できる. よって, count_between_lines に帰着できる.

また, 以下を満たす整数の組 $(x, y)$ の数は上での問題における $a = b = k$ とした特別バージョンである.

• $p \leq x \leq q$.
• $r \leq y \leq s$.
• $x + y = k$.

count_diff_eq, count_diff_range

以下を満たす整数の組 $(x, y)$ の数を求める.

• $p \leq x \leq q$.
• $r \leq y \leq s$.
• $a \leq x - y \leq b$.

$y = -z$ と変数変換すると,

• $p \leq x \leq q$.
• $-s \leq z \leq -r$.
• $a \leq x + z \leq b$.

となり, count_sum_range に帰着できる.

Contents

count_between_lines

template<typename T>
T count_between_lines(T a, T b, T c, T d, T l, T r)

• 以下を満たす整数の組 $(x, y)$ の数を求める.
  • $ax+b \leq cx+d$.
  • $l \leq x \leq r$.

template<typename T>
T count_between_lines(T a, T b, T c, T d, T e, T f, T g, T h, T l, T r)

• 以下を満たす整数の組 $(x, y)$ の数を求める.
  • $ax+b \leq cx+d$.
  • $ex+f \leq gx+h$.
  • $l \leq x \leq r$.

count_sum_eq

template<typename T>
T count_sum_eq(T p, T q, T r, T s, T k)

• 以下を満たす整数の組 $(x, y)$ の数を求める.
  • $p \leq x \leq q$.
  • $r \leq y \leq s$.
  • $x + y = k$.

count_sum_range

template<typename T>
T count_sum_eq(T p, T q, T r, T s, T a, T b)

• 以下を満たす整数の組 $(x, y)$ の数を求める.
  • $p \leq x \leq q$.
  • $r \leq y \leq s$.
  • $a \leq x + y \leq b$.

count_diff_eq

template<typename T>
T count_diff_eq(T p, T q, T r, T s, T k)

• 以下を満たす整数の組 $(x, y)$ の数を求める.
  • $p \leq x \leq q$.
  • $r \leq y \leq s$.
  • $x - y = k$.

count_diff_range

template<typename T>
T count_diff_eq(T p, T q, T r, T s, T a, T b)

• 以下を満たす整数の組 $(x, y)$ の数を求める.
  • $p \leq x \leq q$.
  • $r \leq y \leq s$.
  • $a \leq x - y \leq b$.

Depends on

• 特殊な数列の総和 (Summation/Summation.hpp)
• template/bitop.hpp
• template/exception.hpp
• template/inout.hpp
• template/macro.hpp
• template/math.hpp
• template/template.hpp
• template/utility.hpp

Code

#pragma once

#include "Summation.hpp"
#include <vector>
#include <algorithm>

namespace counting {
    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: l <= x <= r, ax + b <= y <= cx + d
     * 
     * @tparam T 整数型
     * @param a 不等式 ax + b <= y の係数 a
     * @param b 不等式 ax + b <= y の定数 b
     * @param c 不等式 y <= cx + d の係数 c
     * @param d 不等式 y <= cx + d の定数 d
     * @param l x の下限
     * @param r x の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_between_lines(T a, T b, T c, T d, T l, T r) {
        return summation::relu_linear(c - a, d - b + 1, l, r);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: l <= x <= r, ax + b <= y <= cx + d, ex + f <= y <= gx + h
     * 
     * @tparam T 整数型
     * @param a 不等式 ax + b <= y の係数 a
     * @param b 不等式 ax + b <= y の定数 b
     * @param c 不等式 y <= cx + d の係数 c
     * @param d 不等式 y <= cx + d の定数 d
     * @param e 不等式 ex + f <= y の係数 e
     * @param f 不等式 ex + f <= y の定数 f
     * @param g 不等式 y <= gx + h の係数 g
     * @param h 不等式 y <= gx + h の定数 h
     * @param l x の下限
     * @param r x の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_between_lines(T a, T b, T c, T d, T e, T f, T g, T h, T l, T r) {
        // 各 x に対して y の個数は
        // max(0, min(cx + d, gx + h) - max(ax + b, ex + f) + 1)
        // = max(0, min(
        //    (c - a)x + (d - b + 1),
        //    (c - e)x + (d - f + 1),
        //    (g - a)x + (h - b + 1),
        //    (g - e)x + (h - f + 1)
        // ))
        // となる. 4つの直線の最小値(ただし負なら0)の和を求める問題に帰着される.

        struct Line { T k, m; };
        std::vector<Line> lines = {
            {c - a, d - b + 1},
            {c - e, d - f + 1},
            {g - a, h - b + 1},
            {g - e, h - f + 1}
        };

        T L = l, R = r;

        // 値が負になる範囲を除外する (L_i(x) >= 0 を満たす範囲に限定)
        for (const auto& line : lines) {
            if (line.k == 0) {
                if (line.m < 0) return 0;
            } else if (line.k > 0) {
                L = max<T>(L, div_ceil(-line.m, line.k));
            } else {
                R = min<T>(R, div_floor(-line.m, line.k));
            }
        }

        if (L > R) return 0;

        vector<T> points;
        points.push_back(L - 1);
        points.push_back(R);

        // 直線の交点を求めて区間を分割する
        for (size_t i = 0; i < lines.size(); ++i) {
            for (size_t j = i + 1; j < lines.size(); ++j) {
                if (lines[i].k == lines[j].k) continue;

                T num = lines[j].m - lines[i].m;
                T den = lines[i].k - lines[j].k;
                T x = div_floor(num, den);
                if (x >= L - 1 && x < R) points.push_back(x);
            }
        }

        sort(points.begin(), points.end());
        points.erase(unique(points.begin(), points.end()), points.end());

        T total = 0;
        for (size_t i = 0; i < points.size() - 1; ++i) {
            T seg_l = points[i] + 1;
            T seg_r = points[i+1];
            
            // 区間内では最小値を与える直線は変わらないので, 先頭の点で判定する
            T min_val = lines[0].k * seg_l + lines[0].m;
            size_t min_idx = 0;
            for (size_t j = 1; j < lines.size(); ++j) {
                T val = lines[j].k * seg_l + lines[j].m;
                if (val < min_val) {
                    min_val = val;
                    min_idx = j;
                }
            }
            
            total += summation::linear(lines[min_idx].k, lines[min_idx].m, seg_l, seg_r);
        }

        return total;
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, x + y = k
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param k 和 k
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_sum_eq(T p, T q, T r, T s, T k) {
        return count_sum_range(p, q, r, s, k, k);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, a <= x + y <= b
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param a 和 x + y の下限
     * @param b 和 x + y の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_sum_range(T p, T q, T r, T s, T a, T b) {
        return count_between_lines<T>(0, r, 0, s, -1, a, -1, b, p, q);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, x - y = k
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param k 差 k
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_diff_eq(T p, T q, T r, T s, T k) {
        return count_sum_eq(p, q, -s, -r, k);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, a <= x - y <= b
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param a 差 x - y の下限
     * @param b 差 x - y の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_diff_range(T p, T q, T r, T s, T a, T b) {
        return count_sum_range(p, q, -s, -r, a, b);
    }
}

#line 2 "Summation/Counting.hpp"

#line 2 "Summation/Summation.hpp"

#line 2 "template/template.hpp"

using namespace std;

// intrinstic
#include <immintrin.h>

#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <cctype>
#include <cfenv>
#include <cfloat>
#include <chrono>
#include <cinttypes>
#include <climits>
#include <cmath>
#include <complex>
#include <concepts>
#include <cstdarg>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <fstream>
#include <functional>
#include <initializer_list>
#include <iomanip>
#include <ios>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <optional>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <streambuf>
#include <string>
#include <tuple>
#include <type_traits>
#include <typeinfo>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>

// utility
#line 2 "template/utility.hpp"

using ll = long long;

// a ← max(a, b) を実行する. a が更新されたら, 返り値が true.
template<typename T, typename U>
inline bool chmax(T &a, const U b){
    return (a < b ? a = b, 1: 0);
}

// a ← min(a, b) を実行する. a が更新されたら, 返り値が true.
template<typename T, typename U>
inline bool chmin(T &a, const U b){
    return (a > b ? a = b, 1: 0);
}

// a の最大値を取得する.
template<typename T>
inline T max(const vector<T> &a){
    if (a.empty()) throw invalid_argument("vector is empty.");

    return *max_element(a.begin(), a.end());
}

// vector<T> a の最小値を取得する.
template<typename T>
inline T min(const vector<T> &a){
    if (a.empty()) throw invalid_argument("vector is empty.");

    return *min_element(a.begin(), a.end());
}

// vector<T> a の最大値のインデックスを取得する.
template<typename T>
inline size_t argmax(const vector<T> &a){
    if (a.empty()) throw std::invalid_argument("vector is empty.");

    return distance(a.begin(), max_element(a.begin(), a.end()));
}

// vector<T> a の最小値のインデックスを取得する.
template<typename T>
inline size_t argmin(const vector<T> &a){
    if (a.empty()) throw invalid_argument("vector is empty.");

    return distance(a.begin(), min_element(a.begin(), a.end()));
}
#line 61 "template/template.hpp"

// math
#line 2 "template/math.hpp"

// 演算子
template<typename T>
T add(const T &x, const T &y) { return x + y; }

template<typename T>
T sub(const T &x, const T &y) { return x - y; }

template<typename T>
T mul(const T &x, const T &y) { return x * y; }

template<typename T>
T neg(const T &x) { return -x; }

template<integral T>
T bitwise_and(const T &x, const T &y) { return x & y; }

template<integral T>
T bitwise_or(const T &x, const T &y) { return x | y; }

template<integral T>
T bitwise_xor(const T &x, const T &y) { return x ^ y; }

// 除算に関する関数

// floor(x / y) を求める.
template<integral T, integral U>
auto div_floor(T x, U y){
    return x / y - ((x % y != 0) && ((x < 0) != (y < 0)));
}

// ceil(x / y) を求める.
template<integral T, integral U>
auto div_ceil(T x, U y){
    return x / y + ((x % y != 0) && ((x < 0) == (y < 0)));
}

// x を y で割った余りを求める.
template<integral T, integral U>
auto safe_mod(T x, U y){
    auto q = div_floor(x, y);
    return x - q * y ;
}

// x を y で割った商と余りを求める.
template<integral T, integral U>
auto divmod(T x, U y){
    auto q = div_floor(x, y);
    return make_pair(q, x - q * y);
}

// 四捨五入を求める.
template<integral T, integral U>
auto round(T x, U y){
    auto [q, r] = divmod(x, y);
    if (y < 0) return (r <= div_floor(y, 2)) ? q + 1 : q;
    return (r >= div_ceil(y, 2)) ? q + 1 : q;
}

// 奇数かどうか判定する.
template<integral T>
bool is_odd(const T &x) { return x % 2 != 0; }

// 偶数かどうか判定する.
template<integral T>
bool is_even(const T &x) { return x % 2 == 0; }

// m の倍数かどうか判定する.
template<integral T, integral U>
bool is_multiple(const T &x, const U &m) { return x % m == 0; }

// 正かどうか判定する.
template<typename T>
bool is_positive(const T &x) { return x > 0; }

// 負かどうか判定する.
template<typename T>
bool is_negative(const T &x) { return x < 0; }

// ゼロかどうか判定する.
template<typename T>
bool is_zero(const T &x) { return x == 0; }

// 非負かどうか判定する.
template<typename T>
bool is_non_negative(const T &x) { return x >= 0; }

// 非正かどうか判定する.
template<typename T>
bool is_non_positive(const T &x) { return x <= 0; }

// 指数に関する関数

// x の y 乗を求める.
ll intpow(ll x, ll y){
    ll a = 1;
    while (y){
        if (y & 1) { a *= x; }
        x *= x;
        y >>= 1;
    }
    return a;
}

// x の y 乗を z で割った余りを求める.
template<typename T, integral U>
T modpow(T x, U y, T z) {
    T a = 1;
    while (y) {
        if (y & 1) { (a *= x) %= z; }

        (x *= x) %= z;
        y >>= 1;
    }

    return a;
}

template<typename T>
T sum(const vector<T> &X) {
    T y = T(0);
    for (auto &&x: X) { y += x; }
    return y;
}

template<typename T>
T gcd(const T x, const T y) {
    return y == 0 ? x : gcd(y, x % y);
}

// a x + b y = gcd(a, b) を満たす整数の組 (a, b) に対して, (x, y, gcd(a, b)) を求める.
template<integral T>
tuple<T, T, T> Extended_Euclid(T a, T b) {
    T s = 1, t = 0, u = 0, v = 1;
    while (b) {
        auto [q, r] = divmod(a, b);
        a = b;
        b = r;
        tie(s, t) = make_pair(t, s - q * t);
        tie(u, v) = make_pair(v, u - q * v);
    }

    return make_tuple(s, u, a);
}

// floor(sqrt(N)) を求める (N < 0 のときは, 0 とする).
ll isqrt(const ll &N) { 
    if (N <= 0) { return 0; }

    ll x = sqrtl(N);
    while ((x + 1) * (x + 1) <= N) { x++; }
    while (x * x > N) { x--; }

    return x;
}

// floor(sqrt(N)) を求める (N < 0 のときは, 0 とする).
ll floor_sqrt(const ll &N) { return isqrt(N); }

// ceil(sqrt(N)) を求める (N < 0 のときは, 0 とする).
ll ceil_sqrt(const ll &N) {
    ll x = isqrt(N);
    return x * x == N ? x : x + 1;
}
#line 64 "template/template.hpp"

// inout
#line 1 "template/inout.hpp"
// 入出力
template<class... T>
void input(T&... a){ (cin >> ... >> a); }

void print(){ cout << "\n"; }

template<class T, class... Ts>
void print(const T& a, const Ts&... b){
    cout << a;
    (cout << ... << (cout << " ", b));
    cout << "\n";
}

template<typename T, typename U>
istream &operator>>(istream &is, pair<T, U> &P){
    is >> P.first >> P.second;
    return is;
}

template<typename T, typename U>
ostream &operator<<(ostream &os, const pair<T, U> &P){
    os << P.first << " " << P.second;
    return os;
}

template<typename T>
vector<T> vector_input(int N, int index){
    vector<T> X(N+index);
    for (int i=index; i<index+N; i++) cin >> X[i];
    return X;
}

template<typename T>
istream &operator>>(istream &is, vector<T> &X){
    for (auto &x: X) { is >> x; }
    return is;
}

template<typename T>
ostream &operator<<(ostream &os, const vector<T> &X){
    int s = (int)X.size();
    for (int i = 0; i < s; i++) { os << (i ? " " : "") << X[i]; }
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const unordered_set<T> &S){
    int i = 0;
    for (T a: S) {os << (i ? " ": "") << a; i++;}
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const set<T> &S){
    int i = 0;
    for (T a: S) { os << (i ? " ": "") << a; i++; }
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const unordered_multiset<T> &S){
    int i = 0;
    for (T a: S) { os << (i ? " ": "") << a; i++; }
    return os;
}

template<typename T>
ostream &operator<<(ostream &os, const multiset<T> &S){
    int i = 0;
    for (T a: S) { os << (i ? " ": "") << a; i++; }
    return os;
}

template<typename T>
std::vector<T> input_vector(size_t n, size_t offset = 0) {
    std::vector<T> res;
    // 最初に必要な全容量を確保（再確保を防ぐ）
    res.reserve(n + offset);
    // offset 分をデフォルト値で埋める（特別 indexed 用）
    res.assign(offset, T());
    
    for (size_t i = 0; i < n; ++i) {
        T el;
        if (!(std::cin >> el)) break;
        res.push_back(std::move(el));
    }
    return res;
}
#line 67 "template/template.hpp"

// macro
#line 2 "template/macro.hpp"

// マクロの定義
#define all(x) x.begin(), x.end()
#define len(x) ll(x.size())
#define elif else if
#define unless(cond) if (!(cond))
#define until(cond) while (!(cond))
#define loop while (true)

// オーバーロードマクロ
#define overload2(_1, _2, name, ...) name
#define overload3(_1, _2, _3, name, ...) name
#define overload4(_1, _2, _3, _4, name, ...) name
#define overload5(_1, _2, _3, _4, _5, name, ...) name

// 繰り返し系
#define rep1(n) for (ll i = 0; i < n; i++)
#define rep2(i, n) for (ll i = 0; i < n; i++)
#define rep3(i, a, b) for (ll i = a; i < b; i++)
#define rep4(i, a, b, c) for (ll i = a; i < b; i += c)
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)

#define foreach1(x, a) for (auto &&x: a)
#define foreach2(x, y, a) for (auto &&[x, y]: a)
#define foreach3(x, y, z, a) for (auto &&[x, y, z]: a)
#define foreach4(x, y, z, w, a) for (auto &&[x, y, z, w]: a)
#define foreach(...) overload5(__VA_ARGS__, foreach4, foreach3, foreach2, foreach1)(__VA_ARGS__)
#line 70 "template/template.hpp"

// bitop
#line 2 "template/bitop.hpp"

// 非負整数 x の bit legnth を求める.
ll bit_length(ll x) {
    if (x == 0) { return 0; }
    return (sizeof(long) * CHAR_BIT) - __builtin_clzll(x);
}

// 非負整数 x の popcount を求める.
ll popcount(ll x) { return __builtin_popcountll(x); }

// 正の整数 x に対して, floor(log2(x)) を求める.
ll floor_log2(ll x) { return bit_length(x) - 1; }

// 正の整数 x に対して, ceil(log2(x)) を求める.
ll ceil_log2(ll x) { return bit_length(x - 1); }

// x の第 k ビットを取得する
int get_bit(ll x, int k) { return (x >> k) & 1; }

// x のビット列を取得する.
// k はビット列の長さとする.
vector<int> get_bits(ll x, int k) {
    vector<int> bits(k);
    rep(i, k) {
        bits[i] = x & 1;
        x >>= 1;
    }

    return bits;
}

// x のビット列を取得する.
vector<int> get_bits(ll x) { return get_bits(x, bit_length(x)); }
#line 73 "template/template.hpp"

// exception
#line 2 "template/exception.hpp"

class NotExist: public exception {
    private:
    string message;

    public:
    NotExist() : message("求めようとしていたものは存在しません.") {}

    const char* what() const noexcept override {
        return message.c_str();
    }
};
#line 4 "Summation/Summation.hpp"

namespace summation {
    /**
     * @brief 等差数列の和を求めます.
     * 
     * sum_{k=l}^{r} (a * k + b)
     * 
     * @tparam T 整数型
     * @param a 係数 a
     * @param b 定数 b
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T linear(T a, T b, T l, T r) {
        if (l > r) return 0;

        T num_terms = r - l + 1;
        T sum_k = num_terms * (l + r) / 2;
        T sum_b = num_terms * b;
        return a * sum_k + sum_b;
    }

    /**
     * @brief ReLU(一次関数) の和を求めます.
     * 
     * sum_{k=l}^{r} max(0, a * k + b)
     * 
     * @tparam T 整数型
     * @param a 係数 a
     * @param b 定数 b
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T relu_linear(T a, T b, T l, T r) {
        if (a == 0) { return (r - l + 1) * max<T>(0, b); }
        if (a < 0) { return relu_linear(-a, a * (r + l) + b, l, r); }

        // ここに到達した段階で a > 0 が保証される

        if (a * r + b <= 0) return 0;

        T t = max<T>(l, div_ceil(-b, a));

        return linear(a, b, t, r);
    }

    /**
     * @brief 2つの一次関数の最大値の和を求めます.
     * 
     * sum_{k=l}^{r} max(a * k + b, c * k + d)
     * 
     * @tparam T 整数型
     * @param a 第1の一次関数の係数 a
     * @param b 第1の一次関数の定数 b
     * @param c 第2の一次関数の係数 c
     * @param d 第2の一次関数の定数 d
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T max_linear(T a, T b, T c, T d, T l, T r) {
        return relu_linear(a - c, b - d, l, r) + linear(c, d, l, r);
    }

    /**
     * @brief 一次関数と定数の最大値の和を求めます.
     * 
     * sum_{k=l}^{r} max(a * k + b, d)
     * 
     * @tparam T 整数型
     * @param a 係数 a
     * @param b 定数 b
     * @param d 定数 d
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T max_linear(T a, T b, T d, T l, T r) { return max_linear(a, b, 0, d, l, r); }

    /**
     * @brief 2つの一次関数の最小値の和を求めます.
     * 
     * sum_{k=l}^{r} min(a * k + b, c * k + d)
     * 
     * @tparam T 整数型
     * @param a 第1の一次関数の係数 a
     * @param b 第1の一次関数の定数 b
     * @param c 第2の一次関数の係数 c
     * @param d 第2の一次関数の定数 d
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T min_linear(T a, T b, T c, T d, T l, T r) {
        return -relu_linear(-(a - c), -(b - d), l, r) + linear(c, d, l, r);
    }

    /**
     * @brief 一次関数と定数の最小値の和を求めます.
     * 
     * sum_{k=l}^{r} min(a * k + b, d)
     * 
     * @tparam T 整数型
     * @param a 係数 a
     * @param b 定数 b
     * @param d 定数 d
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T min_linear(T a, T b, T u, T l, T r) { return min_linear(a, b, 0, u, l, r); }

    /**
     * @brief 一次関数の絶対値の和を求めます.
     * 
     * sum_{k=l}^{r} |a * k + b|
     * 
     * @tparam T 整数型
     * @param a 係数 a
     * @param b 定数 b
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T abs_linear(T a, T b, T l, T r) {
        return max_linear(a, b, -a, -b, l, r);
    }

    /**
     * @brief 一次関数を範囲 [d, u] に収めた値の和を求めます.
     * 
     * sum_{k=l}^{r} clamp(a * k + b, d, u)
     * clamp(x, d, u) = min(max(x, d), u)
     * 
     * @tparam T 整数型
     * @param a 係数 a
     * @param b 定数 b
     * @param d 下限値 d
     * @param u 上限値 u
     * @param l 総和の下限
     * @param r 総和の上限
     * @return T 総和
     */
    template<typename T>
    T clamp_linear(T a, T b, T d, T u, T l, T r) {
        return linear(a, b, l, r) - relu_linear(a, b - u, l, r) + relu_linear(-a, d - b, l, r);
    }

    /**
     * @brief 0 から n までの二乗和を求めます.
     * 
     * sum_{k=0}^{n} k^2
     * 
     * @tparam T 整数型
     * @param n 上限
     * @return T 総和
     */
    template<typename T>
    T sum_sq(T n) {
        if (n < 0) return 0;
        return n * (n + 1) * (2 * n + 1) / 6;
    }

    /**
     * @brief 一次関数の二乗の和を求めます.
     * 
     * sum_{k=l}^{r} (a * k + b)^2
     * 
     * @tparam T 整数型
     * @param a 係数 a
     * @param b 定数 b
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T square_linear(T a, T b, T l, T r) {
        if (l > r) return 0;
        T sum_1 = (r - l + 1) * (l + r) / 2;
        T sum_2 = sum_sq(r) - sum_sq(l - 1);
        return a * a * sum_2 + 2 * a * b * sum_1 + b * b * (r - l + 1);
    }

    /**
     * @brief 2つの一次関数の積の和を求めます.
     * 
     * sum_{k=l}^{r} (a * k + b) * (c * k + d)
     * 
     * @tparam T 整数型
     * @param a 第1の一次関数の係数 a
     * @param b 第1の一次関数の定数 b
     * @param c 第2の一次関数の係数 c
     * @param d 第2の一次関数の定数 d
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T sum_product_linear(T a, T b, T c, T d, T l, T r) {
        if (l > r) return 0;
        T n = r - l + 1;
        T sum_k = n * (l + r) / 2;
        T sum_k2 = sum_sq(r) - sum_sq(l - 1);
        return a * c * sum_k2 + (a * d + b * c) * sum_k + b * d * n;
    }

    /**
     * @brief 2つの ReLU(一次関数) の積の和を求めます.
     * 
     * sum_{k=l}^{r} relu(a * k + b) * relu(c * i + d)
     * 
     * @tparam T 整数型
     * @param a 第1の一次関数の係数 a
     * @param b 第1の一次関数の定数 b
     * @param c 第2の一次関数の係数 c
     * @param d 第2の一次関数の定数 d
     * @param l 下限
     * @param r 上限
     * @return T 総和
     */
    template<typename T>
    T product_relu_linear(T a, T b, T c, T d, T l, T r) {
        T L = l, R = r;

        if (a > 0) L = max<T>(L, div_ceil(-b, a));
        else if (a < 0) R = min<T>(R, div_floor(-b, a));
        else if (b < 0) return 0;

        if (c > 0) L = max<T>(L, div_ceil(-d, c));
        else if (c < 0) R = min<T>(R, div_floor(-d, c));
        else if (d < 0) return 0;

        if (L > R) return 0;

        return sum_product_linear(a, b, c, d, L, R);
    }
}
#line 6 "Summation/Counting.hpp"

namespace counting {
    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: l <= x <= r, ax + b <= y <= cx + d
     * 
     * @tparam T 整数型
     * @param a 不等式 ax + b <= y の係数 a
     * @param b 不等式 ax + b <= y の定数 b
     * @param c 不等式 y <= cx + d の係数 c
     * @param d 不等式 y <= cx + d の定数 d
     * @param l x の下限
     * @param r x の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_between_lines(T a, T b, T c, T d, T l, T r) {
        return summation::relu_linear(c - a, d - b + 1, l, r);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: l <= x <= r, ax + b <= y <= cx + d, ex + f <= y <= gx + h
     * 
     * @tparam T 整数型
     * @param a 不等式 ax + b <= y の係数 a
     * @param b 不等式 ax + b <= y の定数 b
     * @param c 不等式 y <= cx + d の係数 c
     * @param d 不等式 y <= cx + d の定数 d
     * @param e 不等式 ex + f <= y の係数 e
     * @param f 不等式 ex + f <= y の定数 f
     * @param g 不等式 y <= gx + h の係数 g
     * @param h 不等式 y <= gx + h の定数 h
     * @param l x の下限
     * @param r x の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_between_lines(T a, T b, T c, T d, T e, T f, T g, T h, T l, T r) {
        // 各 x に対して y の個数は
        // max(0, min(cx + d, gx + h) - max(ax + b, ex + f) + 1)
        // = max(0, min(
        //    (c - a)x + (d - b + 1),
        //    (c - e)x + (d - f + 1),
        //    (g - a)x + (h - b + 1),
        //    (g - e)x + (h - f + 1)
        // ))
        // となる. 4つの直線の最小値(ただし負なら0)の和を求める問題に帰着される.

        struct Line { T k, m; };
        std::vector<Line> lines = {
            {c - a, d - b + 1},
            {c - e, d - f + 1},
            {g - a, h - b + 1},
            {g - e, h - f + 1}
        };

        T L = l, R = r;

        // 値が負になる範囲を除外する (L_i(x) >= 0 を満たす範囲に限定)
        for (const auto& line : lines) {
            if (line.k == 0) {
                if (line.m < 0) return 0;
            } else if (line.k > 0) {
                L = max<T>(L, div_ceil(-line.m, line.k));
            } else {
                R = min<T>(R, div_floor(-line.m, line.k));
            }
        }

        if (L > R) return 0;

        vector<T> points;
        points.push_back(L - 1);
        points.push_back(R);

        // 直線の交点を求めて区間を分割する
        for (size_t i = 0; i < lines.size(); ++i) {
            for (size_t j = i + 1; j < lines.size(); ++j) {
                if (lines[i].k == lines[j].k) continue;

                T num = lines[j].m - lines[i].m;
                T den = lines[i].k - lines[j].k;
                T x = div_floor(num, den);
                if (x >= L - 1 && x < R) points.push_back(x);
            }
        }

        sort(points.begin(), points.end());
        points.erase(unique(points.begin(), points.end()), points.end());

        T total = 0;
        for (size_t i = 0; i < points.size() - 1; ++i) {
            T seg_l = points[i] + 1;
            T seg_r = points[i+1];
            
            // 区間内では最小値を与える直線は変わらないので, 先頭の点で判定する
            T min_val = lines[0].k * seg_l + lines[0].m;
            size_t min_idx = 0;
            for (size_t j = 1; j < lines.size(); ++j) {
                T val = lines[j].k * seg_l + lines[j].m;
                if (val < min_val) {
                    min_val = val;
                    min_idx = j;
                }
            }
            
            total += summation::linear(lines[min_idx].k, lines[min_idx].m, seg_l, seg_r);
        }

        return total;
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, x + y = k
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param k 和 k
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_sum_eq(T p, T q, T r, T s, T k) {
        return count_sum_range(p, q, r, s, k, k);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, a <= x + y <= b
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param a 和 x + y の下限
     * @param b 和 x + y の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_sum_range(T p, T q, T r, T s, T a, T b) {
        return count_between_lines<T>(0, r, 0, s, -1, a, -1, b, p, q);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, x - y = k
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param k 差 k
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_diff_eq(T p, T q, T r, T s, T k) {
        return count_sum_eq(p, q, -s, -r, k);
    }

    /**
     * @brief 以下の条件を満たす整数の組 (x, y) の個数を求めます.
     * 
     * 条件: p <= x <= q, r <= y <= s, a <= x - y <= b
     * 
     * @tparam T 整数型
     * @param p x の下限
     * @param q x の上限
     * @param r y の下限
     * @param s y の上限
     * @param a 差 x - y の下限
     * @param b 差 x - y の上限
     * @return T 条件を満たす (x, y) の個数
     */
    template<typename T>
    T count_diff_range(T p, T q, T r, T s, T a, T b) {
        return count_sum_range(p, q, -s, -r, a, b);
    }
}

Back to top page