1. 程式人生 > >排列組合演算法總結(基於C++實現)

排列組合演算法總結(基於C++實現)

1.排列

全排列n!

1.1 遞迴法

設一組數p = {r1, r2, r3, … ,rn}, 全排列為perm(p),pn = p – {rn}。則perm(p) = r1perm(p1), r2perm(p2), r3perm(p3), … , rnperm(pn)。當n = 1時perm(p} = r1。

如:求{1, 2, 3, 4, 5}的全排列

1、首先看最後兩個數4, 5。 它們的全排列為4 5和5 4, 即以4開頭的5的全排列和以5開頭的4的全排列。

由於一個數的全排列就是其本身,從而得到以上結果。

2、再看後三個數3, 4, 5。它們的全排列為3 4 5、3 5 4、 4 3 5、 4 5 3、 5 3 4、 5 4 3 六組數。

即以3開頭的和4,5的全排列的組合、以4開頭的和3,5的全排列的組合和以5開頭的和3,4的全排列的組合.

#include <iostream>
using namespace std;

void Perm(int start, int end, int a[]) {
    //得到全排列的一種情況,輸出結果
    if (start == end) {
        for (int i = 0; i < end; i++)
            cout << a[i] << ' ';
        cout << endl;
        return
; } for (int i = start; i < end; i++) { swap(a[start], a[i]); //交換 Perm(start + 1, end, a); //分解為子問題a[start+1,...,end-1]的全排列 swap(a[i], a[start]); //回溯 } } int main() { int i, n, a[10]; while (cin >> n, n) { for (i = 0; i < n; i++) { a[i] = i + 1
; } Perm(0, n, a); } return 0; }

2.組合

C(n,k),n個數中任取k個數

2.1 遞迴法

實際上就是在n個數中,標記k個數,然後輸出這k個數的過程。使用一個visited陣列來記錄相應下標的數是否被選中。

#include <iostream>
using namespace std;

void dfs(int pos, int cnt, int n, int k, int a[],bool visited[]) {
    //已標記了k個數,輸出結果
    if (cnt == k) {
        for (int i = 0; i < n; i++)
            if (visited[i]) cout << a[i] << ' ';
        cout << endl;
        return;
    }

    //處理到最後一個數,直接返回
    if (pos == n) return;

    //如果a[pos]沒有被選中
    if (!visited[pos]) {
        //選中a[pos]
        visited[pos] = true;
        //處理在子串a[pos+1, n-1]中取出k-1個數的子問題
        dfs(pos + 1, cnt + 1, n, k, a,visited);
        //回溯
        visited[pos] = false;   
    }
    //處理在子串a[pos+1, n-1]中取出k個數的問題
    dfs(pos + 1, cnt, n, k, a, visited);
}
int main() {
    int i, n, k;
    while (cin >> n >> k, n || k) 
    {
        int *a = new int[n];
        bool *visited = new bool[n];
        for (i = 0; i < n; i++)
        {
            a[i] = i + 1;
            visited[i] = false;
        }
        dfs(0, 0, n, k, a, visited);
        delete[] a;
        delete[] visited;
    }
    getchar();
    return 0;
}

2.2 ‘01’轉換法

本程式的思路是開一個數組,其下標表示1到n個數,陣列元素的值為1表示其代表的數被選中,為0則沒選中。

首先初始化,將陣列前n個元素置1,表示第一個組合為前n個數。

然後從左到右掃描陣列元素值的“10”組合,找到第一個“10”組合後將其變為“01”組合,同時將其左邊的所有“1”全部移動到陣列的最左端。

當第一個“1”移動到陣列的n-m的位置,即n個“1”全部移動到最右端時,就得到了最後一個組合。

例如求5中選3的組合:

1 1 1 0 0 //1,2,3

1 1 0 1 0 //1,2,4

1 0 1 1 0 //1,3,4

0 1 1 1 0 //2,3,4

1 1 0 0 1 //1,2,5

1 0 1 0 1 //1,3,5

0 1 1 0 1 //2,3,5

1 0 0 1 1 //1,4,5

0 1 0 1 1 //2,4,5

0 0 1 1 1 //3,4,5

#include <iostream>

using namespace std;

//輸出結果
void printRes(int* a, bool* index, int n)
{
    for (int i=0;i<n;i++)
    {
        if (index[i])
        {
            cout << a[i] << " ";
        }
    }
    cout << endl;
}

//檢查最後k個位置是否已全變成0
bool hasDone(bool* index, int n, int k)
{
    for (int i=n-1;i>=n-k;i--)
    {
        if (!index[i])
        {
            return false;
        }
    }
    return true;
}

void Comb(int* a, int n, int k)
{
    bool *index = new bool[n]();
    //選中前k個位置
    for (int i = 0; i < k; i++)
    {
        index[i] = true;
    }
    printRes(a, index, n);

    while (!hasDone(index, n, k))
    {
        //從左到右掃描陣列
        for (int i = 0; i < n - 1; i++)
        {
            //找到第一個“10”組合將其變成"01"組合
            if (index[i] && !index[i + 1])
            {
                index[i] = false;
                index[i + 1] = true;

                //將"01"組合左邊的1移到最左邊
                int count = 0;
                for (int j = 0; j < i; j++)
                {
                    if (index[j])
                    {
                        index[j] = false;
                        index[count++] = true;
                    }
                }
                printRes(a, index, n);
                break;
            }
        }
    }
    delete[] index;
}
int main()
{
    int n,k;
    while (cin>>n>>k)
    {
        int *a = new int[n]();
        for (int i = 0; i < n; i++)
        {
            a[i] = i+1;
        }
        Comb(a, n, k);
        delete[] a;
    }

    return 0;
}