前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >【原创】90%的人都不会做的一道笔试题

【原创】90%的人都不会做的一道笔试题

作者头像
田维常
发布2020-09-10 11:55:31
2760
发布2020-09-10 11:55:31
举报

数组操作的题目,有的确实比较容易,但并非每个问题都是如此。今天就来看道90%的人都不会做的笔试题。请看题:

给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。

不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例:

给定 nums = [3,1,4,3], val = 3,函数应该返回新的长度 2,并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。

请暂时不要看下面,先思考一下,有没有解答思路。

解题思路

现在考虑数组包含很少的要删除的元素的情况。例如,num=[1,2,3,5,4],Val=4。按照很多人的思维会对前四个元素进行复制操作,其实咱们没必要对前四个元素做复制操作。另一个例子是 num=[4,1,2,3,5],Val=4。也似乎没有必要将 [1,2,3,5][1,2,3,5] 这几个元素左移一步,因为问题描述中提到元素的顺序可以更改。

实现

当我们遇到 nums[i] = val 时,我们可以将当前元素与最后一个元素进行交换,并释放最后一个元素。这实际上使数组的大小减少了 1。

请注意,被交换的最后一个元素可能是您想要移除的值。但是不要担心,在下一次迭代中,我们仍然会检查这个元素。

最多遍历 n 步。在这个方法中,赋值操作的次数等于要删除的元素的数量。因此,如果要移除的元素很少,效率会更高。

代码如下:

代码语言:javascript
复制
    public static int removeElement(int[] nums, int val) {
        if(nums == null){
            return 0;
        }
        int i = 0;
        int n = nums.length;
        while (i < n) {
            //如果相等
            //(注意当当前元素是5的时候,此时i并没有加1)
            if (nums[i] == val) {
                //当前这个位置
                nums[i] = nums[n - 1];
                //数组大小减1
                n--;
            } else {
                i++;
            }
        }
        return n;
    }

测试

场景1:数组为null;

代码语言:javascript
复制
    public static void main(String[] args) {
        int[] nums =null; 
        int val = 5;
        System.out.println(removeElement(nums, val));
    }

结果

场景2:数组中只有一个5

代码语言:javascript
复制
 public static void main(String[] args) { 
        int[] nums ={5}; 
        int val = 5;
        System.out.println(removeElement(nums, val));
    }

结果

场景三:数组中有一个5

代码语言:javascript
复制
public static void main(String[] args) { 
        int[] nums ={1, 3, 5, 6}; 
        int val = 5;
        System.out.println(removeElement(nums, val));
}

结果

场景4:数组中有多个5

代码语言:javascript
复制
public static void main(String[] args) { 
        int[] nums ={1, 3, 5,5, 6}; 
        int val = 5;
        System.out.println(removeElement(nums, val));
}

结果:

场景4:数组中没有5

代码语言:javascript
复制
public static void main(String[] args) { 
        int[] nums ={1, 3, 4, 6}; 
        int val = 5;
        System.out.println(removeElement(nums, val));
}

结果

场景5:数组的最后一个元素是5

代码语言:javascript
复制
public static void main(String[] args) { 
        int[] nums ={1, 3, 4, 5}; 
        int val = 5;
        System.out.println(removeElement(nums, val));
}

结果

场景6:数组中有多个5,并且最后一个元素是5

代码语言:javascript
复制
public static void main(String[] args) {
        int[] nums = {1, 3, 5, 6, 8, 9, 5, 5};
        int val = 5;
        System.out.println(removeElement(nums, val));
}

结果

最后一种场景:数组中全是5

代码语言:javascript
复制
public static void main(String[] args) {
        int[] nums = {5, 5, 5, 5};
        int val = 5;
        System.out.println(removeElement(nums, val));
}

结果

ok,以上几种场景都涉及到了,并且结果正确。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-09-02,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Java后端技术全栈 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档