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 1for (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 2for (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;}}// rotatedvector<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;}
};