自学内容网 自学内容网

LeetCode 1861. Rotating the Box

🔗 https://leetcode.com/problems/rotating-the-box

题目

  • 给一个 m * n 的二维数组,# 代表花,* 代表墙
  • 二维数组顺时针旋转 90 度,# 会根据重力下落,要么到 * 被挡住,要么到底被挡住
  • 返回经过旋转后的二维数组

思路

  • M * N 数组,经过旋转之后 [i, j] 会变成 [j][m - i -1]
  • solution 1 是先进行重力处理,把 # 往右移动,直到碰到 * 或者到底。这里有两种处理方式,一种是遍历,另外一种是记录当前可填充 # 的位置
  • solution 2 时边重力处理,边重力处理 #,记录当前可填充 # 的位置,性能较优

代码

class Solution {
public:
    vector<vector<char>> rotateTheBox(vector<vector<char>>& box) {
        int m = box.size(), n = box[0].size();
        /* gravity 1
        for (int i = 0; i < m; i++) {
            for (int j = n - 2; j >=0; j--) {
                if (box[i][j] == '#') {
                    int k = j;
                    while (k < n - 1 && box[i][k + 1] == '.') {
                        k++;
                    }
                    if (k > j) {
                        box[i][k] = '#';
                        box[i][j] = '.';
                    }
                }
            }
        }*/

        // gravity 2
        for (int i = 0; i < m; i++) {
            int bottom = n -1;
            for (int j = n - 1; j >=0; j--) {
                if (box[i][j] == '#') {
                    box[i][bottom] = '#';
                    if (j != bottom) box[i][j] = '.'; 
                    bottom--;         
                } else if (box[i][j] == '*') bottom = j - 1;
            }
        }
        // rotated
        vector<vector<char>> ans;
        for (int i = 0; i < n; i++) {
            vector<char> tmp;
            for (int j = 0; j < m; j++) {
                tmp.push_back(box[m - j -1][i]);
            }
            ans.push_back(tmp);
        }
        return ans;

        
    }
};

原文地址:https://blog.csdn.net/weixin_42383726/article/details/143992937

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!