LeetCode-27~Remove Element

Given an array and a value, remove all instances of that value in place and return the new length. Do not allocate extra space for another array, you must do this in place with constant memory. The order of elements can be changed. It doesn't matter what you leave beyond the new length.
Example:Given input array nums = [3,2,2,3] , val = 3
Your function should return length = 2, with the first two elements of nums being 2.
给定一个array和一个值,将数组中相同的值去掉,返回新的数组的长度。

算法分析

方法一:

算法跟LeetCode-26思路一样。具体参见Remove Duplicates from Sorted Array

代码
public class Solution {
    public int removeElement(int[] nums, int val) {
        int n = nums.length;
        if (n == 0) return 0;
        
        int index = 0;
        for (int i = 0; i < n; i++) {
            if (nums[i] != val) {
                nums[index] = nums[i];
                index ++;
            }
        }
        return index;
    }
}
方法二

现在考虑如下情况:nums=[1,2,3,5,4],val=4,方法一会复制前面四个值,很没有必要。因此可以考虑把这些没有必要的步骤去掉。如下情况:nums = [4,1,2,3,5], val = 4nums[i] == val时,将左后一个值与当前值交换。

代码
public class Solution {
    public int removeElement(int[] nums, int val) {
        int n = nums.length;
        int i = 0;
        while (i < n) {
            if (nums[i] == val) {
                nums[i] = nums[n - 1];
                n --;
            } else {
                i ++;
            }
        }
        return i;
    }
}

参考

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容