LeetCode——Spiral Matrix

最新推荐文章于 2024-07-31 23:50:13 发布

原创最新推荐文章于 2024-07-31 23:50:13 发布 · 170 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java #leetcode

算法专栏收录该内容

83 篇文章

订阅专栏

本文介绍了一种螺旋遍历矩阵的算法实现，提供了两种Java代码示例，一种使用额外的布尔矩阵来跟踪已访问元素，另一种则通过动态调整边界来简化逻辑。这两种方法都能有效地按螺旋顺序返回矩阵的所有元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

Example 1:
Input:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
Output: [1,2,3,6,9,8,7,4,5]
Example 2:
Input:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
Output: [1,2,3,4,8,12,11,10,9,5,6,7]

解法

一个复杂又白痴的做法，按要求螺旋遍历数组一个一个把数据加进res

public List<Integer> spiralOrder(int[][] matrix) {
		List<Integer> res=new ArrayList<Integer>();
		if(matrix==null||matrix.length==0)
            return res;
		int n=matrix.length;
		int m=matrix[0].length;
		boolean[][] visited=new boolean[n][m];
		for(int i=0;i<n;i++)
		{	if(i>m-1||visited[i][i])
				break;
			right(matrix,visited,res,i,i,m-1-i);
			if(i>m-1||i>=n-1||visited[i+1][m-1-i])
				break;
			down(matrix,visited,res,m-1-i,i+1,n-1-i);
			if(i>m-2||i>n-1||visited[n-1-i][m-2-i]||i>m-2)
				break;
			left(matrix,visited,res,n-1-i,m-2-i,i);
			if(i>n-2||i>=m||visited[n-2-i][i])
				break;
			up(matrix,visited,res,i,n-2-i,1+i);
		}
		return res;
	}
	private void up(int[][] matrix, boolean[][] visited, List<Integer> res, int col, int start, int end) {
		// TODO Auto-generated method stub
		for(int i=start;i>=end;i--)
		{
			if(visited[i][col])
				break;
			visited[i][col]=true;
			res.add(matrix[i][col]);
		}
	}

	private void left(int[][] matrix, boolean[][] visited, List<Integer> res, int row, int start, int end) {
		// TODO Auto-generated method stub
		for(int i=start;i>=end;i--)
		{
			if(visited[row][i])
				break;
			visited[row][i]=true;
			res.add(matrix[row][i]);
		}
	}

	private void down(int[][] matrix, boolean[][] visited, List<Integer> res, int col, int start, int end) {
		// TODO Auto-generated method stub
		for(int i=start;i<=end;i++)
		{
			if(visited[i][col])
				break;
			visited[i][col]=true;
			res.add(matrix[i][col]);
		}
	}
    private void right(int[][] matrix, boolean[][] visited, List<Integer> res, int row,int start,int end) {
		// TODO Auto-generated method stub
		for(int i=start;i<=end;i++)
		{
			if(visited[row][i])
				break;
			visited[row][i]=true;
			res.add(matrix[row][i]);
		}
    }

Runtime: 1 ms, faster than 100.00% of Java online submissions for Spiral Matrix.
Memory Usage: 37 MB, less than 64.52% of Java online submissions for Spiral Matrix.
简洁一点的写法

public List<Integer> spiralOrder(int[][] matrix) {
    	List<Integer> res=new ArrayList<Integer>();
		if(matrix==null||matrix.length==0)
            return res;
		int n=matrix.length;
		int m=matrix[0].length;
		int up=0,down=n-1,left=0,right=m-1;
		while(true){
			for(int j=left;j<=right;j++)
				res.add(matrix[up][j]);
			if(++up>down)
				break;
			for(int j=up;j<=down;j++)
				res.add(matrix[j][right]);
			if(--right<left)
				break;
			for(int j=right;j>=left;j--)
				res.add(matrix[down][j]);
			if(--down<up)
				break;
			for(int j=down;j>=up;j--)
				res.add(matrix[j][left]);
			if(++left>right)
				break;
		}
		return res;
    }