LeetCode 283.Move Zeros

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ASJBFJSB/article/details/82084190

题目描述: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++;    
                }
            }
    }
}

猜你喜欢

转载自blog.csdn.net/ASJBFJSB/article/details/82084190