移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
算法流程
-
可以保留两个指针 i 和 j,其中 i 是慢指针,j 是快指针。
-
当 nums[j] 与给定的值相等时,递增 j 以跳过该元素。只要 nums[j]!=val,我们就复制 nums[j] 到 nums[i] 并同时递增两个索引。重复这一过程,直到 j 到达数组的末尾,该数组的新长度为 i。
public int removeElement(int[] nums, int val) {
int i = 0;
for (int j = 0; j < nums.length; j++) {
if (nums[j] != val) {
nums[i] = nums[j];
i++;
}
}
return i;
}
三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,
使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
算法流程
- 特判,对于数组长度 nn,如果数组为 nullnull 或者数组长度小于 33,返回 [][]。
- 对数组进行排序。
- 遍历排序后数组:
- 若 nums[i]>0:因为已经排序好,所以后面不可能有三个数加和等于 0,直接返回结果。
- 对于重复元素:跳过,避免出现重复解
- 令左指针 L=i+1,右指针 R=n−1,当 L<R 时,执行循环:
- 当nums[i]+nums[L]+nums[R]==0,执行循环,判断左界和右界是否和下一位置重复,去除重复解。并同时将 L,RL,R 移到下一位置,寻找新的解
- 若和大于 0,说明 nums[R] 太大,R 左移
- 若和小于 0,说明 nums[L] 太小,LL 右移
AC代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int> &nums) {
int size = nums.size();
vector<vector<int>> ans;
if (size <= 2) return ans;
sort(nums.begin(), nums.end());
for (int i = 0; i < size; i++) {
if (nums[i] > 0) return ans;
if (i > 0 && nums[i] == nums[i - 1]) continue;
int L = i + 1, R = size - 1;
while (L < R) {
int sum = nums[i] + nums[L] + nums[R];
if (sum == 0) {
vector<int> t_ans = {nums[i], nums[L], nums[R]};
ans.push_back(t_ans);
while (L < R && nums[L] == nums[L + 1]) L++;
while (L < R && nums[R] == nums[R - 1]) R--;
L++, R--;
} else if (sum > 0) {
R--;
} else {
L++;
}
}
}
return ans;
}
};
四数之和
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> ans = new ArrayList<>();
if (nums == null || nums.length <= 2) return ans;
Arrays.sort(nums);
for (int i = 0; i <= nums.length - 4; i++) {
if (i > 0 && nums[i] == nums[i - 1]) continue;
for (int j = i + 1; j <= nums.length - 3; j++) {
if (j > i + 1 && nums[j] == nums[j - 1]) continue;
int target2 = target - nums[i] - nums[j];
int left = j + 1, right = nums.length - 1;
while (left < right) {
if (nums[left] + nums[right] == target2) {
ans.add(new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[left], nums[right])));
left++;
right--;
while (left < right && nums[left] == nums[left - 1]) left++;
while (left < right && nums[right] == nums[right + 1]) right--;
} else if (nums[left] + nums[right] < target2) {
left++;
} else {
right--;
}
}
}
}
return ans;
}
Comments | 0 条评论