算法面试题-牛客网数组变换

题目：

解析：

首先简单的情况是，两个数，判断能否通过题目中的规则，每次乘以2，最后能否相等，这个问题其实很简单，设两个数为a和b，则有：

a*2ⁿ=b*2^m

化简后得到：

a/b=2^m-n

求对数：

log₂(a/b)=m-n

当条件成立时，n和m均为正数，也就是m-n也为整数。

当数组中有多于两个数时，要判断整个数组是否符合，只需要判断相邻两个数是否符合规则即可，也就是只需要遍历一次，从两个数迁移到多个数的情况是一样的。

代码：

import math


def log2(num):
    return math.log(num, 2)


def isInt(n):
    return n - int(n) == 0


while True:
    try:
        n = int(raw_input())
        ns = map(float, raw_input().split(' '))
        for i in range(n-1):
            if not isInt(log2(ns[i] / ns[i+1])):
                print 'NO'
                break
        else:
            print 'YES'
    except EOFError:
        break