使用C语言和JAVA语言版本双指针来解决复写零问题

力扣链接:https://leetcode.cn/problems/duplicate-zeros/description/

题意:对一个数组进行复写,遇到0写两遍,非0写一遍,复写结果不能超过原数组长度,即当复写结果达到数组长度时,后面的结果数组元素直接舍弃.

例子

思路:找到复写的最后一个值,从右向左进行遍历,根据值进行不同复写即可.

步骤:1.找到复写的最后一个值:使用双指针来寻找复写的最后一个值,创建下标变量dest,cur.当cur对应值为零时,dest += 2,非零,dest++,当dest >= 数组长度时,此时cur对应值就是复写的最后一个值

2.判断dest的位置:因为存在特殊情况,当复写的最后一个值为0时,dest会前进两个位置,当进行复写时,就会导致越界.

3.复写:根据下标值情况进行复写

复写的情况

非零

为零

C语言代码

void duplicateZeros(int* arr, int arrSize) {
    //1.双指针法获取复写之后的最后一个数
    int cur = 0;
    int dest = -1;
    //1.1cur判断情况
    while(cur < arrSize) {
    //1.2cur,dest根据判断情况前进
        if(arr[cur] != 0) {
            dest++;
        }else {
            dest += 2;
        }
        //1.3判断dest是否已经越界
        if(dest >= arrSize - 1) {
            break;
        }
            cur++;
    }
    //1.4dest遍历完成,cur下标对应值为复写的最后一个数
    //判断dest情况,防止越界
    if(dest == arrSize) {
        arr[arrSize - 1] = 0;
        dest -= 2;
        cur -= 1; 
    }
    //2.从后往前进行遍历复写(保证不被覆盖)
    //2.1判断cur下标值情况
    while(cur >= 0) {
    //2.2根据情况进行不同情况复写
        if(arr[cur] != 0) {
            arr[dest] = arr[cur];
            dest -= 1;
            cur -= 1;
        }else {
                arr[dest] = 0;
                arr[dest - 1] = 0;
                dest -= 2;
                cur -= 1;
        }
    }
    //2.3cur结束,完成复写
}

JAVA代码如下

class Solution {
    public void duplicateZeros(int[] arr) {
        //1.找到复写的最后一个值
        int dest = -1;
        int cur = 0;
        int length = arr.length;
        while(cur < length) {
            if(arr[cur] != 0) {
                dest++;
            }else {
                dest += 2;
            }
            //判断dest下标情况,如果dest前移值超过数组长度则cur不前移
            if(dest >= length - 1) {
                break;
            }
            cur++;
        }
        //2.判断dest位置,防止dest下标进行复写时越界复写
        if(dest == length) {
            arr[length - 1] = 0;
            cur--;
            dest -= 2;
        }
        //3.从右到左进行复写
        while(cur >= 0) {
            if(arr[cur] != 0) {
                arr[dest] = arr[cur];
                dest--;
                cur--;
            }else {
                arr[dest] = arr[cur];
                arr[dest - 1] = arr[cur];
                cur--;
                dest -= 2;
            }
        }
    }
}