原题链接:. - 力扣(LeetCode)
目录
给你一个长度固定的整数数组 arr
,请你将该数组中出现的每个零都复写一遍,并将其余的元素向右平移。
注意:请不要在超过该数组长度的位置写入元素。请对输入的数组 就地 进行上述修改,不要从函数返回任何东西。
双指针。
先根据“异地”操作,优化成双指针下的“就地”操作。
1. 先找到最后一个复写的数(也使用双指针算法):
(1)先判断cur位置的值
(2)决定dest向后移动一步或者两步
(3)判断一下dest是否已经到结束位置
(4)cur++
2. 处理一下边界情况:
n-1 —> 0
cur--
dest -= 2
3.“从后向前”完成复写操作
- class Solution {
- public:
- void duplicateZeros(vector<int>& arr) {
- int cur=0,dest=-1,n=arr.size();
- while(cur
- if(arr[cur]) dest++;
- else dest+=2;
- if(dest>=n-1) break;
- cur++;
- }
- if(dest==n){
- arr[n-1]=0;
- cur--,dest-=2;
- }
- while(cur>=0){
- if(arr[cur]) arr[dest--]=arr[cur--];
- else{
- arr[dest--]=0;
- arr[dest--]=0;
- cur--;
- }
- }
- }
- };