剑指Offer-顺时针打印矩阵

        最近一直在复习一些算法及数据结构方面的东西,就找了一个适合找工作笔试的题目,在剑指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;
   }
    
}

点赞