[Из песочницы] Сумма сумм арифметических прогрессий

Пускай у нас есть некий ряд ячеек, часть которых можно пометить как «занятые»:

01


Нам нужно узнать, сколько всего существует вариантов расположения занятых ячеек.

К этой схеме сводится множество задач. Например, разбиение периода из N + 1 календарных дней на l + 1 следующих друг за другом меньших периодов. Допустим, мы хотим провести оптимизационный расчет методом «грубой силы», рассчитав целевую функцию для каждого возможного варианта разбиения периода, чтобы выбрать наилучший вариант. Чтобы заранее оценить время расчета, нужно определить количество вариантов. Это поможет принять решение, стоит ли вообще начинать расчет. Согласитесь — полезно будет заранее предупредить пользователя вашей программы, что с теми параметрами, которые он задал, расчет займет 10000 лет.

Существует простая формула, позволяющая для любых N и l получить количество всех возможных вариантов расположения занятых ячеек:

$S_l(n) = \prod_{i=1}^l\frac{n+i-1}{i}$

$S_0(n) =1$

, где n = N — l + 1, n ≠ 0.

Это формула суммы сумм сумм … сумм арифметических прогрессий вида 1 + 2 + 3 + … + n; слово «сумма» в разных склонениях в этой фразе повторяется ровно l — 1 раз. Другими словами, эта формула позволяет быстро вычислять вот такие суммы:

$S_l(n) = \sum_{i=1}^nS_{l-1}(n)$

$S_0(n) = 1$


У обычного настольного ПК уходит слишком много времени на вычисление этих сумм «в лоб» (особенно с использованием длинной арифметики —, а без нее не обойтись).

Когда я только начинал решать эту задачу (и еще не знал этой формулы), я забивал в поисковик фразу «сумма сумм арифметических прогрессий» и ничего не находил. Именно так я и назвал свою статью — в расчете, что следующий искатель этой формулы сможет быстро найти ее здесь.

Ссылки на авторитетный источник касательно этой формулы у меня нет. Я ее вывел аналитически и проверил численно, проведя свыше 4000 тестов. Ее вывод, если кому интересно, приведен в конце статьи.

Скоро я расскажу, какое отношение имеют арифметические прогрессии к подсчету вариантов расположения ячеек, а пока приведу еще несколько готовых формул.

Введем дополнительное условие: допустим, в любом раскладе должна быть хотя бы 1 группа из r смежных незанятых ячеек. В этом случае, количество вариантов определяется так:

$S_l^r(n) = (l+1)\cdot S_l(n-r)-\sum_{j=0}^{l-1}\sum_{i=r}^{n-r-1}\left[S_j^r(i+1)\cdot S_{l-j-1}(n-r-i)\right]$

$S_l^r(n) = S_l(n),\;\;\;\; при\;\;\; n+l-1\geq r(l+1)$

$S_l^0(n) = S_l(n)$

$S_l^1(n) = \begin{cases} 0 & \text{, при n = 1} \\ S_l(n) & \text{, при n > 1} \end{cases}$» /></p> <p><img src=

$S_0^r(n) = \begin{cases} 1 & \text{, при n > r} \\ 0 & \text{, при n ≤ r} \end{cases}$» /></p><p>Вывод этой формулы следует в самом конце статьи.</p>

<p>Если ваша программа проводит какие-либо расчеты для каждого варианта расположения ячеек, то расчеты для разных вариантов наверняка однотипные. Это значит, что их легко распределить по нескольким потокам. В статье будет рассказано, как организовать итерацию по вариантам, чтобы каждому потоку досталось равное количество вариантов.</p>

<h4>1. Причем тут прогрессии? </h4>

<p><br />
Это самый легкий вопрос. Сгруппируйте все занятые ячейки в левом конце вашего ряда ячеек — чтобы они следовали одна за другой, начиная с крайней левой ячейки в ряду: </p>

<div><img src=


