leetcode第一刷_Remove Element

来源:互联网 发布:c语言制表符是什么 编辑:程序博客网 时间:2024/06/16 16:27

水题。

这道题的题干直接把思路说出来了啊。。惊呆了。

两个指针,一前一后,遇到要去掉的数字就把后面的换上来,呵呵。

inline void swap(int &a, int &b){    int tp = a;    a = b;    b = tp;}class Solution {public:    int removeElement(int A[], int n, int elem) {        if(A == 0)            return 0;        int end = n-1;        while(A[end--] == elem);        for(int i=0;i<end;i++){            if(A[i] == elem){                swap(A[i], A[end]);                while(A[--end] == elem);            }        }        return end+1;    }};


0 0
原创粉丝点击