Арифметико-геометрическая прогрессия является поэлементным произведением арифметической последовательности и геометрической последовательности. Например, 1 -4 12 -32является произведением арифметической последовательности 1 2 3 4и геометрической последовательности 1 -2 4 -8. N-й член целочисленной арифметико-геометрической последовательности может быть выражен как

$$a_n = r^n \cdot (a_0 + nd)$$

для некоторого вещественного числа $d$ , отлична от нуля реального $r$ , а целое число $a_0$ . Обратите внимание, что $r$ и $d$ не обязательно являются целыми числами.

Например, последовательность 2 11 36 100 256 624 1472 3392имеет $a_0 = 2$ , $r = 2$ , и $d = 3.5$ .

вход

Упорядоченный список из $n \ge 2$ целых чисел в качестве ввода в любом разумном формате. Поскольку некоторые определения геометрической последовательности допускают $r=0$ и определяют $0^0 = 1$ , то, является ли вход арифметико-геометрической последовательностью, не будет зависеть от того, разрешено ли значение $r$ равным 0. Например, 123 0 0 0 0не будет входить в качестве ввода.

Выход

Будь то арифметико-геометрическая последовательность. Выведите истинное / ложное значение или два разных последовательных значения.

Контрольные примеры

Правда:

1 -4 12 -32
0 0 0
-192 0 432 -1296 2916 -5832 10935 -19683
2 11 36 100 256 624 1472 3392
-4374 729 972 567 270 117 48 19
24601 1337 42
0 -2718
-1 -1 0 4 16
2 4 8 16 32 64
2 3 4 5 6 7
0 2 8 24

Ложь:

4 8 15 16 23 42
3 1 4 1
24601 42 1337
0 0 0 1
0 0 1 0 0
1 -1 0 4 16

Perl 6 , 184 128 135 байтов

{3>$_||->\x,\y,\z{?grep ->\r{min (x,{r&&r*$_+(y/r -x)*($×=r)}...*)Z==$_},x??map (y+*×sqrt(y²-x*z).narrow)/x,1,-1!!y&&z/y/2}(|.[^3])}

Попробуйте онлайн!

$r$$d$

$a_n=r \cdot a_{n-1}+r^n \cdot d$

Некоторые улучшения вдохновлены ответом Арнаулда на JavaScript.

объяснение

3>$_||  # Return true if there are less than three elements

->\x,\y,\z{ ... }(|.[^3])}  # Bind x,y,z to first three elements

# Candidates for r
x  # If x != 0
??map (y+*×sqrt(y²-x*z).narrow)/x,1,-1  # then solutions of quadratic equation
!!y&&z/y/2  # else solution of linear equation or 0 if y==0

?grep ->\r{ ... },  # Is there an r for which the following is true?

    ( ,                         ...*)  # Create infinite sequence
     x  # Start with x
       {                       }  # Compute next term
        r&&  # 0 if r==0
                (y/r -x)  # d
           r*$_  # r*a(n-1)
                          ($×=r)  # r^n
                +        *  # r*a(n-1)+d*r^n
                                     Z==$_  # Compare with each element of input
min  # All elements are equal?

JavaScript (ES7), 135 127 байт

a=>!([x,y,z]=a,1/z)|!a.some(n=>n)|[y/x+(d=(y*y-x*z)**.5/x),y/x-d,z/y/2].some(r=>a.every((v,n)=>(v-(x+n*y/r-n*x)*r**n)**2<1e-9))

Попробуйте онлайн!

Как?

$r$$d$$<10^{-9}$

Особый случай № 1: менее 3-х сроков

Если есть менее 3 терминов, всегда можно найти подходящую последовательность. Таким образом, мы добиваемся истинного значения.

Особый случай № 2: только нули

$0$$a_0=0$$d=0$$r\neq 0$

$a_0=0$

$a_0=0$

$$a_n=r^n\times n\times d$$

Который дает:

$$a_1=r\times d\\ a_2=2r^2\times d$$

$d$$0$$a_1\neq 0$

$$r=\frac{a_2}{2a_1}$$

$a_0\neq 0$

$a_{n+1}$$a_n$

$$a_{n+1}=r.a_n+r^{n+1}d$$

$a_{n+2}$

$$\begin{align}a_{n+2}&=r.a_{n+1}+r^{n+2}d\\ &=r(r.a_n+r^{n+1}d)+r^{n+2}d\\ &=r^2a_n+2r.r^{n+1}d\\ &=r^2a_n+2r(a_{n+1}-r.a_n)\\ &=-r^2a_n+2r.a_{n+1} \end{align}$$

В частности, у нас есть:

$$a_2=-r^2a_0+2r.a_1$$

Ведущий в следующий квадратик:

$$r^2a_0-2r.a_1+a_2=0$$

Чьи корни:

$$r_0=\frac{a_1+\sqrt{{a_1}^2-a_0a_2}}{a_0}\\ r_1=\frac{a_1-\sqrt{{a_1}^2-a_0a_2}}{a_0}$$

Wolfram Language (Mathematica) , 55 байтов

{}!=Solve[i=Range@Tr[1^#];(a+i*d)r^i==#,{r,a,d},Reals]&

Попробуйте онлайн!

Solveвернуть все формы решения. Результат сравнивается с тем, {}чтобы проверить, есть ли какое-либо решение.