最近一直在复习一些算法及数据结构方面的东西,就找了一个适合找工作笔试的题目,在剑指Offer上刷了几道题目,发现对复习知识点还是很有用的,推荐要找工作的伙伴去剑指Offer刷题。
题目描述
输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字,例如,如果输入如下4 X 4矩阵: 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.
这种题目一看就是环形取数,直接分四个方向控制xy的大小,不过需要注意数组越界的情况,一不小心就越界了,在上下左右转换方向的时候可能会出现重复判断的情况,这部分情况需要过滤掉,我这里用了一个辅助矩阵来标记。也相当于是一种记忆化dp。
代码如下:
import java.util.ArrayList;
public class Solution {
public static ArrayList<Integer> printMatrix(int [][] matrix) {
ArrayList<Integer> lists = new ArrayList<Integer>();
int len = matrix[0].length,int i = 0 , j = 0,count = 0;
int column = 0;
for(int a[]: matrix) column++;
int flag[][] = new int[column][len];
for(int k = 0 ; k < column ; k++){
for(int l = 0 ; l < len ; l++){
flag[k][l] = 0;
}
}
int countAllData = 0;
while(true){
while(i < len - count){ //右
if(i < len && j < column && flag[j][i] == 0 ){
lists.add(matrix[j][i]);
flag[j][i] = 1;
countAllData++;
}
i++;
}
i--;
while(j < column - count){ //下
if(i < len && j < column && flag[j][i] == 0 ){
lists.add(matrix[j][i]);
flag[j][i] = 1;
countAllData++;
}
j++;
}
j--;
while(i >= count){ //左
if( i < len && j < column && flag[j][i] == 0 ){
lists.add(matrix[j][i]);
flag[j][i] = 1;
countAllData++;
}
i--;
}
i++;
while(j > count){ //上
if(i < len && j < column && flag[j][i] == 0 ){
lists.add(matrix[j][i]);
flag[j][i] = 1;
countAllData++;
}
j--;
}
j++;
count++;
if(countAllData == len*column) {
break;
}
}
return lists;
}
}