初刷力扣,还是先从基础的做起,而双指针法在基础题中的应用非常广泛。
LeetCode 27、977、15、18就是我这几天所做到的可以用双指针法很好解决的题目:
27:移除元素
相当于把数组分为两部分:前一部分为有效部分,不等于val的数,后一部分为无效部分,即等于val的数,最后返回前一部分的长度,直接上代码:
class Solution {
public int removeElement(int[] nums, int val) {
int j = nums.length - 1;
for (int i = 0; i <= j; i++) {
if (nums[i] == val) {
swap(nums, i--, j--);
}
}
return j + 1;
}
void swap(int[] nums, int i, int j) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
}
题干的说明也要稍稍注意一下:
977:有序数组的平方
因为给定数组为非递增顺序,且返回平方后的数组也要是非递增顺序,即平方后的最大值要么在参数数组的头部和尾部。此时即可用双指针的思想,在数组头部和尾部各放置一个指针进行操作。
《代码随想录》中有一张一目了然的动图:
class Solution {
public int[] sortedSquares(int[] nums) {
int r=nums.length-1;
int l=0;
int []res=new int[r+1];
int index=res.length-1;//这里不能令index=0
//因为只能确定平方后最大的数在两端,所以必须这样倒着将结果放入
while(l<=r)
{
if(nums[l]*nums[l] > nums[r]*nums[r])
{
res[index--]=nums[l]*nums[l];
l++;
}
else{
res[index--]=nums[r]*nums[r];
r--;
}
}
return res;
}
}
15:三数之和
这题一开始我想用哈希法来着,不过试着试着就放弃了,似乎还是双指针法更简洁:
大致思路是:首先将数组排序,然后for循环,i从下标0的地方开始,同时定一个下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上。
当nums[left]+nums[right]+nums[i]>0时,说明三数之和过大,right下标就应该向左移动,这样才能让三数之和小一些。<0时同理,移动left下标。
而nums[left]+nums[right]+nums[i]=0时,也并不是万事大吉了,因为不能计入重复出现的数,所以还需要有进一步的优化操作。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List <List<Integer>> result= new ArrayList<>();
for(int i=0;i<nums.length;i++)
{
if (i > 0 && nums[i] == nums[i - 1])
continue;//因为答案不包含重复,所以遇见先前已遍历的数字就跳出本次循环
int left=i+1;
int right=nums.length-1;
while(left<right)
{
int sum=nums[left]+nums[right]+nums[i];
if(sum>0)
right--;
else if(sum<0)
left++;
//当sum为0时
else{
result.add(Arrays.asList(nums[left],nums[right],nums[i]));
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) left++;
//以上均是为了防止计入重复的数
left++;right--;
}
}
}
return result;
}
}
18:四数之和
这题算是上一题的翻版,运用的思想都是一样的:
三数之和的双指针解法是一层for循环num[i]为确定值,然后循环内有left和right下标作为双指针,找到nums[i] + nums[left] + nums[right] == 0。
四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值,依然是循环内有left和right下标作为双指针,找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况,三数之和的时间复杂度是O(n2),四数之和的时间复杂度是O(n3) 。
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
Arrays.sort(nums);
List<List<Integer>> result=new ArrayList<>();
for(int i=0;i<nums.length;i++)
{
if(i>0 &&nums[i]==nums[i-1])
continue;
for(int j=i+1;j<nums.length;j++)
{
if(j>i+1 &&nums[j]==nums[j-1])
continue;
int left=j+1;
int right=nums.length-1;
while(left<right)
{
int sum=nums[i]+nums[j]+nums[left]+nums[right];
if(sum>target)
right--;
else if(sum<target)
left++;
else{
result.add(Arrays.asList(nums[left],nums[right],nums[i],nums[j]));
while(right>left &&nums[right]==nums[right-1]) right--;
while(right>left &&nums[left]==nums[left+1]) left++;
left++;right--;
}
}
}
}
return result;
}
}
参考资料:代码随想录