题目地址:https://leetcode.com/problems/longest-increasing-path-in-a-matrix/#/description
题目描述:Given an integer matrix, find the length of the longest increasing path.
From each cell, you can either move to four directions: left, right, up or down. You may NOT move diagonally or move outside of the boundary (i.e. wrap-around is not allowed).
我的代码:
class Solution {
public:
struct Point{
int x;
int y;
Point(int a,int b):x(a),y(b){};
};
int longestIncreasingPath(vector<vector<int>>& matrix) {
int n=matrix.size();
if(n==0) return 0;
int m=matrix[0].size();
if(m==0) return 0;
vector<vector<int>> mt;
vector<vector<Point>> pt;
for(int i=0;i<n;i++){
//vector<int> a;
mt.push_back(vector<int>());
for(int j=0;j<m;j++)
mt[i].push_back(0);
}
int dis[4][2]={-1,0,1,0,0,-1,0,1};
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
for(int k=0;k<4;k++)
if(i+dis[k][0]>=0&&i+dis[k][0]<n&&j+dis[k][1]>=0&&j+dis[k][1]<m&&matrix[i][j]<matrix[i+dis[k][0]][j+dis[k][1]])
mt[i][j]+=1;
int num=0;
pt.push_back(vector<Point>());
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
if(mt[i][j]==0)
pt[num].push_back(Point(i,j));
while(!pt[num].empty()){
pt.push_back(vector<Point>());
int l=pt[num].size();
for(int i0=0;i0<l;i0++){
int i=pt[num][i0].x;
int j=pt[num][i0].y;
for(int k=0;k<4;k++)
if(i+dis[k][0]>=0&&i+dis[k][0]<n&&j+dis[k][1]>=0&&j+dis[k][1]<m&&matrix[i][j]>matrix[i+dis[k][0]][j+dis[k][1]]){
mt[i+dis[k][0]][j+dis[k][1]]-=1;
if(mt[i+dis[k][0]][j+dis[k][1]]==0) pt[num+1].push_back(Point(i+dis[k][0],j+dis[k][1]));
}
}
num++;
}
return num;
}
};
解题思路:
题目要求找出最长的递减路,最简单的思路就是,找到所有可能的起点,然后广搜或者深搜,但那样的最坏复杂度为O(n^2*m^2*k),k是可能的起点的个数。对这一思路做一点点小的改变,首先构造一个相同大小的矩阵,矩阵的每个位置的值等于原矩阵对应位置的四个方向上比它大的点的个数,这样,一个由0,1,2,3,4构成的矩阵就形成了,而且对应于原矩阵中的最小值(非空情况下)的位置一定是0。显然,为0的位置就是所有可能的起点位置。然后进行广搜,对于每一个可能的起点,如果去掉这个点(设为无穷大),显然其四个方向上的点中比它大的位置现在变得比它小了,所以这个位置的对应新矩阵的值要减1,对所有的起点都进行该操作之后,所有的新的变成0的位置就是所求递增路径上的第二个点,然后依次找出所有第三个点,第四个点……。当再没有新的0产生时,路径就结束了,最长路径的长度也就可以确定了。
由于每一个点变成0后都只需操作一次,因此复杂度为O(n*m)。