欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode 283.Move Zeros

程序员文章站 2024-02-17 10:31:58
...

题目描述:Given an array nums, write a function to move all 0’s to the end of it while maintaining the relative order of the non-zero elements.
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。

样例输入:[0,1,0,3,12]
样例输出:[1,3,12,0,0]

//直观的方法就是,开辟一个数组用于保存所有非零元素
//遍历完成后回写到原数组中,之后添加对应个数的0元素
void moveZeroes(int* nums, int numsSize) {
    int nonzero[numsSize];
    int k=0;//统计非零元素的个数
    for(int i=0; i<numsSize; ++i){
        if(nums[i])
            nonzero[k++] = nums[i];
    }
    //2.回写非零元素
    int j=0;//记录新数组当前下标
    for(int i=0; i<k; ++i){
        nums[j++] = nums[i];
    }
    //3.添加0,numsSize-k为零元素的个数
    fpr(int i=0; i<numsSize-k; ++i){
        nums[k++] = 0;
    }
}

上述算法的时间复杂度为:O(n),空间复杂度也为O(n)。
上边的算法开辟了额外的数组用于存储非零元素,那么如何空间复杂度上进行优化,能能在原数组上直接操作即数据结构中的就地完成?

//two pointers思想
void moveZeroes(int* nums, int numsSize){
    int k=0;//标记非零元素
    for(int i=0;i<numsSize;++i){
        if(nums[i]!=0)
            nums[k++]=nums[i]
    }
    for(int i=k;i<numsSize;++i){
        nums[i]=0;
    }
}

交换零元素和非零元素

void moveZeroes(int* nums, int numsSize){
    int k=0;
    for(int i=0;i<numsSize;++i){
        if(nums[i]){
                //防止发生自交换,若数组中全为非零元素
                if(i!=k){
                        swap(nums[k++],nums[i]);
                    }
                else{
                    k++;    
                }
            }
    }
}