给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

100 阅读 0 评论 66 点赞

我是靠谱客的博主喜悦小懒猪，这篇文章主要介绍给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。，现在分享给大家，希望可以做个参考。

1.给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。
不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。

2.代码展示

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
int removeElement(int* nums, int numsSize, int val)
{
    int src =0;
    int dst =0;
    while (dst<numsSize)
    {
        if ( nums[dst] != val)
        {
            nums[src] = nums[dst];
            ++dst;
            ++src;
        }
        else
        {
            ++dst;
        }
    }
    return src;
}

3.解题思路
双指针，在原空间下存储新的数组，相同就跳过这个数字，不相同就赋值。
非常简单的。
让我比较费解的是下面的代码不能通过，不过在vs上可以通过，我这里写的太麻烦了。

复制代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#define _CRT_SECURE_NO_WARNINGS 1
#include<stdio.h>
int removeElement2(int* nums, int numsSize, int val)
{

	int count = 0;
	for (int i = 0; i < numsSize; i++)
	{
		if (nums[i] == val)
		{
			count++;
		}
	}
	int* dst = nums;
	int* src = nums;
	while (nums <= (dst + numsSize-1))
	{
		src = nums;
		if (*nums == val)
		{
			while (*nums == val && nums <= (dst + numsSize-1))
			{
				nums++;
			}
			if (nums >( dst + numsSize-1))
			{
				break;
			}
			int tmp = *nums;
			*nums = *src;
			*src = tmp;
		}
		src++;
		nums = src;
	}
	nums = dst;
	return numsSize-count;
}
int main() 
{
	int arr[] = {1,2,2,1,3,1};
	int val = 1;
	int ret = removeElement2(arr, 6, 1);
	printf("%dn",ret);
	for (int i = 0; i < 6; i++)
	{
		printf("%d", arr[i]);
	}
	system("pause");
	return 0;

}