给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。

请注意 ,必须在不复制数组的情况下原地对数组进行操作。

示例 1:

输入: nums = [0,1,0,3,12]
输出: [1,3,12,0,0]

示例 2:

输入: nums = [0]
输出: [0]

解题思路:

这题采用双指针,将一个指针指向已排序好的序列的末尾,另一个指针指向未排序好的序列头部,从头到尾排下去即可

AC代码:

#include <bits/stdc++.h>
using namespace std;
int main(){
    int n;
    cin >> n;
    vector<int> a(n);
    for (int i=0;i<n;i++){
        int temp;
        cin >> temp;
        a[i] = temp;
    }
    int i = 0;
    int j = 0;
    while(i<n&&j<n){
        while(a[j]==0){
            j++;
        }
        int temp = a[i];
        a[i] = a[j];
        a[j] = temp;
        i++;
        j++;
    }
    for(int i=0;i<n;i++){
        cout << a[i] << ' ';
    }
    return 0;
}

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