题解 | #向左移动牛群II#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param n int整型
     * @param k int整型
     * @return int整型二维数组
     */
    public int[][] rotateII (int n, int k) {
        // write code here
        int index = 1;
        int[][] arr = new int[n][n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                arr[i][j] = index;
                index++;
            }
        }

        int[][] result = new int[n][n];
        index = 0;
        while (index < k) {
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    result[i][(j - 1 + n) % n] = arr[i][j];
                }
            }
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    arr[i][j] = result[i][j];
                }
            }

        
            for (int i = n - 1; i >= 0; i--) {
                for (int j = 0; j < n; j++) {
                    result[(i - 1 + n) % n][j] = arr[i][j];
                }
            }
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    arr[i][j] = result[i][j];
                }
            }
            index++;
        }
        return result;
    }
}

知识点：

数组元素的移动

解题思路：

先初始化一个新的二维数组，然后使用一个循环来模拟k次移动。每次移动，它将当前牛的编号放入新位置，并根据规则计算新的位置。然后，它会继续填充剩余的牛群直到所有位置都被填充。