面试题12. 矩阵中的路径
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。
[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]
但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。
示例 1:
输入:board = [[“A”,“B”,“C”,“E”],[“S”,“F”,“C”,“S”],[“A”,“D”,“E”,“E”]], word = “ABCCED”
输出:true
示例 2:
输入:board = [[“a”,“b”],[“c”,“d”]], word = “abcd”
输出:false
提示:
1 <= board.length <= 200
1 <= board[i].length <= 200
思路
回溯法。
Go实现需要注意的一点。
go函数传参的时候,切片传递的实际上是指针,因此形参的值会影响实参的值。
而在C++/Java中,传递vector或者数组都是创建副本,形参不影响实参。
因此用C++/Java可以直接操作传进来的board,但是Go则另外需要建立一个矩阵来记录状态。
最下面做了传参实验,感兴趣的可以看一眼。
我的解答
func exist(board [][]byte, word string) bool {
for i:=0;i<len(board);i++{
for j:=0;j<len(board[0]);j++{
mark:=make([][]bool,len(board))
for k:=0;k<len(board);k++{
mark[k]= make([]bool,len(board[0]))
}
if search(board,mark,i,j,word){
return true
}
}
}
return false
}
func search(board [][]byte,mark [][]bool,i,j int,word string)bool{
if len(word)==0{
return true
}
if i<len(board)&&i>=0&&j<len(board[0])&&j>=0{
if mark[i][j]==false&&board[i][j]==byte(word[0]){
mark[i][j]=true
down:=search(board,mark,i+1,j,word[1:])
up:=search(board,mark,i-1,j,word[1:])
right:=search(board,mark,i,j+1,word[1:])
left:=search(board,mark,i,j-1,word[1:])
if down||up||right||left{
return true
}else{
mark[i][j]=false
}
}
}
return false
}
传参实验
Go
代码:
package main
import "fmt"
func main() {
a:=[]int{1,2,3}
fmt.Println(a)
modify(a)
fmt.Println(a)
}
func modify(a []int) {
a[0]=4
}
结果:
C++
#include <iostream>
#include <vector>
using namespace std;
void modify(vector<int> nums) {
nums[0] = 4;
}
void printVector(vector<int> nums) {
for (int i = 0; i < nums.size(); i++) {
cout << nums[i] << " ";
}
}
int main()
{
vector<int>nums = { 1,2,3 };
printVector(nums);
cout << endl<<"--------" << endl;
modify(nums);
printVector(nums);
return 0;
}
结果: