力扣 (LeetCode)542. 01 矩阵(C语言)

清墨韵染

已于 2022-10-09 17:47:16 修改

阅读量404

点赞数 1

分类专栏：小白leetcode 文章标签： leetcode c语言广度优先动态规划

于 2022-10-09 17:46:36 首次发布

本文链接：https://blog.csdn.net/Innocence02/article/details/127230244

版权

小白leetcode 专栏收录该内容

44 篇文章 4 订阅

订阅专栏

一、环境说明

本文是 leetcode 542题：01 矩阵，使用c语言实现
方法一使用广度优先遍历实现。
方法二使用动态规划实现。
测试环境:Visual Studio 2019

二、题目说明：

三、代码展示

方法1 广度优先遍历

const int dx[4] = { 1,-1,0,0 };//偏移的坐标值
const int dy[4] = { 0,0,1,-1 };
typedef struct queue {//队列存储坐标
    int x;
    int y;
}queue;
int** updateMatrix(int** mat, int matSize, int* matColSize, int* returnSize, int** returnColumnSizes) {
    int m = matSize, n = matColSize[0];//m行 n列
    *returnSize = m;//行数
    *returnColumnSizes = (int*)calloc(m, sizeof(int));//列数
    for (int i = 0; i < m; i++) {
        returnColumnSizes[0][i] = n;//每一行有n个元素
    }
    //BFS(mat,m,n);
    int** ans = (int**)calloc(m, sizeof(int*));//答案矩阵
    int front = 0, rear = 0;
    queue* q = (queue*)calloc(m * n, sizeof(queue));//队列元素，存储坐标
    for (int i = 0; i < m; i++) {
        ans[i] = (int*)calloc(n, sizeof(int));//每一行申请n个元素
        for (int j = 0; j < n; j++) {
            if (mat[i][j]) {//mat当前元素是1
                ans[i][j] = INT_MAX;//1的距离为INT_MAX
            }
            else {//当前元素是0
                q[rear].x = i;//0的坐标入队
                q[rear++].y = j;
            }
        }
    }
    while (front != rear) {//队列非空
        int x = q[front].x;//出发坐标
        int y = q[front++].y;
        for (int i = 0; i < 4; i++) {//向4个方向遍历
            int new_x = x + dx[i], new_y = y + dy[i];//被遍历到的坐标
            if (new_x >= 0 && new_x < m && new_y >= 0 && new_y < n) {//边界保护
                if (ans[new_x][new_y] > ans[x][y] + 1) {
                    ans[new_x][new_y] = ans[x][y] + 1;//更新距离
                    q[rear].x = new_x;
                    q[rear++].y = new_y;
                }
            }
        }
    }
    return ans;
}

方法2 动态规划

#define Min(a,b) a<b?a:b
int** updateMatrix(int** mat, int matSize, int* matColSize, int* returnSize, int** returnColumnSizes) {
    int m = matSize, n = matColSize[0];//m行 n列
    *returnSize = m;//行数
    *returnColumnSizes = (int*)calloc(m, sizeof(int));//列数
    for (int i = 0; i < m; i++) {
        returnColumnSizes[0][i] = n;//每一行有n个元素
    }
    //DP(mat,m,n);
    int** ans = (int**)calloc(m, sizeof(int*));//答案矩阵
    for (int i = 0; i < m; i++) {
        ans[i] = (int*)calloc(n, sizeof(int));//每一行申请n个元素
        for (int j = 0; j < n; j++) {
            if (mat[i][j]) {//mat当前元素是1
                ans[i][j] = INT_MAX-100000;//1的距离为INT_MAX-100000，防溢出
            }
        }
    }
    for(int i = 0; i<m;i++){//从上到下
        for(int j = 0;j<n;j++){//从左到右遍历
            if(i > 0){//第0行不向上遍历
                ans[i][j]=Min(ans[i][j],ans[i-1][j]+1);
            }
            if(j > 0){//第0列不向左遍历
                ans[i][j]=Min(ans[i][j],ans[i][j-1]+1);
            }
        }
    }
    for(int i = m-1;i>=0;i--){
        for(int j = n-1;j>=0;j--){
            if(i<m-1){//最后一行不向下遍历
                ans[i][j]=Min(ans[i][j],ans[i+1][j]+1);
            }
            if(j<n-1){//最后一列不向右遍历
                ans[i][j]=Min(ans[i][j],ans[i][j+1]+1);
            }
        }
    }
    return ans;
}

四、思路分析

广度优先遍历思路分析

这道题类似图的遍历，树有从根结点出发的单源遍历，图则有从多个结点出发的多源遍历。其实二者本质上是一样的，只不过图的遍历于没有从第一个结点出发，而是直接从多个结点出发，图的多源遍历相当于树的遍历第二步。
用一个队列，记录待遍历结点的坐标。
创建ans矩阵，大小等于mat，将所有0到0的距离设为0，将所有1到0的距离设为INT_MAX(int的最大值)
将所有的0，看做一个超级0，这个超级0到每个0的距离为1。将所有0的坐标入队，等待遍历。
出队队首元素，对它遍历，遍历到的下一个坐标，如果它目前到0的最近距离＜（队首元素到0的距离+1），则更新下一个坐标到0的最近距离为二者最小值，同时下一个坐标入队。
注意设置边界。
细节看代码注释