【算法实验】回溯/分支限界法

• 话不多说，直接上代码，思路读一读代码就明白了，很简单。
• 需要注意的是，直接执行下段代码的话，只能测试本地的样例文本，所以在上传OJ前需要修改main函数和构造函数。

//实验四：回溯/分支限界法 0-1互斥矩阵
//
//设计一个算法，找出一对互斥集合 A 和 B，使得 A和 B 包含的列的总数最大。
//输入格式：
//一个 1000 行 20 列的矩阵 matrix 作为测试用例，每行输入 20 列的元素，对矩阵找出满足题意的互斥集合。
//输出格式：
//输出两行，使得 A 的元素 + B 的元素个数最大。
//第一行输出 A 集合中的所有元素（下标从 0 开始），以空格分开
//第二行输出 B 集合中的所有元素，格式同上
//如果，没有找到非空集合 A 和 B，则输出两行空行

#include<iostream>
#include<fstream>
using namespace std;
class matrix {
    int m[1000][20];

    int judgeMap[20][20] = { 0 };
    int temp[20] = { 0 };

    int bestAns[20] = { 0 };//存放20列各自状态
    int bestSum = 0;

    int sumA = 0;
    int sumB = 0;
    int diff = 0;
public:
    matrix(ifstream &afile) {
        //本地文件读取
        for (int i = 0; i < 1000; i++) {
            for (int j = 0; j < 20; j++) {
                afile >> m[i][j];
            }
        }
    }
    //matrix() {
    //    //上传OJ时读取
    //    for (int i = 0; i < 1000; i++) {
    //        for (int j = 0; j < 20; j++) {
    //            cin >> m[i][j];
    //        }
    //    }
    //}
    void getAns() {
        for (int i = 0; i < 20; i++) {
            if (bestAns[i] == 1) cout << i << ' ';
        }
        cout << endl;
        for (int i = 0; i < 20; i++) {
            if (bestAns[i] == 2) cout << i << ' ';
        }
        cout << endl;
    }
    void buildJudgeMap() {
        for (int i = 0; i < 20; i++) {
            for (int j = i + 1; j < 20; j++) {
                for (int k = 0; k < 1000; k++) {
                    if (m[k][i] == 1 && m[k][j] == 1) {
                        judgeMap[i][j] = 1;
                        judgeMap[j][i] = 1;
                        break;
                    }
                }
            }
        }
    }
    int ifMutex(int j, int tag) {
        //tag==1，判断j列放入A是否可行
        //tag==2，判断j列放入B是否可行
        if (tag == 1) {
            for (int i = 0; i < 20; i++) {
                if (temp[i] == 2) {
                    if (judgeMap[i][j] == 1) return 0;
                }
            }
            return 1;
        }
        else {
            for (int i = 0; i < 20; i++) {
                if (temp[i] == 1) {
                    if (judgeMap[i][j] == 1) return 0;
                }
            }
            return 1;
        }
    }
    int ifOkayAndBest() {
        if (sumA != 0 && sumB != 0 && sumA > sumB && sumA + sumB >= bestSum) {
            if (sumA + sumB == bestSum) { 
                //考虑A、B的差最小：
                if (abs(sumA - sumB) < diff) return 1;
                else return 0;
            }
            return 1;
        }
        return 0;
    }
    void copyAns() {
        for (int j = 0; j < 20; j++) {
            bestAns[j] = temp[j];
        }
        bestSum = sumA + sumB;
        diff = sumA - sumB;
    }
    void solutionInDFS(int j) {
        //对j+1到19列，递归调用
        //cout <<"now j is "<< j << endl;
        if (j == 20) {
            if (ifOkayAndBest()) {
                copyAns();
            }
            return;
        }
        else {
            if (ifMutex(j, 1)) {
                sumA++;
                temp[j] = 1;
                solutionInDFS(j + 1);
                sumA--;
                temp[j] = 0;
            }
            if (ifMutex(j, 2)) {
                sumB++;
                temp[j] = 2;
                solutionInDFS(j + 1);
                sumB--;
                temp[j] = 0;
            }
            if (sumA + sumB + 19 - j >= bestSum) {
            	//剪枝
                //若j=0，其潜力>bestSum时，递归j+1
                solutionInDFS(j + 1);
            }
            else return;
        }
    }
};
int main() {

    //本地文件读取
    ifstream afile;
    afile.open("D:/exp4_data.txt", ios::in);
    matrix m(afile);

    //上传OJ时读取
    //matrix m;
    
    m.buildJudgeMap();
    m.solutionInDFS(0);
    m.getAns();
    return 0;
}