广搜和深搜都能解决,但是LeetCode上使用深搜时会栈溢出
DFS:
<span style="font-size:18px;">/*LeetCode Surrounded Regions * 题目:给定一个字符数组,由'X'和'O'组成,找到所有被x包围的o并将其替换为x * 思路:只要替换被包围的o就行,如果有一个o是边界或者上下左右中有一个是o且这个o不会被替换,则该点也不会被替换 * 从四条边开始,因为在这4周的一定不是被包围的所以用他们开始找到广搜的队列,如果队列为空,那么就是所有的o都被包围 */ package javaTrain; public class Train25 { public static void solve(char[][] board) { long n = board.length; if(n==0) return; long m = board[0].length; for(long i = 0;i < m;i++){ //对第一行和最后一行的字符进行广搜 bfs(board,0,i); bfs(board,n-1,i); } for(long j = 1;j < n-1;j++){ //对第一列和最后一列的字符进行广搜,去除4条边重复的字符 bfs(board,j,0); bfs(board,j,m-1); } for(int i = 0;i < n;i++){ for(int j = 0;j < m;j++){ if(board[i][j] == 'O') board[i][j] = 'X'; //被包围的o需取代 else if(board[i][j] == '$') board[i][j] = 'O'; //标记的不被包围的o保持原样 } } } private static void bfs(char[][] board,int i,int j){ long n = board.length; long m = board[0].length; if(i < 0 || i>=n||j<0||j>=m||board[i][j] != 'O') return; //边界的点都不被包围 board[i][j] = '$'; bfs(board,i-1,j); bfs(board,i,j-1); bfs(board,i+1,j); bfs(board,i,j+1); } public static void main(String args[]){ char board[][] = {{'O','X','O'},{'X','O','X'},{'O','X','O'}}; solve(board); for(int i = 0;i < board.length;i++){ for(int j = 0;j < board[0].length;j++){ System.out.print(board[i][j]); } System.out.println(); } } } </span>
<span style="font-size:18px;">// LeetCode, Surrounded Regions // BFS,时间复杂度O(n),空间复杂度O(n) class Solution { public: void solve(vector<vector<char>> &board) { if (board.empty()) return; const int m = board.size(); const int n = board[0].size(); for (int i = 0; i < n; i++) { bfs(board, 0, i); bfs(board, m - 1, i); } for (int j = 1; j < m - 1; j++) { bfs(board, j, 0); bfs(board, j, n - 1); } for (int i = 0; i < m; i++) for (int j = 0; j < n; j++) if (board[i][j] == 'O') board[i][j] = 'X'; else if (board[i][j] == '+') board[i][j] = 'O'; } private: void bfs(vector<vector<char>> &board, int i, int j) { typedef pair<int, int> state_t; queue<state_t> q; const int m = board.size(); const int n = board[0].size(); auto is_valid = [&](const state_t &s) { const int x = s.first; const int y = s.second; if (x < 0 || x >= m || y < 0 || y >= n || board[x][y] != 'O') return false; return true; }; auto state_extend = [&](const state_t &s) { vector<state_t> result; const int x = s.first; const int y = s.second; // 上下左右 const state_t new_states[4] = {{x-1,y}, {x+1,y}, {x,y-1}, {x,y+1}}; for (int k = 0; k < 4; ++k) { if (is_valid(new_states[k])) { // 既有标记功能又有去重功能 board[new_states[k].first][new_states[k].second] = '+'; result.push_back(new_states[k]); } } return result; }; state_t start = { i, j }; if (is_valid(start)) { board[i][j] = '+'; q.push(start); } while (!q.empty()) { auto cur = q.front(); q.pop(); auto new_states = state_extend(cur); for (auto s : new_states) q.push(s); } } }; </span>
【LeetCode】 Surrounded Regions (BFS && DFS)
原文:http://blog.csdn.net/ymzmdx/article/details/44709089