本文主要记录一下leetcode之复写零数组
给你一个长度固定的整数数组 arr,请你将该数组中出现的每一个零都复写一遍,并将其他的元素向右平移。 注意:请不要在超过该数组长度的位置写入元素。 要求:请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。 示例 1: 输入:[1,0,2,3,0,4,5,0] 输出:null 解释:调用函数后,输入的数组将被修改成:[1,0,0,2,3,0,0,4] 示例 2: 输入:[1,2,3] 输出:null 解释:调用函数后,输入的数组将被修改成:[1,2,3] 提示: 1 <= arr.length <= 10000 0 <= arr[i] <= 9 来源:力扣(LeetCode) 连接:https://leetcode-cn.com/problems/duplicate-zeros 著做权归领扣网络全部。商业转载请联系官方受权,非商业转载请注明出处。
class Solution { public void duplicateZeros(int[] arr) { int count = 0; for (int i = 0; i < arr.length; i++) { count += arr[i] == 0 ? 1 : 0; } for(int i=arr.length-1; i>=0; i--) { if(arr[i] == 0){ count--; } if(i + count < arr.length){ arr[i + count] = arr[i]; if(arr[i] == 0 && i + count + 1 < arr.length){ arr[i + count + 1] = 0; } } } } }
这里遍历数组,计算值为0的个数count,而后从后遍历数据,遇到0就递减该值,而后判断i + count是否超出数组长度,没有的话则将i的值设置到i + count,紧接着再当前位置值为0且i + count + 1不超出数组长度的时候将i + count + 1的值设置为0。网络