算法 实验四 回溯算法的设计与实现

1．用回溯法求解n后问题。

n后问题描述：要求在一个n*n格的棋盘上放置n个皇后，使得她们彼此不受攻击。一个皇后可以攻击与之在同一行或同一斜线上的其他任何棋子。因此，n后问题等价于：任何两个皇后不能在同行、同列、同一斜线上。由于要求不同的皇后不能放在同一行，不失一般性，可设皇后i只放在第i行。

2．用回溯法求解求解子集合和问题。

       子集合问题描述：给定集合S，S中有n个正整数，M是一个正整数。子集合问题判定是否存在S的一个子集S1，使得S1中个元素之和等于M。例如：

 若S = {7, 5, 1, 2, 10}，M = 9，则存在一个S的子集合S1 = {7, 2}，使得S1中个元素之和等于9。

#include <iostream>
using namespace std;
#define MAXSIZE 20

int n,sum=0; //n 是皇后个数，sum 可行解的个数
int y[]={};//y 数组记当前解

int m,M;//s数组为集合元素，x数组存放解，m为集合元素个数，M为目标正整数
//int s[]={0,7,5,1,2,10};
int s[MAXSIZE]={0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};

bool Place(int k) //可行性约束函数
{
    for (int j=1; j<k; j++)
        if ( (abs(k-j)==abs(y[k]-y[j])) || (y[k]==y[j]) ) //不满足约束条件
            return false;
    return true;
}
void Backtrack(int t) //深度优先搜索解空间中第 t 层子树
{
    if (t > n) //已搜索到一叶结点，得到一个互不攻击的放置方案
    {
        sum++;
        for (int i=1;i<=n;i++)
            cout << y[i]<< " " ;
        cout << endl;
    }
    else //结点 x[t]是内部结点，他有 n 个儿子 1，2，…，n
        for (int i=1;i<=n;i++)
        { y[t]=i;
            if ( Place(t) ) //检查当前扩展结点的可行性，否则剪去相应的子树
                Backtrack(t+1) ; //以深度优先方式递归地对可行子树搜索
        }
}
//7 5 2 1 10
int count=0;  //累计解个数
void dispsolution(int x[])
{
    cout<<"第"<<++count<<"个解：";
    for(int i=1;i<=m;i++)
        if(x[i]==1)
            cout<<s[i]<<" ";
    cout<<endl;

}
void dfs(int ssum,int rsum,int x[],int i) //求解子集和
{
    //sum 考虑第i个整数时选取的整数和，rsum为剩下的整数和
    if(i>m)   //找到一个叶子结点
    {
        if(ssum==M)    //找到一个满足条件的解输出
            dispsolution(x);
    }
    else    //尚未找完所有整数
    {
        if(ssum+s[i]<=M)  //左孩子结点剪枝：选取满足条件的整数w[i]
        {
            x[i]=1;
            dfs(ssum+s[i],rsum-s[i],x,i+1);
        }
        if(ssum+rsum>M)  //右孩子结点剪枝：剪出不可能存在解的结点
        {
            x[i]=0;
            dfs(ssum,rsum-s[i],x,i+1);
        }
    }
}

void Nnum()
{
    cout<<"n皇后问题"<<endl<<"请输入皇后个数：";
    cin>>n;
    Backtrack(1);
}
void Size()
{

    cout<<"求解子集合和问题"<<endl;
    cout<<"请输入集合元素个数:"<<endl;
    cin>>m;

    cout<<"请输入集合S："<<endl;
    for(int i=1;i<=m;i++)
      cin>>s[i];
    cout<<"请输入正整数M"<<endl;
    cin>>M;

    int x[20];
    int rsum=0;
    for(int j=1;j<=m;j++)   //求所有整数和
        rsum+=s[j];
    dfs(0,rsum,x,1);   //i从1开始
    if(count==0)
        cout<<"无解"<<endl;
}

int main() {
    Nnum();
    Size();
}