题解 | #向左移动牛群II# java
向左移动牛群II
https://www.nowcoder.com/practice/2467ddd80a2942abbaa752f3c874dd79
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[][] arr = new int[n][n]; k %= n; int count = 1; for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { arr[i][j] = count; count++; } } if (k == 0) { return arr; } for (int time = 0; time < k; time++) { int[][] res = new int[n][n]; for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { res[(i + n - 1) % n][(j + n - 1) % n] = arr[i][j]; } } arr = res; } return arr; } }
使用的是Java语言。
这道题考察的是对二维数组旋转操作的实现。
代码中的解题思路是先创建一个二维数组,然后用1到n^2的数字依次填充数组。接下来判断k是否为0,如果是,则直接返回原数组。如果k不为0,就开始进行旋转操作。旋转操作的实现是创建一个新的n x n的二维数组res,遍历原数组arr中的每个元素,并将其根据旋转规则放置到新数组res中相应的位置。旋转规则是将arr[i][j]放到res[(i + n - 1) % n][(j + n - 1) % n]的位置上。然后将新数组res赋值给原数组arr,重复k次旋转操作后,返回最终的数组arr。