Leetcode - Remove Elements

iteye_17352

于 2015-07-27 10:19:34 发布

阅读量86

点赞数

CC 4.0 BY-SA版权

分类专栏： Two pointers

本文链接：https://blog.youkuaiyun.com/iteye_17352/article/details/82638058

Two pointers 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一个使用双指针技术在原地修改数组、删除特定值并返回新长度的算法。该算法通过设置两个指针分别从两端遍历数组，实现元素交换与移除，确保空间复杂度为O(1)，适用于多种编程场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array and a value, remove all instances of that value in place and return the new length.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

[分析] 注意内部while的判断条件有等号，考虑nums]{1}, val=1


public class Solution {
    public int removeElement(int[] nums, int val) {
        if (nums == null || nums.length == 0)
            return 0;
        int p = 0, q = nums.length - 1;
        while (true) {
            while (p <= q && nums[p] != val) p++;
            while (p <= q && nums[q] == val) q--;
            if (p < q) {
                nums[p++] = nums[q];
                nums[q] = val;
            } else {
                break;
            }
        }
        return q + 1;
    }
}