Сколько существует вариантов расположения самой правой из занятых ячеек (будем называть ее последней занятой ячейкой), при условии, что остальные ячейки остаются на своих местах? Ровно n. Теперь сдвиньте ПРЕДпоследнюю занятую ячейку вправо на 1 позицию. Сколько останется позиций для последней ячейки? n — 1. Предпоследнюю ячейку можно сдвинуть вправо n раз. Значит, вариантов расположения двух последних занятых ячеек будет вот столько: S2(n) = 1 + 2 + 3 + … + n. Сдвиньте третью справа занятую ячейку на 1 позицию и снова подсчитайте количество вариантов расположения последней и предпоследней ячеек. Получится S2(n — 1). Третью ячейку тоже можно сдвинуть вправо только n раз, и всего вариантов расположения трех последних ячеек будет S3(n) = S2(1) + S2(2) + … + S2(n).

Рассуждая так, доберемся наконец и до количества вариантов расположения всех l занятых ячеек:

$S_l(n) = \sum_{i=1}^nS_{l-1}(n)$

$S_0(n) = 1$

2. Дополнительное условие: хотя бы 1 группа из r незанятых смежных ячеек


Сначала рассмотрим частные случаи. Если n + l — 1 ≥ r (l + 1), то ни при каком раскладе мы не получим варианта, в котором нет хотя бы 1 группы из r незанятых ячеек. Даже в самом худшем случае — когда занятые ячейки в ряду расположены равномерно, так чтобы расстояние между ними и краями ряда было равно r — 1, количества занятых ячеек просто не хватит на то, чтобы ВСЕ эти расстояния были не больше r — 1. Хотя бы одно будет равно как минимум r. Следовательно:

$S_l^r(n) = S_l(n),\;\;\;\; при\;\;\; n+l-1\geq r(l+1)$

r = 0 означает, что нас устраивает любое расположение ячеек. Условие «n + l — 1 ≥ r (l + 1)» в этом случае выполняется при любых n и l:

$S_l^0(n) = S_l(n)$

Если же r = 1, то условие «n + l — 1 ≥ r (l + 1)» выполняется только при n > 1:

$S_l^1(n) = \begin{cases} 0 & \text{, при n = 1} \\ S_l(n) & \text{, при n > 1} \end{cases}$» /></p><p>n ≤ r означает, что число незанятых ячеек меньше r, и нет никакой возможности получить r незанятых ячеек, смежных или нет. Поэтому в предыдущей формуле и стоит 0 при n = 1.</p>

<p><img src=

Если занятых ячеек нет, то вариант их расположения всего один:

$S_0^r(n) = \begin{cases} 1 & \text{, при n > r} \\ 0 & \text{, при n ≤ r} \end{cases}$» /></p><p>Теперь — обобщенная формула для всех остальных случаев.</p>

<p>Если зафиксировать самую левую занятую ячейку в крайнем левом положении, вариантов расположения всех остальных ячеек будет Sr (l — 1) (n). Если сдвинуть ее на 1 позицию вправо, вариантов будет Sr (l — 1) (n — 1). Начиная с позиции r + 1, у нас всегда будет промежуток длиной не менее r в левом конце ряда, так что остальные варианты можно будет посчитать без рекурсии: Sl (n — r). Получается вот что: </p>

<p><img src=

Формула верная, но я предпочитаю пользоваться вот этой:

$S_l^r(n) = (l+1)\cdot S_l(n-r)-\sum_{j=0}^{l-1}\sum_{i=r}^{n-r-1}\left[S_j^r(i+1)\cdot S_{l-j-1}(n-r-i)\right]$

По моим замерам, она в десятки раз быстрее предыдущей. Выводить ее несложно, но описывать вывод долго. В конце статьи расскажу.

Теперь мы можем перейти к распределению итераций по потокам. Чтобы разобраться в этом, вы должны ясно понимать, как выводилась первая формула Srl (n), т.к. в этом выводе заложен определенный порядок итерации по вариантам расположения занятых ячеек.

