Арифметико-геометрична прогресія є поелементний твором арифметичної послідовності і геометричній послідовності. Наприклад, 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не відбуватиметься в якості вхідних даних.

Вихідні дані

Будь це арифметико-геометрична послідовність. Виведіть значення truthy / falesy або два різних послідовних значення.

Тестові справи

Правда:

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 термінів

Якщо є менше трьох термінів, завжди можна знайти послідовність відповідності. Таким чином, ми примушуємо дотримати значення.

Спеціальний випадок №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}$$

Мова Вольфрама (Mathematica) , 55 байт

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

Спробуйте в Інтернеті!

Solveповернути всі форми рішення. Результат порівнюється з тим, {}щоб перевірити, чи є рішення.