【算法设计与分析习题】索引与序列值问题 | 序列查找 | 整数的均方根

1. 给定n个元素的有序序列A，对于A中的索引i，给定一个算法判断是否A[i]=i。

def seq_index(seq):
    for i in range(len(seq)):
        if i != seq[i]:
            return False
    return True

该算法时间复杂度O(n)。

2.给定一个元素只有0或1的序列，该序列中的1后面一定为0，设计算法中找出序列中0的个数。

def seq_find(seq):
    size, i, res = len(seq), 0, 0
    while i < size - 2:
        for j in range(i + 1, size):
            if seq[j] == 0:
                res += 1
                i += 1
            else:
                i = j
    return res

该算法的时间复杂度为O(n)。

3.给定一个整数x，设计算法找出x均方根值。如果没有均方根，则取最近的整数。如x=4，则输出2；x=11，则输出3。

def average_sqrt(x):
    for i in range(0, x + 1):
        pow_i = i * i
        pow_ip = (i + 1) * (i + 1)
        if pow_i < x < pow_ip:
            return i if x - pow_i < pow_ip - x else i + 1
        if pow_i == x:
            return i
        if pow_ip == x:
            return i + 1

该算法的时间复杂度为O(n)。

如果利用Python的特性，则只需一行

def single_line_average_sqrt(x):
    return list(filter(lambda n: n is not None, [(i if x - i*i < (i+1)*(i+1) -x else i + 1) if (i*i < x < (i+1)*(i+1)) else (i if i*i == x else (i + 1 if (i+1)*(i+1) == x else None)) for i in range(x + 1)]))[0]

其时间复杂度仍为O(n)。