Python程序员面试算法宝典---解题总结: 第7章 排列组合与概率 7.7 如何判断还有几盏灯亮着

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

'''
Python程序员面试算法宝典---解题总结: 第7章 排列组合与概率 7.7 如何判断还有几盏灯亮着

题目:
100个灯泡排成一排，第一轮将所有灯泡打开；第二轮每隔一个灯泡关掉一个，即排在偶数的
灯泡被关掉，第三轮每隔两个灯泡，将开着的灯泡关掉，关掉的打开。
依次类推，第100轮结束的时候，还有几盏灯亮着?

分析:
被操作奇数次的灯打开，偶数次的关闭，问题转化为求每个灯被操作的次数。


关键:
1 书上解法
每个灯被操作的次数依赖于这个灯的编号有几个约数，约数个数为奇数的
灯最后亮着，
其中约数个数为奇数的有:
1, 4, 9,16,...,81, 100

2 没有想到
灯被操作的次数取决于这个灯的编号的约数的个数，
只有约数个数为奇数的才会亮着。

参考:
Python程序员面试算法宝典
'''
def isFactorOdd(num):
    count = 0
    for i in range(1, num + 1):
        if num % i == 0:
            count += 1
    if count & 1 == 1:
        return True
    else:
        return False

def getLightLampNum(n):
    results = []
    for i in range(1, n + 1):
        if isFactorOdd(i):
            results.append(i)
    return results


def process():
    results = getLightLampNum(100)
    print results


if __name__ == "__main__":
    process()