Python程序员面试算法宝典---解题总结: 第6章 基本数字运算 6.1 如何判断一个自然数是否是某个数的平方

# -*- coding: utf-8 -*-

'''
Python程序员面试算法宝典---解题总结: 第6章 基本数字运算 6.1 如何判断一个自然数是否是某个数的平方

题目:
设计一个算法，判断给定的一个数n是否是某个数的平方，不能使用开方运算。
例如16就满足条件，因为它是4的平方，而15则不满足条件，因为不存在
一个数使得其平方的值为15.

分析:
方法1:
二分查找，从0~n/2开始查找

方法2:
事先建立所有数到其开根号的字典

用二分查找实现，时间复杂度为O(logN)

关键:
1 二分查找要注意一种特殊情况
        # 为了防止无限循环，low=mid=high-1,直接对这种情况，跳出循环
        if low == high - 1:
            break

参考:
Python程序员面试算法宝典
'''

def getSquareRoot(n):
    if n < 0:
        return -1
    high = n / 2 + 1
    low = 0
    while low < high:
        mid = (low + high) / 2
        value = mid ** 2
        if value == n:
            return mid
        elif value < n:
            low = mid
        else:
            high = mid
        # 为了防止无限循环，low=mid=high-1,直接对这种情况，跳出循环
        if low == high - 1:
            break
    return -1


def process():
    n = 15
    result = getSquareRoot(n)
    if result == -1:
        print "{n} has no square root".format(n=n)
    else:
        print "The square root of {n} is {result}".format(
            n=n,
            result=result
        )
    n = 16
    result = getSquareRoot(n)
    if result == -1:
        print "{n} has no square root".format(n=n)
    else:
        print "The square root of {n} is {result}".format(
            n=n,
            result=result
        )


if __name__ == "__main__":
    process()