题目描述
输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字,例如,如果输入如下矩阵: 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.
一开始以为是n*n的矩阵,写出了如下代码:
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> printMatrix(int [][] matrix) {
ArrayList<Integer> list = new ArrayList<>();
int row = matrix.length;
int col = matrix[0].length;
int rowCount = 0;
int colCount = col-1;
int revRowCount = row-1;
int revColCount = 0;
while(rowCount<=revRowCount&&colCount>=revColCount){
//打印顶部一行
for(int i=revColCount; i<=colCount; i++){
list.add(matrix[rowCount][i]);
}
rowCount++;
//打印右侧一列
for(int i=rowCount; i<=revRowCount; i++){
list.add(matrix[i][colCount]);
}
colCount--;
//打印底部一行
for(int i=colCount; i>=revColCount; i--){
list.add(matrix[revRowCount][i]);
}
revRowCount--;
//打印左侧一列
for(int i=revRowCount; i>=rowCount; i--){
list.add(matrix[i][revColCount]);
}
revColCount++;
}
return list;
}
}
在测试n*1矩阵的时候没有通过,因为在一次循环时至少打印两行、两列。如下代码加了对矩阵只有一行或只有一列的判断。
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> printMatrix(int [][] matrix) {
ArrayList<Integer> list = new ArrayList<>();
int row = matrix.length;
int col = matrix[0].length;
int rowCount = 0;
int colCount = col-1;
int revRowCount = row-1;
int revColCount = 0;
while(rowCount<=revRowCount&&colCount>=revColCount){
//打印顶部一行
for(int i=revColCount; i<=colCount; i++){
list.add(matrix[rowCount][i]);
}
rowCount++;
//打印右侧一列
for(int i=rowCount; i<=revRowCount; i++){
list.add(matrix[i][colCount]);
}
colCount--;
//打印底部一行,先排除只有一行的情况
if(rowCount>revRowCount){
break;
}
for(int i=colCount; i>=revColCount; i--){
list.add(matrix[revRowCount][i]);
}
revRowCount--;
//打印左侧一列,先排除只有一列的情况
if(colCount<0){
break;
}
for(int i=revRowCount; i>=rowCount; i--){
list.add(matrix[i][revColCount]);
}
revColCount++;
}
return list;
}
}