LeetCode__Surrounded Regions

問題描述:
Given a 2D board containing 'X' and 'O', capture all regions surrounded by 'X'.
A region is captured by flipping all 'O's into 'X's in that surrounded region.

For example,
X X X X
X O O X
X X O X
X O X X
After running your function, the board should be:

X X X X
X X X X
X X X X
X O X X

解法:題目要求將被'X'包圍的字符'O'變成'X',因此這題的解法就是搜索邊界上的字符'O',並找出所有與邊界字符'O'連通的字符'O',將這些字符做個標記,最後再次遍歷此2D board,將所有字符'O'變成'X',並將剛纔做標記的字符變換爲原來的'O'.所以關鍵是找出所有與邊界上'O'字符連通的'O'字符,這是圖的遍歷問題,可以用DFS或BFS方法,注意用DFS遍歷的時候最好要用堆棧而不是遞歸,我剛開始用遞歸時,當矩陣大於250*250時,出現了stackoverflow問題。接下來上代碼:

1.遞歸版DFS(RunTimeError)
class Solution {
public:
	void solve(vector<vector<char> > &board) 
	{
		int m = board.size();
		if (m < 3)
			return;
		int n = board[0].size();
		if (n < 3)
			return;
		for (int i = 1; i < m - 1; i++)
		{
			dfs(board, i, 0);
			dfs(board, i, n - 1);
		}
		for (int j = 0; j < n; j++)
		{
			dfs(board, 0, j);
			dfs(board, m - 1, j);
		}
		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] == 'A') board[i][j] = 'O';
			}
		}
	}
private:
	void dfs(vector<vector<char> > &board, int x, int y)
	{
		if (x<0 || x >= board.size()|| y < 0 || y >= board[0].size())
			return;
		if (board[x][y] == 'O')
		{
			board[x][y] = 'A';
			dfs(board, x - 1, y);
			dfs(board, x + 1, y);
			dfs(board, x, y - 1);
			dfs(board, x, y + 1);
		}
	}
};

2.用堆棧實現DFS(Accept)
class Solution {
public:
	void solve(vector<vector<char> > &board) 
	{
		int m = board.size();
		if (m < 3)
			return;
		int n = board[0].size();
		if (n < 3)
			return;

		for (int i = 1; i < m - 1; i++)
		{
			dfs(board, i, 0);
			dfs(board, i, n - 1);
		}
		for (int j = 0; j < n; j++)
		{
			dfs(board, 0, j);
			dfs(board, m - 1, j);
		}
		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] == 'A') 
					board[i][j] = 'O';
			}
		}
	}
private:
	stack<int> st;
	void dfs(vector<vector<char> > &board, int x, int y)
	{
		fill_stack(board, x, y);
		while (!st.empty())
		{
			int curr = st.top();
			st.pop();
			int i = curr / board[0].size();
			int j = curr % board[0].size();

			fill_stack(board, i - 1, j);
			fill_stack(board, i + 1, j);
			fill_stack(board, i, j - 1);
			fill_stack(board, i, j + 1);
		}
	}

	void fill_stack(vector<vector<char> > &board, int x, int y)
	{
		if (x < 0 || x >= board.size() || y<0 || y >= board[0].size() || board[x][y] != 'O')
			return;
		st.push(x*board[0].size() + y);
		board[x][y] = 'A';
	}
};

3.BFS
class Solution {
public:
	void solve(vector<vector<char> > &board) 
	{
		int m = board.size();
		if (m < 3)
			return;
		int n = board[0].size();
		if (n < 3)
			return;

		for (int i = 1; i < m - 1; i++)
		{
			bfs(board, i, 0);
			bfs(board, i, n - 1);
		}
		for (int j = 0; j < n; j++)
		{
			bfs(board, 0, j);
			bfs(board, m - 1, j);
		}
		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] == 'A') 
					board[i][j] = 'O';
			}
		}
	}
private:
	queue<int> Que;
	void fill(vector<vector<char> > &board, int x, int y){
		if (x < 0 || x >= board.size() || y<0 || y >= board[0].size() || board[x][y] != 'O') 
			return;
		Que.push(x*board[0].size() + y);
		board[x][y] = 'A';
	}

	void bfs(vector<vector<char> > &board,int x, int y){
		fill(board, x, y);

		while (!Que.empty()){
			int curr = Que.front();
			Que.pop();
			int i = curr / board[0].size();
			int j = curr % board[0].size();

			fill(board, i - 1, j);
			fill(board, i + 1, j);
			fill(board, i, j - 1);
			fill(board, i, j + 1);
		}
	}
};




發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章