这种题目有好几种写法
1、裸BFS,有点像SPFA
class Solution {
public:
int swimInWater(vector<vector<int>>& heights) {
int n = heights.size(), m = heights[0].size();
vector<vector<int>> f(n, vector<int>(m, INT_MAX));
int dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};
f[0][0] = heights[0][0];
queue<int> q;
q.push(0);
while(!q.empty()) {
auto t = q.front();
q.pop();
int x = t / m;
int y = t % m;
for (int d = 0; d < 4; d++) {
int a = x + dx[d], b = y + dy[d];
if (a >= 0 && a < n && b >= 0 && b < m) {
int tmp = max(f[x][y], heights[a][b]);
if (tmp >= f[a][b]) continue;
f[a][b] = tmp;
q.push(a*m + b);
}
}
}
return f[n-1][m-1];
}
};
2、并查集
并查集的思路挺奇妙的,首先我们从小到大枚举时间,如果说 时间 大于矩阵的某一个格子的高度,如果同时也大于格子上下左右的格子高度,那么俩个格子就是联通的,就可以加一条边了,每个格子最多加四条边。如果说第一个格子和最后一个格子联通了,那么算法就可以结束了。
class Solution {
public:
int pre[2500+7];
int dir[4][2]={{0,1},{1,0},{0,-1},{-1,0}};
int find(int x){
return pre[x]==x ? x:pre[x]=find(pre[x]);
}
void merge(int x,int y){
int fx=find(x);
int fy=find(y);
if(fx!=fy) pre[fx]=fy;
}
vector<int >ve[2500+7];
int swimInWater(vector<vector<int>>& grid) {
int n = grid.size();
int m = grid[0].size();
int end = n*m-1;
for(int i=0;i<=end;i++){
ve[grid[i/m][i%m]].push_back(i);
}
for(int i=0;i<=end;i++) pre[i]=i;
for(int ans=0;ans<=end;ans++){
for(int j=0;j<ve[ans].size();j++){
int i=ve[ans][j];
int tx=i/m;
int ty=i%m;
if(ans<grid[tx][ty]) continue ;
for(int j=0;j<4;j++){
int dx=tx+dir[j][0];
int dy=ty+dir[j][1];
if(dx<0||dy<0||dx>=n||dy>=m) continue ;
if(grid[dx][dy]>ans) continue ;
merge(dx*m+dy,i);
}
}
if(find(0)==find(n*m-1)) return ans;
}
return -1;
}
};
3、最短路(dijkstra)
待补
4、二分+bfs
可以二分一个时间,然后用bfs判断能否到达终点。