知识点
数组,反转
解题思路
首先k是可能大于数组长度n的,所以k对n取模减少许多不必要的操作。
定义一个反正数组的方法reverse,当start小于end时交换数组中的两个元素。
对数组进行三次反转,第一次反转整个数组,第二次反转0到k的,第三次反转k到n的。
Java代码
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param nums int整型一维数组 * @param k int整型 * @return int整型一维数组 */ public int[] rotateCows (int[] nums, int k) { // write code here int n = nums.length; k = k % n; reverse(nums,0,n - 1); reverse(nums,0,k - 1); reverse(nums,k,n - 1); return nums; } public void reverse(int[] nums, int start, int end){ while(start < end){ nums[start] = nums[start] ^ nums[end]; nums[end] = nums[start] ^ nums[end]; nums[start] = nums[start] ^ nums[end]; start ++; end --; } } }