使用python内置库解决排列组合问题

我就是我 2022-11-03 10:53 70阅读 0赞

### 笛卡尔积：itertools.product(\*iterables\[, repeat\]) ###

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 </td> 
 <td> <code>import</code>&nbsp;<code>itertools</code> <code>for</code>&nbsp;<code>i&nbsp;</code><code>in</code>&nbsp;<code>itertools.product(</code><code>'BCDEF'</code><code>, repeat&nbsp;</code><code>=</code>&nbsp;<code>2</code><code>):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(''.join(i),end</code><code>=</code><code>","</code><code>)</code> <code>print</code><code>(</code><code>'\n'</code><code>)</code> &nbsp; <code># 输出 BB BC BD BE BF CB CC CD CE CF DB DC DD DE DF EB EC ED EE EF FB FC FD FE FF</code> </td> 
 </tr> 
 </tbody> 
</table>

两个元组进行笛卡尔积：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 </td> 
 <td> <code>import</code>&nbsp;<code>itertools</code> <code>a&nbsp;</code><code>=</code>&nbsp;<code>(</code><code>1</code><code>,&nbsp;</code><code>2</code><code>)</code> <code>b&nbsp;</code><code>=</code>&nbsp;<code>(</code><code>'A'</code><code>,&nbsp;</code><code>'B'</code><code>,&nbsp;</code><code>'C'</code><code>)</code> <code>c&nbsp;</code><code>=</code>&nbsp;<code>itertools.product(a,b)</code> <code>for</code>&nbsp;<code>i&nbsp;</code><code>in</code>&nbsp;<code>c:</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(i,end</code><code>=</code><code>","</code><code>)</code> &nbsp; <code># 输出(1, 'A') (1, 'B') (1, 'C') (2, 'A') (2, 'B') (2, 'C')</code> </td> 
 </tr> 
 </tbody> 
</table>

### 排列：itertools.permutations(iterable\[, r\]) ###

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 </td> 
 <td> <code>import</code>&nbsp;<code>itertools</code> <code>for</code>&nbsp;<code>i&nbsp;</code><code>in</code>&nbsp;<code>itertools.permutations(</code><code>'BCD'</code><code>,&nbsp;</code><code>2</code><code>):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(''.join(i),end</code><code>=</code><code>","</code><code>)</code> <code># 输出 BC BD CB CD DB DC</code> <code>print</code><code>(</code><code>'\n'</code><code>)</code> </td> 
 </tr> 
 </tbody> 
</table>

### 组合：itertools.combinations(iterable, r) ###

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 </td> 
 <td> <code>import</code>&nbsp;<code>itertools</code> <code>for</code>&nbsp;<code>i&nbsp;</code><code>in</code>&nbsp;<code>itertools.combinations(</code><code>'BCDEF'</code><code>,&nbsp;</code><code>2</code><code>):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(''.join(i),end</code><code>=</code><code>" "</code><code>)</code> <code># 输出 BC BD BE BF CD CE CF DE DF EF</code> <code>print</code><code>(</code><code>'\n'</code><code>)</code> </td> 
 </tr> 
 </tbody> 
</table>

### 组合（包含自身重复）：itertools.combinations\_with\_replacement(iterable, r) ###

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 </td> 
 <td> <code>import</code>&nbsp;<code>itertools</code> <code>for</code>&nbsp;<code>i&nbsp;</code><code>in</code>&nbsp;<code>itertools.combinations_with_replacement(</code><code>'ABC'</code><code>,&nbsp;</code><code>3</code><code>):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code>&nbsp;<code>('</code><code>'.join(i),end='</code>&nbsp;<code>')</code> &nbsp; <code># 输出 AAA AAB AAC ABB ABC ACC BBB BBC BCC CCC</code> <code>print</code><code>(</code><code>'\n'</code><code>)</code> </td> 
 </tr> 
 </tbody> 
</table>

### 举例 ###

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 </td> 
 <td> <code>'BCDEF五个字母组合问题'</code> <code>import</code>&nbsp;<code>itertools</code> &nbsp; <code>print</code><code>(</code><code>"1个组合："</code><code>)</code> <code>for</code>&nbsp;<code>i, val&nbsp;</code><code>in</code>&nbsp;<code>enumerate</code><code>(</code><code>list</code><code>(itertools.combinations(</code><code>'BCDEF'</code><code>,&nbsp;</code><code>1</code><code>))):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(</code><code>"序号：%s&nbsp;&nbsp; 值：%s"</code>&nbsp;<code>%</code>&nbsp;<code>(i&nbsp;</code><code>+</code>&nbsp;<code>1</code><code>, ''.join(val)))</code> <code>print</code><code>(</code><code>"2个组合："</code><code>)</code> <code>for</code>&nbsp;<code>i, val&nbsp;</code><code>in</code>&nbsp;<code>enumerate</code><code>(</code><code>list</code><code>(itertools.combinations(</code><code>'BCDEF'</code><code>,&nbsp;</code><code>2</code><code>))):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(</code><code>"序号：%s&nbsp;&nbsp; 值：%s"</code>&nbsp;<code>%</code>&nbsp;<code>(i&nbsp;</code><code>+</code>&nbsp;<code>1</code><code>, ''.join(val)))</code> <code>print</code><code>(</code><code>"3个组合："</code><code>)</code> <code>for</code>&nbsp;<code>i, val&nbsp;</code><code>in</code>&nbsp;<code>enumerate</code><code>(</code><code>list</code><code>(itertools.combinations(</code><code>'BCDEF'</code><code>,&nbsp;</code><code>3</code><code>))):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(</code><code>"序号：%s&nbsp;&nbsp; 值：%s"</code>&nbsp;<code>%</code>&nbsp;<code>(i&nbsp;</code><code>+</code>&nbsp;<code>1</code><code>, ''.join(val)))</code> <code>print</code><code>(</code><code>"4个组合："</code><code>)</code> <code>for</code>&nbsp;<code>i, val&nbsp;</code><code>in</code>&nbsp;<code>enumerate</code><code>(</code><code>list</code><code>(itertools.combinations(</code><code>'BCDEF'</code><code>,&nbsp;</code><code>4</code><code>))):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(</code><code>"序号：%s&nbsp;&nbsp; 值：%s"</code>&nbsp;<code>%</code>&nbsp;<code>(i&nbsp;</code><code>+</code>&nbsp;<code>1</code><code>, ''.join(val)))</code> <code>print</code><code>(</code><code>"5个组合："</code><code>)</code> <code>for</code>&nbsp;<code>i, val&nbsp;</code><code>in</code>&nbsp;<code>enumerate</code><code>(</code><code>list</code><code>(itertools.combinations(</code><code>'BCDEF'</code><code>,&nbsp;</code><code>5</code><code>))):</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>print</code><code>(</code><code>"序号：%s&nbsp;&nbsp; 值：%s"</code>&nbsp;<code>%</code>&nbsp;<code>(i&nbsp;</code><code>+</code>&nbsp;<code>1</code><code>, ''.join(val)))</code> </td> 
 </tr> 
 </tbody> 
</table>