前言python
在程序設計的過程當中,全排列是比較常常遇到的一類問題,有時候本身寫仍是有點麻煩,也比較浪費時間。在這裏我介紹一種python中的全排列函數——itertools.permutations。更重要的是itertools是一個標準庫,不須要額外安裝只要import便可,要知道正式比賽中是不容許使用第三方庫的。函數
正文spa
咱們先看下函數的語法格式和參數:設計
itertools.permutations(iterable[, r])
r 指定生成排列的元素的長度,若是不指定,則默認爲迭代對象的元素長度。code
示例:對象
>>> list(itertools.permutations([1,2,3],2)) [(1, 2), (1, 3), (2, 1), (2, 3), (3, 1), (3, 2)] >>> list(itertools.permutations([1,2,3],3)) [(1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)]
下面是我在洛谷上用該方法解決的一道題:blog
題目描述it
將1,2,⋯ ,9共9個數分紅3組,分別組成3個三位數,且使這3個三位數構成1:2:3的比例,試求出全部知足條件的3個三位數。io
輸入格式程序設計
木有輸入
輸出格式
若干行,每行3個數字。按照每行第1個數字升序排列。
import itertools nums = [1,2,3,4,5,6,7,8,9] for num in itertools.permutations(nums, 9): a = 100*num[0] + 10*num[1] + num[2] b = 100*num[3] + 10*num[4] + num[5] c = 100*num[6] + 10*num[7] + num[8] if b == 2*a and c == 3*a: print(a, b, c)
該段函數輸出以下:
192 384 576
219 438 657
273 546 819
327 654 981
能夠看到這些數字都是有序的,正好符合題目的要求。可是要注意的是在排列前迭代對象必須有序,不然不會有序輸出。
下面介紹python中的組合函數——combinations(),一樣也是itertools庫中的函數,用法也和permutations()同樣,這裏就再也不贅述了,只要搞懂了排列函數,組合函數也就懂了。