3. Как организовать итерацию в нескольких потоках


Порядок итерации будет общим для всех потоков. Вначале все l ячеек располагаются в левом конце ряда, занимая позиции с 1 по l. Это первая итерация. Крайняя правая ячейка на каждой итерации сдвигается вправо на 1 позицию, пока не окажется в конце ряда, затем ячейка слева от нее сдвигается на 1 позицию вправо, и крайняя ячейка снова проходит все возможные положения между ячейкой слева и правым концом ряда. Когда обе ячейки оказываются в крайнем правом положении, ячейка слева от них сдвигается на 1 позицию вправо. И так пока все ячейки не окажутся в правом конце ряда. При итерации мы пропускаем варианты, в которых нет ни одной группы из r смежных незанятых ячеек.

Выберем согласно этому порядку итерации первые k вариантов и назначим их первому потоку, затем следующие k вариантов назначим второму потоку, и так далее. Допустим, мы решили задействовать nt потоков, количество итераций в каждом из которых мы определили как ki, i = 1, …, nt.

Порядковый номер первой итерации для каждого потока назовем hi:

$h_i=1+\sum_{j=1}^{i-1}k_j$

Имея начальное расположение ячеек для варианта под номером hi, не составит труда провести ki итераций, начиная с этого варианта (я даже описывать не буду, как это делается). Однако нам понадобится функция, вычисляющая положение занятых ячеек по порядковому номеру варианта:

using Anylong = boost::multiprecision::cpp_int;
using Opt_UVec = boost::optional >;
Opt_UVec arrangement(Anylong index, unsigned n, unsigned l, unsigned r = 0);


Функция arrangement принимает номер варианта index и параметры ряда: n = N — l + 1, l и r, —, а возвращает вектор с позициями занятых ячеек в ряду. Позиция занятой ячейки — это целое число от 1 до N.

Число вариантов очень быстро растет с увеличением параметров l и n, поэтому для представления этого числа нам требуется длинная арифметика. Я использовал класс boost: multiprecision: cpp_int, способный представлять целые числа любой длины.

Если параметр index превышает число возможных вариантов расположения ячеек, функция возвращает пустой объект boost: optional. Если параметр index или параметр n равен 0, это рассматривается как ошибка программиста, и функция генерирует исключение.

Сначала переведем на C++ формулы для определения числа вариантов:

struct Exception
{
    enum class Code { INVALID_INPUT = 1 } code;

    Exception(Code c) : code(c) {}
    static Exception InvalidInput() { return Exception(Code::INVALID_INPUT); }
};

Anylong S(unsigned n, unsigned l)
{
    if (!n) throw Exception::InvalidInput();
    if (l == 1) return n;
    if (n == 1 || l == 0) return 1;
    Anylong res = 1;
    for (unsigned i = 1; i <= l; ++i) {
        res = res * (n + i - 1) / i; // порядок действий важен!
    }
    return res;
}

Anylong S(unsigned n, unsigned l, unsigned r)
{
    if (!n) throw Exception::InvalidInput();
    if (r == 0) return S(n, l);
    if (r == 1) if (n == 1) return 0; else return S(n, l);
    if (n + l - 1 >= r * (l + 1)) return S(n, l);
    if (n <= r) return 0; else if (l == 0) return 1;

    Anylong res = (l + 1) * S(n - r, l);
    for (unsigned j = 0; j <= l - 1; ++j)
        for (unsigned i = r; i <= n - r - 1; ++i)
            res -= S(i + 1, j, r) * S(n - r - i, l - j - 1);
    return res;
}


Обратите внимание на эту строчку:

        res = res * (n + i - 1) / i;


Здесь важен порядок действий. Произведение множителей res и (n + i — 1) всегда делится нацело на i, а каждый из них в отдельности — нет. Нарушение порядка действий приведет к искажению результатов.

