1. 程式人生 > >回溯法——八皇后問題

回溯法——八皇后問題

回溯法的基本做法是搜尋,或是一種組織得井井有條的,能避免不必要搜尋的窮舉式搜尋法。這種方法適用於解一些組合數相當大的問題。

回溯法在問題的解空間樹中,按深度優先策略,從根結點出發搜尋解空間樹。演算法搜尋至解空間樹的任意一點時,先判斷該結點是否包含問題的解。如果肯定不包含,則跳過對該結點為根的子樹的搜尋,逐層向其祖先結點回溯;否則,進入該子樹,繼續按深度優先策略搜尋。

回溯法指導思想——走不通,就掉頭。設計過程:確定問題的解空間;確定結點的擴充套件規則;搜尋。

n皇后問題

要在n*n的國際象棋棋盤中放n個皇后,使任意兩個皇后都不能互相吃掉。規則:皇后能吃掉同一行、同一列、同一對角線的任意棋子。求所有的解。n=8是就是著名的八皇后問題了。

設八個皇后為xi,分別在第i行(i=1,2,3,4……,8);

問題的解狀態:可以用(1,x1),(2,x2),……,(8,x8)表示8個皇后的位置;

由於行號固定,可簡單記為:(x1,x2,x3,x4,x5,x6,x7,x8);

問題的解空間:(x1,x2,x3,x4,x5,x6,x7,x8),1≤xi≤8(i=1,2,3,4……,8),共88個狀態;

約束條件:八個(1,x1),(2,x2) ,(3,x3),(4,x4) ,(5,x5), (6,x6) , (7,x7), (8,x8)不在同一行、同一列和同一對角線上。

盲目的列舉演算法:通過8重迴圈模擬搜尋空間中的88個狀態,從中找出滿足約束條件的“答案狀態”。程式如下:

 

/*
 *說明:八皇后——盲目迭代法
 *日期:2013-12-18
 */
#include <iostream>
using namespace std;
bool check_1(int a[],int n)
{
for(int i=2;i<=n;i++)
{
    for(int j=1;j<=i-1;j++)
    {
        if ((a[i]==a[j])||(abs(a[i]-a[j])==i-j))
        {
            return false;
        }
    }
}
return true;//不衝突
}

void queens_1()
{
    int a[9];
    int count = 0;
    for(a[1]=1;a[1]<=8;a[1]++)
    {
        for(a[2]=1;a[2]<=8;a[2]++)
        {
            for(a[3]=1;a[3]<=8;a[3]++)
            {
                for(a[4]=1;a[4]<=8;a[4]++)
                {
                    for(a[5]=1;a[5]<=8;a[5]++)
                    {
                        for(a[6]=1;a[6]<=8;a[6]++)
                        {
                            for(a[7]=1;a[7]<=8;a[7]++)
                            {
                                for(a[8]=1;a[8]<=8;a[8]++)
                                {
                                    if(!check_1(a,8))   
                                        continue;
                                    else
                                    {
                                        for(int i=1;i<=8;i++)  
                                        {
                                            cout<<a[i];
                                        }
                                        cout<<endl;
                                        count++;
                                    }
                                }
                            }
                        }
                    }
                }
            }

        }
    }
    cout<<count<<endl;
}

void main()
{
    queens_1();
}

 

程式思想比較簡單,最後可知共92種擺放方法。如果能夠排除那些沒有前途的狀態,會節約時間——回溯法(走不通,就回頭)。

 

bool check_2 (int a[ ],int n)
{//多次被呼叫,只需一重迴圈 
    for(int i=1;i<=n-1;i++)
    {
        if((abs(a[i]-a[n])==n-i)||(a[i]==a[n]))
            return false;
    }      
    return true;
}

void queens_2()
{
    int a[9];
    int count = 0;
    for(a[1]=1;a[1]<=8;a[1]++)
    {
        for(a[2]=1;a[2]<=8;a[2]++)
        {
            if (!check_2(a,2))  continue;
            for(a[3]=1;a[3]<=8;a[3]++)
            {
                if (!check_2(a,3))  continue;
                for(a[4]=1;a[4]<=8;a[4]++)
                {
                    if (!check_2(a,4))  continue;
                    for(a[5]=1;a[5]<=8;a[5]++)
                    {
                        if (!check_2(a,5))  continue;
                        for(a[6]=1;a[6]<=8;a[6]++)
                        {
                            if (!check_2(a,6))  continue;
                            for(a[7]=1;a[7]<=8;a[7]++)
                            {
                                if (!check_2(a,7))  continue;
                                for(a[8]=1;a[8]<=8;a[8]++)
                                {
                                    if (!check_2(a,8))  
                                        continue;
                                    else
                                    {
                                        for(int i=1;i<=8;i++)  
                                        {
                                            cout<<a[i];
                                        }
                                        cout<<endl;
                                        count++;
                                    }
                                }
                            }
                        }
                    }
                }
            }

        }
    }
    cout<<count<<endl;
}

void main()
{
    queens_2();
}

 

n此演算法可讀性很好,體現了“回溯”。但它只針對八皇后問題,解決任意的n皇后問題還要修改程式結構。如果要解決n皇后的問題,就需要將n作為引數傳遞給函式,函式需要重寫來實現回溯(不能採用級聯的for迴圈,n不確定);從另一方面,程式中出現了大量的for迴圈,而且for中的函式結構很相似,自然想到的是遞迴迭代回溯。這就是回溯比較常用的兩種實現方法:非遞歸回溯和遞歸回溯。

非遞歸回溯的程式實現:

 

void backdate (int n)
{  
    int count = 0;
    int a[100];

    int k = 1;
    a[1]=0;   
    while(k>0)
    {
        a[k]=a[k]+1;//對應for迴圈的1~n
        while((a[k]<=n)&&(!check_2(a,k)))//搜尋第k個皇后位置
        {
            a[k]=a[k]+1;
        }
          
        if(a[k]<=n)//找到了合理的位置
        {
            if(k==n )
            {//找到一組解
                for(int i=1;i<=8;i++)  
                {
                    cout<<a[i];
                }
                cout<<endl;
                count++;
            } 
            else 
            {
                k=k+1;//繼續為第k+1個皇后找到位置,對應下一級for迴圈 
                a[k]=0;//下一個皇后一定要從頭開始搜尋
            }
        }
        else
        {
            k=k-1;//回溯,對應執行外內層for迴圈回到更上層 
        }
    }
    cout<<count<<endl;
}

void main()
{
    backdate(8);
}

 

這樣也可以得到,8皇后問題的92中結果。更簡單、可讀的方法是採用遞迴的方式,如下:

 

int a[100], n, count;
void backtrack(int k)
{
    if (k>n)//找到解
    {
        for(int i=1;i<=8;i++)  
        {
            cout<<a[i];
        }
        cout<<endl;
        count++;
    }
    else
    {
        for (int i = 1;i <=n; i++)
        {
            a[k] = i;
            if (check_2(a,k) == 1)
            {backtrack(k+1);}
        }
    }
    
}

void main()
{
    n=8,count=0;
    backtrack(1);
    cout<<count<<endl;
}

 

眨眼可見,遞迴呼叫大大減少了程式碼量,也增加了程式的可讀性。給出其中的一個解,如下:

image