回溯算法之选书问题

选书问题

学校放寒假时，信息学竞赛辅导老师有A，B，C，D，E五本书，要分给参加培训的张、王、刘、孙、李五位同学，每人只能选一本书。老师事先让每个人将自己喜欢的书填写在如下的表格中。然后根据他们填写的表来分配书本，希望设计一个程序帮助老师求出所有可能的分配方案，使每个学生都满意。

输入

无输入。

输出

输出所有选书方案，每种方案六行；

每一行输出方案数，格式：answer XX:（XX为方案数）

接下来五行输出五位学生所选的书，五位学生姓名用"Student Zhang, Student Wang, Student Liu, Student Sun, Student Li"表示，参考样例输出。

样例输出

answer 1:
Student Zhang:C
Student Wang:A
Student Liu:B
Student Sun:D
Student Li:E

代码如下：

思路分析：

利用经典的回溯算法模板解决问题。

#include<iostream>
using namespace std;

class SelectBook{
public:
    static int num;		//记录有几种方案
    int depth=0;
    //初始化每人喜欢的书籍
    const bool PreferBook[5][5]={{0,0,1,1,0},
                           {1,1,0,0,1},
                           {0,1,1,0,0},
                           {0,0,0,1,0},
                           {0,1,0,0,1}
                           };
    string man[5]={"Zhang","Wang","Liu","Sun","Li"};
    bool used[5]={false,false,false,false,false};	//是否遍历过某结点
    char Solution[5]={'0','0','0','0','0'};			//定义一个数组记录最终的答案
public:
    void addTrack(char *solution,int depth,int y){
        char temp;
        switch (y) {
            case 0: temp='A';break;
            case 1: temp='B';break;
            case 2: temp='C';break;
            case 3: temp='D';break;
            case 4: temp='E';break;
        }
       solution[depth]=temp;
    }

    void OutPut(char *Solution){
        cout<<"answer "<<++num<<":\n";
        for(int i=0;i<5;i++){
            cout<<"Student "<<man[i]<<":"<<Solution[i]<<endl;
        }
    }

    void BackTracking(int depth,char *solution,bool *used){
        if(depth==5){
            OutPut(solution);
            return ;
        }

        for(int i=0;i<5;i++){
            if(used[i]==false&&PreferBook[depth][i]==true){
                //达到进行一步的条件
                used[i]= true;
                addTrack(solution,depth,i);
                BackTracking(depth+1,solution,used);	//递归进入下一步
                //回溯上一步
                used[i]=false;
                solution[depth]='0';
            }
        }
    }
};

int SelectBook::num=0;

int main(){
    SelectBook a;
    a.BackTracking(a.depth,a.Solution,a.used);
    return 0;
}