Теперь стоит вопрос, как определить вариант расположения по индексу.

Вспомним принятый нами порядок итерации. Чтобы сдвинуть крайнюю левую ячейку на 1 позицию вправо, потребуется X1=Srl — 1 (n) итераций. Сдвинуть ее еще на 1 ячейку правее можно за X2=Srl — 1 (n — 1) итераций. Если в какой-то момент X1 + X2 + … + Xi + Xi + 1 оказалось больше, чем index, значит, мы только что определили положение первой занятой ячейки. Она должна находиться на i-й позиции. Далее вычисляем, сколько итераций требуется, чтобы сдвинуть вправо вторую ячейку: X1=Srl — 2 (n — i + 1). Так продолжается, пока мы не доберемся до варианта под номером index. Если хоть раз i оказалось больше r, все последующие вызовы Srl (n) заменяются на Sl (n), ведь у нас уже есть как минимум один промежуток длиной не меньше r слева от текущей ячейки.

Короче будет написать код, чем объяснять словами.

Opt_UVec arrangement(Anylong index, unsigned n, unsigned l, unsigned r = 0)
{
    if (index == 0) throw Exception::InvalidInput();
    if (index > S(n, l, r)) return {};
    std::vector oci(l); /* occupied cells indices - позиции
                              занятых ячеек в ряду (индексы от 1 до N) */
    if (l == 0) return oci;
    if (l == 1) {
        assert(index <= std::numeric_limits::max());
        oci[0] = (unsigned)index;
        return oci;
    }
    oci[0] = 1;

    unsigned N = n + l - 1;
    unsigned prev = 1;
    
    Anylong i = 0; auto it = oci.begin();
    while (true) {
        auto s = S(n, --l, r);
        while (i + s <= index) {
            if ((i += s) == index) {
                auto it1 = --oci.end();
                if (it1 == it) return oci;
                *it1 = N;
                for (it1--; it1 != it; it1--) *it1 = *(it1 + 1) - 1;
                return oci;
            }
            s = S(--n, l, r);
            assert(n);
            (*it)++;
            if (r && (*it) > prev + r - 1) r = 0;
        }
        prev = *it++;
        assert(it != oci.end());
        *it = prev + 1;
    }
    assert(false);
}


4. Вывод формул


Порадую любителей школьной алгебры еще одним разделом.

Начнем с суммы сумм арифметических прогрессий:

$S_l(n) = \sum_{i=1}^nS_{l-1}(n)$

$S_0(n) = 1$

Как видим, при l = 0 S0 (n) = 1 — это многочлен 0-й степени. При l = 1 S1 (n) = n — это многочлен 1-й степени. При l = 2 используется известная формула суммы арифметической прогрессии S2 (n) = n (n + 1)/2. И это многочлен 2-й степени. Можно предположить, что при l = 3 у нас будет многочлен 3-й степени. Вычислим вручную значения S3 (n) при n = 1, 2, 3 и 4, составим систему линейных уравнений и найдем коэффициенты этого многочлена. Результат разложим на множители и получим вот это:

$S_3(n)=\frac{n(n+1)(n+2)}{6}$

У нас тут очевидная закономерность. Запишем вот такую формулу:

$S_l(n) = \prod_{i=1}^l\frac{n+i-1}{i}$

Она еще не доказана для l = 3, но уже доказана для l = 1 и l = 2. Воспользуемся методом математической индукции. Предположим, что формула Sl — 1 (n) верна для любых n и докажем, что в этом случае верна и формула Sl (n).

$S_l(n)=S_{l-1}(n)+S_{l-1}(n-1)+...+S_{l-1}(1)=\prod_{i=1}^{l}\frac{n+i-1}{i}$

$S_l(n-1)=S_{l-1}(n-1)+S_{l-1}(n-2)+...+S_{l-1}(1)=\prod_{i=1}^{l}\frac{n+i-2}{i}$

Вычтем второе из первого:

