前提:对于一个给定的二维整数矩阵,如果一个元素i的某个4-相邻位置上的另一个元素j正好值为i-1则i到j可以形成一条通路。
输入:M个N*N整数矩阵;
输出:对于每个输入的矩阵,输出 矩阵中可形成的最长通路长度 及 在该最长通路长度的情况下最小的起始点值。
e.g.
input:
4 5 6 1 2 3 7 0 5 8 2 1 6 7 3 4
output:
4 0
idea:
遍历输入矩阵中每个位置,compute()计算以该位置为起点的最长route长度->找出 最长route长度 及 该长度下最小的起点值。
compute(r,c)
traverse 4个方向上的相邻位置: if 当前位置到该相邻位置(nr,nc)可以形成通路(input[r][c]==input[nr][nc]-1) neighbor_route_length:=compute(nr,nc) end end if traverse 完4个方向上都没形成任何通路,返回0 else 有找到通路,返回 neighbor_route_length的最大值+1
-----------------------------------分割线
可以引入缓存矩阵buffer[N][N]作为hash table存储先前计算过的compute(r,c)值
i.e. 对于某个确定可以形成通路的相邻位置(nr,nc),如果buffer中有缓存已经计算过的route_length值,则不再调用compute,直接提取buffer[nr][nc]。
compute2(r,c)
traverse 4个方向上的相邻位置: if 当前位置到该相邻位置(nr,nc)可以形成通路(input[r][c]==input[nr][nc]-1) if buffer[nr][nc]已有记录 neighbor_route_length:=buffer[nr][nc] else //没有记录 neighbor_route_length:=compute(nr,nc) end end end if traverse 完4个方向上都没形成任何通路,返回0 else 有找到通路,返回 neighbor_route_length的最大值+1
code:
class Solution { public: pair<int,int> getAnswer(vector<vector<int>> input) { int N=input.size(); vector<vector<int>> buffer(N,vector<int>(N,-1)); int longest_length=-1, starting_point=-1; for (int r=0;r<N;r++) { for (int c=0;c<N;c++) { int current_length=compute_route(N,input,buffer,r,c); if (current_length>longest_length|| (current_length==longest_length&&starting_point>input[r][c])) { longest_length=current_length; starting_point=input[r][c]; } } } return pair<int,int>(longest_length, starting_point); }
//虽然突然想起好像原始题目是每个位置的所有4个方向中只有最多一条通路
//但是因为比较无聊就加大一点难度:每个位置有可能有多条通路 要找出最长的一条 罢了 //what if there might be more than 1 route from a starting point int compute_route(int N, vector<vector<int>>& input, vector<vector<int>>& buffer, int r, int c) { int neighbors[4][2]={{r-1,c},{r,c-1},{r,c+1},{r+1,c}}; int neighbor_route_length=-1, answer; for (int i=0;i<4;i++) { int neighbor_r=neighbors[i][0], neighbor_c=neighbors[i][1]; if (neighbor_r>=0&&neighbor_r<N&&neighbor_c>=0&&neighbor_c<N) { if (input[neighbor_r][neighbor_c]==input[r][c]+1) { if (buffer[neighbor_r][neighbor_c]!=-1) { neighbor_route_length=max(neighbor_route_length, buffer[neighbor_r][neighbor_c]); } else { neighbor_route_length=max(neighbor_route_length,compute_route(N, input, buffer, neighbor_r, neighbor_c)); } } } } if (neighbor_route_length==-1) //no route found answer=0; else answer=neighbor_route_length+1; buffer[r][c]=answer; return answer; } };