双指针问题


前言

问题:给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-element
这个问题实现起来并不难,难的是让你使用最优的办法求解,偶尔在b站上看到一个特别妙的解法,满足了文中所给的时间复杂度和空间复杂度的要求,令人拍案叫绝(本人没有学过数据结构,下面只是提供思路,只用一种测试用力,如果考虑不周,自行加上完善,大佬勿cue,狗头)


一、暴力解法,时间复杂度O(n^2)

思路:遍历数组,如果遇到需要删除的数字,使用后面的数字整体移动,进行覆盖

int removeElement(int* nums, int numsSize, int val) {
    int i = 0;
    int j = 0;
    for (i = 0; i < numsSize; i++)
    {
        if (nums[i] == val)//如果相等,后面的数据往前移动,覆盖掉相等的数
        {
            for (j = i; j < numsSize - 1; j++)
            {
                nums[j] = nums[j + 1];
            }
            numsSize--;
            i--;//不能增加,抵消i++作用
        }
    }

}

int main()
{
    int i = 0;
    int nums[8] = { 0, 1, 2, 2, 3, 0, 4, 2 };
    int val = 2;
    int tmp = removeElement(nums, 8, 2);
    printf("%d\n", tmp);
    for (i = 0; i < tmp; i++)
    {
        printf("%d ", nums[i]);
    }
    return 0;
}

二、时间复杂度O(n),空间复杂度O(n)

思路:开辟一个相同大小(足够大)的数组,如果遇到不需要删除的数字,把它复制到我们新开辟的数组中,遍历一遍数组之后,再将新开辟数组中的数据拷贝到原数组中。

int removeElement(int* nums, int numsSize, int val) {
    int tmp[8] = { 0 };
    int i = 0;
    int j = 0;
    int newnumsSize = 0;
    for (i = 0; i < numsSize; i++)
    {
        if (nums[i] != val)
        {
            tmp[j] = nums[i];
            j++;
        }
        
    }
    newnumsSize = j;
    for (i = 0; i < newnumsSize; i++)
    {
        nums[i] = tmp[i];
    }
    return newnumsSize;
}


int main()
{
    int i = 0;
    int nums[8] = { 0, 1, 2, 2, 3, 0, 4, 2 };
    int val = 2;
    int tmp = removeElement(nums, 8, 2);
    printf("%d\n", tmp);
    for (i = 0; i < tmp; i++)
    {
        printf("%d ", nums[i]);
    }
    return 0;
}

三、双指针解法,时间复杂度O(n),空间复杂度O(1)

思路:使用两个下标,src和dst,如果遇到了不需要删除的数字,二者都++,同时把Src的数字复制到dst位置,如果遇到了需要删除的数字,只有src++。

#define _CRT_SECURE_NO_WARNINGS 1
#include <stdio.h>
int removeElement(int* nums, int numsSize, int val) {
    int src = 0;
    int dst = 0;
    while (src < numsSize)
    {
        if (nums[src] != val)
        {
            nums[dst] = nums[src];
            src++;
            dst++;
        }
        else
        {
            src++;
        }
    }
    return dst;

}

int main()
{
	int i = 0;
	int nums[8] = {0, 1, 2, 2, 3, 0, 4, 2};
	int val = 2;
	int tmp = removeElement(nums, 8, 2);
	printf("%d\n", tmp);
    for (i = 0; i < tmp; i++)
    {
        printf("%d ", nums[i]);
    }
	return 0;
}

运行结果如图所示
在这里插入图片描述

总结

双指针解法时间复杂度和空间复杂度都是最优,是一种非常巧妙的思路。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值