$S_{l-1}(n)=\frac{\prod_{i=1}^{l}(n+i-1)}{l!}-\frac{\prod_{i=0}^{l-1}(n+i-1)}{l!}$

$S_{l-1}(n)=\left[\prod_{i=1}^{l-1}\frac{n+i-1}{i}\right]\times\frac{(n+l-1)-(n+0-1)}{l}$

$S_{l-1}(n)=\prod_{i=1}^{l-1}\frac{n+i-1}{i}$

Получилось верное тождество.

Теперь вернемся к формуле подсчета вариантов при r ≠ 0:

$S_l^r(n) = (l+1)\cdot S_l(n-r)-\sum_{j=0}^{l-1}\sum_{i=r}^{n-r-1}\left[S_j^r(i+1)\cdot S_{l-j-1}(n-r-i)\right]$

Выводится она следующим образом.

Подсчитаем количество вариантов, когда промежуток между последней занятой ячейкой и правым концом ряда больше или равен r. Это как если бы последняя занятая ячейка выросла в размерах до r + 1, а размеры ряда при этом остались бы прежними:

03

Такое количество вариантов равно Sl (n — r).

Теперь прибавим к нему количество вариантов, в которых промежуток между последней и предпоследней занятыми ячейками больше или равен r. Это количество вариантов снова будет равно Sl (n — r). Но некоторые из этих вариантов мы уже посчитали, когда вычисляли предыдущие Sl (n — r). А именно — те варианты, в которых промежуток между последней занятой ячейкой и правым концом ряда больше или равен r. Значит, к первым Sl (n — r) вариантов нужно прибавить не Sl (n — r), а Sl (n — r) — X, где X — количество вариантов, в которых промежуток между последней и предпоследней занятыми ячейками больше или равен r, равно как и промежуток между последней занятой ячейкой и правым концом ряда.

Ровно то же самое придется сделать для каждой j-й занятой ячейки — из Sl (n — r) вычесть Xj, равное количеству вариантов, в которых промежуток между j-й ячейкой и ячейкой слева от нее (в случае крайней левой ячейки — промежуток между нею и левым концом ряда), а также хотя бы один из промежутков справа от j-й ячейки — больше или равны r.

Всего у нас l — 1 промежутков между занятыми ячейками, плюс 2 промежутка между занятой ячейкой и концом ряда. Значит, слагаемое Sl (n — r) входит в нашу формулу l + 1 раз. А вот Xj не вычисляется для промежутка между крайней правой занятой ячейкой и правым коном ряда. Значит, этих слагаемых будет только l.

$S_l^r(n)=(l+1)S_l(n-r)-\sum_{j=0}^{l-1}X_j$

j находится в диапазоне от 0 до l — 1. Такой уж я принял порядок индексации. Тогда количество занятых ячеек справа от j-й ячейки равно j.

Посмотрите на этот рисунок:

03


j-я ячейка (та, что увеличивается до размера r + 1 при итерации по j) обозначается как M. Она может находиться в n — r возможных положениях, которые определяются параметром i (i ∈ [0, n — r — 1]). Первые r положений можно не рассматривать, поскольку при i < r все промежутки справа от M будут меньше r. Следовательно, i ∈ [r, n – r – 1].

Количество вариантов расположения ячеек справа от M, в которых хотя бы 1 промежуток больше или равен r, равно Srj (i + 1). Количество вариантов расположения ячеек слева от M равно Sl — j — 1 (n — i — r). Всего вариантов получается Srj (i + 1)Sl — j — 1 (n — i — r) для всех i ∈ [r, n — r — 1] и всех j ∈ [0, l — 1]:

$S_l^r(n) = (l+1)\cdot S_l(n-r)-\sum_{j=0}^{l-1}\sum_{i=r}^{n-r-1}\left[S_j^r(i+1)\cdot S_{l-j-1}(n-r-i)\right]$

Это и есть наша формула.

© Habrahabr.ru