矩阵最小权重对角线(DFS)

今日做图论课作业,4.5节第1题,尝试用代码实现之。
矩阵最小权重对角线(DFS)

思路
dfs搜索,注意恢复现场。
时间复杂度
O(n!)
源码

#include <iostream>

using namespace std;

const int N = 10;

bool st[N];  // 该行对应的列是否被用过,注意恢复现场
int path[N]; // 保存序列
const int n = 5; 
int a[n][n];
int ans = 0x3f3f;
int pr[n];   // 记录最终的答案序列

void dfs(int u)
{
    if(u == n )
    {
        for(int i = 0; i < n; i ++ ) cout << path[i] << " ";
        int b = 0;
        for(int i = 0; i < n; i ++ ){
            b += path[i];
        }
        // cout << "weight: " << b;
        if(b < ans){
            ans = b;
            for(int i = 0; i < n; i ++ ){
                pr[i] = path[i];
            }
        }
        cout << endl;
        return;
    }
    
    for(int i = 0; i < n; i ++)
    {
        if(!st[i])
        {
            st[i] = true;
            path[u] = a[u][i];
            dfs(u + 1);
            st[i] = false; // 恢复现场
        }
    }
}

int main()
{
    for(int i = 0; i < n; i ++ ){
        for(int j = 0; j < n; j ++ ){
            cin >> a[i][j];
        }
    }
    // for(int i = 0; i < n; i ++ ){
    //     for(int j = 0; j < n; j ++ ){
    //         cout << a[i][j] << " ";
    //     }
    //     cout << endl;
    // }
    // cout << endl;
    
    dfs(0);
    
    cout << endl;
    cout << ans << endl;
    for(int i = 0; i < n; i ++ ){
        cout << pr[i] << " ";
    }
    
    return 0;
}
// 输入
// 4 5 8 10 11
// 7 6 5 7 4
// 8 5 12 9 6
// 6 6 13 10 7
// 4 5 7 9 8

结果

4 6 12 10 8 
4 6 12 7 9 
4 6 9 13 8 
4 6 9 7 7 
4 6 6 13 9 
4 6 6 10 7 
4 5 5 10 8 
4 5 5 7 9 
4 5 9 6 8 
4 5 9 7 5 
4 5 6 6 9 
4 5 6 10 5 
4 7 5 13 8 
4 7 5 7 7 
4 7 12 6 8 
4 7 12 7 5 
4 7 6 6 7 
4 7 6 13 5 
4 4 5 13 9 
4 4 5 10 7 
4 4 12 6 9 
4 4 12 10 5 
4 4 9 6 7 
4 4 9 13 5 
5 7 12 10 8 
5 7 12 7 9 
5 7 9 13 8 
5 7 9 7 7 
5 7 6 13 9 
5 7 6 10 7 
5 5 8 10 8 
5 5 8 7 9 
5 5 9 6 8 
5 5 9 7 4 
5 5 6 6 9 
5 5 6 10 4 
5 7 8 13 8 
5 7 8 7 7 
5 7 12 6 8 
5 7 12 7 4 
5 7 6 6 7 
5 7 6 13 4 
5 4 8 13 9 
5 4 8 10 7 
5 4 12 6 9 
5 4 12 10 4 
5 4 9 6 7 
5 4 9 13 4 
8 7 5 10 8 
8 7 5 7 9 
8 7 9 6 8 
8 7 9 7 5 
8 7 6 6 9 
8 7 6 10 5 
8 6 8 10 8 
8 6 8 7 9 
8 6 9 6 8 
8 6 9 7 4 
8 6 6 6 9 
8 6 6 10 4 
8 7 8 6 8 
8 7 8 7 5 
8 7 5 6 8 
8 7 5 7 4 
8 7 6 6 5 
8 7 6 6 4 
8 4 8 6 9 
8 4 8 10 5 
8 4 5 6 9 
8 4 5 10 4 
8 4 9 6 5 
8 4 9 6 4 
10 7 5 13 8 
10 7 5 7 7 
10 7 12 6 8 
10 7 12 7 5 
10 7 6 6 7 
10 7 6 13 5 
10 6 8 13 8 
10 6 8 7 7 
10 6 12 6 8 
10 6 12 7 4 
10 6 6 6 7 
10 6 6 13 4 
10 5 8 6 8 
10 5 8 7 5 
10 5 5 6 8 
10 5 5 7 4 
10 5 6 6 5 
10 5 6 6 4 
10 4 8 6 7 
10 4 8 13 5 
10 4 5 6 7 
10 4 5 13 4 
10 4 12 6 5 
10 4 12 6 4 
11 7 5 13 9 
11 7 5 10 7 
11 7 12 6 9 
11 7 12 10 5 
11 7 9 6 7 
11 7 9 13 5 
11 6 8 13 9 
11 6 8 10 7 
11 6 12 6 9 
11 6 12 10 4 
11 6 9 6 7 
11 6 9 13 4 
11 5 8 6 9 
11 5 8 10 5 
11 5 5 6 9 
11 5 5 10 4 
11 5 9 6 5 
11 5 9 6 4 
11 7 8 6 7 
11 7 8 13 5 
11 7 5 6 7 
11 7 5 13 4 
11 7 12 6 5 
11 7 12 6 4 

30
4 5 5 7 9

类似题的书本标准解法
矩阵最小权重对角线(DFS)

矩阵最小权重对角线(DFS)
矩阵最小权重对角线(DFS)
矩阵最小权重对角线(DFS)

上一篇:dfs入门


下一篇:【leetcode】剑指 Offer 13. 机器人的运动范围