牛客网|蛇形矩阵

我就是我 2022-11-16 15:21 119阅读 0赞

### 题目信息 ###

蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。

例如，当输入5时，应该输出的三角形为：

1 3 6 10 15
    2 5 9 14
    4 8 13
    7 12
    11

请注意本题含有多组样例输入。

**输入描述**

输入正整数N（N不大于100）

**输出描述**

输出一个N行的蛇形矩阵。

### 题解 ###

找规律，以输入4的情况下查看。从1位置开始，先向下，然后斜着向右上方走，走到第一行，再回到最左侧，依次继续。

![20210407215827233.JPG][]  我们给每个位置带上行列值，基本可以得到编码的实现思路![20210407220356259.JPG][]

即：

*  0,0位置向下走
 *  最左侧位置向斜上方走
 *  走到第一行时继续到最左侧
 *  依次继续，直到行、列值小于输入N为止

### 编码 ###

import java.io.BufferedReader;
    import java.io.IOException;
    import java.io.InputStreamReader;
    
    public class SnakeMatrix {
    
    	public static void main(String[] args) throws NumberFormatException, IOException {
    		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    		String input;
    		while ((input = br.readLine()) != null) {
    			
    			// 接收二维数组大小并定义二维数组
    			int size = Integer.parseInt(input);
    			int[][] matrix = new int[size][size];
    			
    			// 定义二维数组行变量i和列变量j
    			int i = 0; 
    			int j = 0;
    			
    			// 定义步骤变量step
    			int step = 1;
    			
    			// 在行和列变量小于矩阵大小的情况下做循环
    			while (i<size && j<size) {
    				// 给矩阵对应位置赋值
    				matrix[i][j] = step;
    				
    				// 如果行和列都为0，则向下走一行
    				if (i == j && i == 0) {
    					i++;
    				// 如果行为0，列不为0，则说明到了顶行，须要转换到最左边竖列
    				} else if (i == 0 && j > 0){
    					i = j+1;
    					j = 0;
    				// 其它情况，行减少，列增加
    				} else {
    					i--;
    					j++;
    				}
    				
    				// 步骤值增加
    				step++;
    			}
    			
    			// 定义缓冲字符，将二维数组内容放进缓冲字符输出
    			StringBuilder sb = new StringBuilder();
    			for (int m=0; m<matrix.length; m++) {
    				for (int n=0; n<matrix.length;n++) {
    					if (matrix[m][n] > 0) {
    						sb.append(matrix[m][n] + " ");
    					}
    				}
    				sb.setCharAt(sb.length()-1, '\n');
    			}
    			System.out.println(sb.substring(0, sb.length()-1));
    		}
    	}
    }

[20210407215827233.JPG]: https://img-blog.csdnimg.cn/20210407215827233.JPG
[20210407220356259.JPG]: /images/20221022/a4d3999c50c6432b929044e2d26465a1.png