当我们遇到移除元素的题目时,通常可以考虑使用双指针的方式来解,下面从leetcode上面总结了几道相关的题目,都是基础入门级别的,非常适合用来入门双指针相关操作的题目。
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
定义快慢指针,快指针遇到目标值,就跳过继续往下找,而慢指针始终指向待替换的位置。
public int removeElement(int[] nums, int val) { int len = nums.length; if (len == 0) { return 0; } int slow = 0; int fast = 0; while (fast < len) { if (nums[fast] != val) { nums[slow] = nums[fast]; slow++; } fast++; } return slow; }
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
和上一题思路差不多,定义快慢指针,慢指针始终指向待替换值的下标,快指针每次和慢指针的前一个下标进行比较,如果相等则跳过,否则快指针用当前值替换慢指针指向的当前值。
public int removeDuplicates(int[] nums) { int len = nums.length; if (len == 0) { return 0; } int slow = 1; int fast = 1; while (fast < len) { if (nums[slow - 1] != nums[fast]) { nums[slow] = nums[fast]; slow++; } fast++; } return slow; }
给定一个数组 nums,编写一个函数将所有 0 移动到数组的末尾,同时保持非零元素的相对顺序。
还是通过快慢指针的套路,最后处理一下慢指针之后的数全为0即可。
public void moveZeroes(int[] nums) { int len = nums.length; if (len < 2) { return; } int slow = 0; int fast = 0; while (fast < len) { if (nums[fast] != 0) { nums[slow] = nums[fast]; slow++; } fast++; } while (slow < len) { nums[slow] = 0; slow++; } }
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
这是一道操作链表的题目,同样可以利用双指针的思路,先从头部节点开始找,找到值不为value的节点,跳出循环,之后的过程就还是定义两个指针,cur就相当于慢指针,next就相当于快指针,如果快指针指向的节点就是要删除的,则先跳过,否则让cur来到next位置。
public ListNode removeElements(ListNode head, int val) { while (head != null) { if (head.val != val) { break; } head = head.next; } ListNode cur = head; while (cur != null) { ListNode next = cur.next; if (next != null && next.val == val) { cur.next = next.next; } else { cur = next; } } return head; }