483. Smallest Good Base

1. 问题
   已知：$a\in\mathbb N, 3 \le a \le 10 ^ {18}$求：
   

   $$\min x \in\mathbb N,x \ge 2, s.t. \exists n\in\mathbb N, \sum_{i=0}^n x^i=a$$

2. 数学推论

   1. $n \ge 1, a=\sum_{i=0}^n x^i= \frac{x^{n+1}-1}{x-1}$
   2. $\forall x,y,m,n \in \mathbb N^+, x \le y, m \le n: \sum_{i=0}^m x^i \le \sum_{i=0}^m y^i \le \sum_{i=0}^n y^i$
   3. 由推论2可得：对于任意的$n$，只有一个可行解$x$
   4. 由推论2可得：可行解$x$随着$n$的增大而减小，即：
      $\forall x,y,m,n \in \mathbb N^+, m \lt n:\sum_{i=0}^m x^i =\sum_{i=0}^n y^i \Rightarrow x > y$
   5. $n=1时, x=a-1$且为最大可行解
   6. 若$x=2$为可行解，则$a$的二进制表达式为全1，且$x=2$就是最小可行解
   7. $n>1, x>2$时, $a=\sum_{i=0}^n x^i > \sum_{i=0}^n 2^i =2^{n+1} -1 \Rightarrow a \ge 2^{n+1}$
      $\Rightarrow n \le \log_2 a - 1$
   8. $\forall n \in \mathbb N, n \gt 1, \forall x \in\mathbb N,x \ge 2: \sum_{i=0}^n x^i \lt (1+x)^n$
      证明：
      $n=2$时，$\sum_{i=0}^2 x^i=1+x+x^2<1+2x+x^2=(1+x)^2$
      假设$n$成立，则$n+1$时，
      $\sum_{i=0}^{n+1} x^i = \sum_{i=0}^n x^i + x^{n+1} \lt (1+x)^n + x^{n+1} \lt (1+x)^n + x(1+x)^n=(1+x)^{n+1}$
   9. 由推论8得：$n > 1$时，可行解$x$必满足：$x^n<\sum_{i=0}^n x^i=a< (1+x)^n$，即得 $x<\sqrt [n]{a} <x+1 \Rightarrow x=\lfloor \sqrt [n]{a} \rfloor$
   10. 由推论1, 9可得：$n > 1$时，可行解$x$存在当且仅当：$x=\lfloor \sqrt [n]{a} \rfloor , \frac{x^{n+1}-1}{x-1} = a$

3. 源码

import math

def binary_base_all_one(a):
    while (a & 1):
        a = (a >> 1)

    return a == 0

class Solution(object):
    def smallestGoodBase(self, a):
        """
        :type n: str
        :rtype: str
        """
        a = int(a)

        if binary_base_all_one(a):
            return '2'

        max_n= int(math.log(a, 2))

        for n in range(max_n, 2, -1):
            x = int(a**((n - 1)**-1))
            if (x**n - 1) // (x - 1) == a:
                return str(x)

        return str(a - 1)