双指针法高效解决「移除元素」问题

双指针法高效解决「移除元素」问题


双指针法高效解决「移除元素」问题

一、问题描述

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

  • 必须原地修改输入数组
  • 不需要考虑数组中超出新长度后面的元素
  • 空间复杂度应为 O(1)

二、解法解析:双指针法

1. 核心思想

采用首尾双指针策略:

  • left 指针从数组头部开始,负责构建新数组
  • right 指针从数组尾部开始,提供替换元素

2. 算法步骤

java 复制代码
public int removeElement(int[] nums, int val) {
    int left = 0, right = nums.length - 1;  // 初始化双指针
    while (left <= right) {  // 循环条件
        if (nums[left] == val) {  // 发现目标值
            nums[left] = nums[right];  // 用末尾元素覆盖
            right--;  // 右指针左移
        } else {
            left++;  // 非目标值,左指针右移
        }
    }
    return left;  // left即为新长度
}

3. 执行过程示例

nums = [3,2,2,3], val = 3 为例:

  1. 初始状态:[3,2,2,3], left=0, right=3
  2. nums[0]==3 → 替换为nums[3] → [3,2,2,3], right=2
  3. nums[0]==3 → 替换为nums[2] → [2,2,2,3], right=1
  4. nums[0]==2 → left=1
  5. nums[1]==2 → left=2
  6. left>right 循环结束
  7. 返回 left=2,新数组为 [2,2,...]

三、关键点分析

  1. 边界条件while (left <= right) 确保处理所有元素
  2. 元素交换:发现目标值时直接覆盖而非交换,减少操作
  3. 指针移动
    • 只有确认当前元素不是目标值时才移动左指针
    • 每次覆盖操作后右指针必定左移

四、复杂度分析

指标 说明
时间复杂度 O(n) 最多遍历数组一次
空间复杂度 O(1) 只使用了常数级别的额外空间

五、与其他解法的比较

1. 快慢指针法

java 复制代码
int slow = 0;
for (int fast = 0; fast < nums.length; fast++) {
    if (nums[fast] != val) {
        nums[slow++] = nums[fast];
    }
}
return slow;
  • 更适合目标值较少的情况
  • 元素移动次数可能更多

2. 本解法的优势

  • 在目标值较多时效率更高
  • 每个目标值只需一次赋值操作
  • 保留了数组末端的原有元素

六、实际应用场景

这种双指针技巧适用于:

  1. 需要原地修改数组的问题
  2. 需要保持部分元素顺序的问题
  3. 资源受限环境下的数组操作

七、总结

这种首尾双指针解法:

  1. 是处理数组原地修改的高效方法
  2. 通过巧妙地指针移动减少不必要的操作
  3. 体现了算法设计中空间换时间的思想
  4. 需要熟练掌握指针边界条件的控制
相关推荐
不穿格子的程序员18 小时前
从零开始写算法——普通数组类题:数组操作中的“翻转技巧”与“前后缀分解”
数据结构·算法
逝雪Yuki18 小时前
简单多源BFS问题
算法·leetcode·bfs·广度优先遍历
curry____30318 小时前
study in PTA(高精度算法与预处理)(2025.12.3)
数据结构·c++·算法·高精度算法
ChoSeitaku18 小时前
高数强化NO6|极限的应用|连续的概念性质|间断点的定义分类|导数与微分
人工智能·算法·分类
代码游侠18 小时前
学习笔记——栈
开发语言·数据结构·笔记·学习·算法
自然语18 小时前
人工智能之数字生命-情绪
人工智能·算法
Ayanami_Reii18 小时前
进阶数据结构应用-维护序列
数据结构·算法·线段树
_w_z_j_18 小时前
mari和shiny() (多状态dp数组)
算法
CoderYanger18 小时前
C.滑动窗口-越长越合法/求最短/最小——2904. 最短且字典序最小的美丽子字符串
java·开发语言·数据结构·算法·leetcode·1024程序员节
Tim_1019 小时前
【算法专题训练】33、堆
算法