Алгоритм

Припустимо, нам задано різних цілих чисел , такі, що для деякої постійної , і для всіх . $n$ $a_1, a_2, \dots, a_n$ $0 \le a_i \le kn$ $k \gt 0$ $i$

Нам цікаво знайти підрахунок усіх можливих парних сум . ( дозволено). $S_{ij} = a_i + a_j$ $i = j$

Один алгоритм полягає в тому, щоб побудувати поліном ступеня , і обчислити його квадрат за допомогою методу перетворення Фур'є і зчитувати повноваження з їх коефіцієнти в отриманому многочлені. Це алгоритм часу . $P(x) = \sum_{j=1}^{n} x^{a_j}$ $\le kn$ $O(n \log n)$

У мене є два питання:

Чи існує алгоритм , який не використовує FFT? $O(n \log n)$
Чи відомі кращі алгоритми (тобто )? (FFT дозволено). $o(n \log n)$

algorithms time-complexity fourier-transform

— Ар'ябхата
джерело

Чому важливо не використовувати FFT? Здається, у вас вже є гарне рішення вашої проблеми. Звідки випливає вимога не використовувати FFT? Це для мене звучить як досить неприродна вимога.

— DW

@DW: Тому що тоді не буде запитання? :-) Мені просто цікаво знати, чи є інший підхід.

— Арябхата

Добре, зрозумів! Зізнаюся, мені теж цікаво. :-) Дякую за цікаве запитання.

— DW

@DW: Ласкаво просимо :-)

— Ар'ябхата

Здавалося б, ця проблема еквівалентна цілочисленному / многочленному квадратуванню:

1. Відомо, що множення полінома еквівалентно цілому множенню .

2. Очевидно, ви вже звели цю проблему до поліноміального / цілого квадратування; отже, ця проблема є не менш важкою, як квадрати.

Тепер я зменшу цілісні квадрати до цієї проблеми:

Припустимо, у вас був алгоритм:

F (\vec{a}) \to P^{2} (x), where P (x) = \sum_{a_{i} \in \vec{a}} x^{a_{i}}

$F(\mathbf{\vec a})\rightarrow P^2(x),\\ \text{where } P(x)=\sum_{a_i \in \mathbf{\vec a}} x^{a_i}$

Цей алгоритм по суті є алгоритмом, про який ви запитуєте у своєму запитанні. Таким чином, якщо у мене був магічний алгоритм, який може це зробити, я можу зробити функцію, яка буде квадратним цілим числом ( о так, я люблю mathjax: P ): ${\rm S{\small QUARE}}\left(y\right)$ $y$

\begin{aligned} Algorithm 1 Squaring \\ 1. : & procedure S Q U A R E (y) : \\ 2. : & \vec{a} \leftarrow () & ▹ \vec{a} starts as empty polynomial sequence \\ 3. : & i \leftarrow 0 \\ 4. : & w h i l e y \neq 0 d o & ▹ break y down into a polynomial of base 2 \\ 5. : & i f y & 1 t h e n & ▹ if lsb of y is set \\ 6. : & \vec{a} \leftarrow \vec{a} i & ▹ append i to \vec{a} (appending x^{i}) \\ 7. : & e n d i f \\ 8. : & i \leftarrow i + 1 \\ 9. : & y \leftarrow y ≫ 1 & ▹ shift y right by one \\ 10. : & e n d w h i l e \\ 11. : & P^{2} (x) \leftarrow F (\vec{a}) & ▹ obtain the squared polynomial via F (\vec{a}) \\ 12. : & r e t u r n P^{2} (2) & ▹ simply sum up the polynomial \\ 13. : & end procedure \end{aligned}

$\begin{array}{rlr}\hline &\mathbf{\text{Algorithm 1}} \text{ Squaring}&\hspace{2em}&\\ \hline {\tiny 1.:}&\mathbf {\text{procedure }} {\rm S{\small QUARE}}\left(y\right):\\ {\tiny 2.:}&\hspace{2em}\mathbf {\vec a} \leftarrow () &&\triangleright~\mathbf {\vec a}\text{ starts as empty polynomial sequence}\\ {\tiny 3.:}&\hspace{2em}i \leftarrow 0\\ {\tiny 4.:}&\hspace{2em}\mathbf{while}~y\ne0~\mathbf{do} &&\triangleright~\text{break }y\text{ down into a polynomial of base }2\\ {\tiny 5.:}&\hspace{4em}\mathbf{if}~y~\&~1~\mathbf{then} &&\triangleright~\text{if lsb of }y\text{ is set}\\ {\tiny 6.:}&\hspace{6em}\mathbf{\vec a} \leftarrow \mathbf{\vec a}i &&\triangleright~\text{append }i\text{ to }\mathbf{\vec a}~\text{(appending }x^i\text{)}\\ {\tiny 7.:}&\hspace{4em}\mathbf{end~if}\\ {\tiny 8.:}&\hspace{4em}i \leftarrow i + 1\\ {\tiny 9.:}&\hspace{4em}y \leftarrow y \gg 1 &&\triangleright~\text{shift }y\text{ right by one}\\ {\tiny 10.:}&\hspace{2em}\mathbf{end~while}\\ {\tiny 11.:}&\hspace{2em}P^2(x) \leftarrow F\left(\mathbf{\vec a}\right) &&\triangleright~\text{obtain the squared polynomial via } F\left(\mathbf{\vec a}\right)\\ {\tiny 12.:}&\hspace{2em}\mathbf{return}~P^2(2) &&\triangleright~\text{simply sum up the polynomial}\\ {\tiny 13.:}&\mathbf {\text{end procedure}}\\ \hline &\end{array}$

Python ( тест з кодовою панеллю ):

#/cs//q/11418/2755

def F(a):
    n = len(a)
    for i in range(n):
        assert a[i] >= 0

    # (r) => coefficient
    # coefficient \cdot x^{r}
    S = {}
    for ai in a:
        for aj in a:
            r = ai + aj

            if r not in S:
                S[r] = 0

            S[r] += 1

    return list(S.items())

def SQUARE(x):
    x = int(x)

    a = []
    i = 0
    while x != 0:
        if x & 1 == 1:
            a += [i]
        x >>= 1
        i += 1

    print 'a:',a
    P2 = F(a)

    print 'P^2:',P2

    s = 0
    for e,c in P2:
        s += (1 << e)*c
    return s

3. Таким чином, квадратик є не менш важким, як ця проблема.

4. Тому ціле квадратування рівнозначно цій задачі. (кожен з них максимум такий же важкий, як один з одним, через ( 2 , 3 , 1 ))

$\mathcal O(n\log n)$ $\mathcal O(n \log n \log \log n)$ $\mathcal O\left(n \log n\,2^{\mathcal O(\log^* n)}\right)$ $\Omega\left(n \log n\right)$

$\mathcal O\left(n\log n\right)$

5. Тепер ваша проблема не є саме множенням, це квадратом. Тож квадратування простіше? Що ж, це відкрита проблема (поки що немає) : невідомо, що квадрати мають швидший алгоритм, ніж множення. Якщо ви могли знайти кращий алгоритм для своєї проблеми, ніж використання множення; то це, ймовірно, буде проривом.

$\mathcal O(n\log n)$ $\mathcal O(n\log n)$ $\mathcal O(n \log n)$ $\mathcal O(n\log n)$ або як найкращий алгоритм множення лише наближається до цієї складності.

— Realz Slaw
джерело