python 生成排列、组合以及选择

from <python cookbook> 19.15

　　任务

　　需要对一个序列的排列（permutation）、组合（combination）或选择（selection）进行迭代操作。即使初始的序列长度并不长，组合计算的规则却显示生成的序列可能非常庞大，比如一个长度为13的序列有超过60亿种可能的排列。所以，你肯定不希望在开始迭代前计算并生成序列中的所有项

　　解决方案

　　生成器允许你在迭代的时候一次一个的计算需要的对象。如果有很多这种对象，而且你也必须逐个的检查他们，那么程序无可避免的会用很长时间才能完成循环。但至少你不用浪费很多内存来保存所有项：

def _combinators(_handle, items, n):
    ''' 抽取下列组合的通用结构'''
    if n == 0:
        yield [ ]
    for i, item in enumerate(items):
        this_one = [item]
        for cc in _combinators(_handle, _handle(items, i), n-1):
            yield this_one + cc

def combinations(items, n):
     ''' 取得n个不同的项， 顺序是有意义的'''
    def skipIthItem(items, i):
        return items[:i] + items[i + 1:]
    return _combinators(skipIthItem, items, n)

def uniqueCombinations(items, n):
    '''取得n个不同的项，顺序无关'''
    def afterIthItem(items, i):
        return items[i+1:]
    return _combinators(afterIthItem, items, n)

def selections(items, n):
    '''取得n项（不一定要不同），顺序是有意义的'''
    def keepAllItems(items, i):
        return items
    return _combinators(keepAllItems, items, n)

def permutations(items):
    ''' 取得所有项， 顺序是有意义的'''
    return combinations(items, len(items))

if __name__ == '__main__':
    print "Permutations of 'bar'"
    print map(''.join, permutations('bar'))
#输出： ['bar', 'bra', 'abr', 'arb', 'rba', 'rab']

    print "Combinations of 2 letters from 'bar'"
    print map(''.join, combinations('bar', 2))
# 输出： ['ba', 'br', 'ab', 'ar', 'rb', 'ra']

    print "Unique Combinations of 2 letters from 'bar'"
    print map(''.join, uniqueCombinations('bar', 2))
# 输出: ['ba', 'br', 'ar']

    print "Selections of 2 letters from 'bar'"
    print [''.join, selections('bar', 2)]
# 输出： ['bb', 'ba', 'br', 'ab', 'aa', 'ar', 'rb', 'ra', 'rr']