这两天事有点多,都没时间学习,再加上神经网络调参不太顺利,还没上火简直是个奇迹。晚上查完作业实在静不下来,做个简单题吧。
做过CPP的作业之后觉得LeetCode真的友好,哪怕不会做,人家的题目描述至少是基本准确的,还有例子和输入输出;CPP那个作业是什么东西。
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。 元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
这个题目描述实在是让人熟悉,显然是用上次那个双指针法,把不需要删除的元素移到前面就行,只需要遍历一次就行:
/**
* ------result------
* memory: 9.7 MB (75%)
* speed: 4 ms (90%)
*/
int removeElement(vector<int> &nums, int val)
{
int length = nums.size();
if (length == 0)
{
return 0;
}
int i = 0;
for (auto j = 0; j < length; ++j)
{
if (nums[j] != val)
{
nums[i] = nums[j];
++i;
}
}
return i;
}
但是不知道为啥结果不是很好。我以为是还有什么奇技淫巧,然后看了题解发现,人家比我少几行代码:
/**
* ------result------
* memory: 8.5 MB (90%)
* speed: 4 ms (90%)
*/
int removeElement(vector<int> &nums, int val)
{
int length = nums.size();
int i = 0;
for (auto j = 0; j < length; ++j)
{
if (nums[j] != val)
{
nums[i] = nums[j];
++i;
}
}
return i;
}
代码也会占空间,那个长度为0的判断显然是多余的。所以,有时候把特殊值放在前面提前判断也未必就是好事,得看有没有用。
感觉简单题是那种一眼看上去就能想到解法,但想要提升还是很麻烦的。