数组相关面试题--3.原地移除数组中所有的元素val

3.原地移除数组中所有的元素val
 

27. 移除元素 - 力扣（LeetCode）

/*
    方法1
    解题思路：
        1. 从前往后遍历nums，找到val第一次出现的位置
        2. 将val之后的所有元素整体往前搬移，即删除该val
        3. nums中有效元素个数减少一个
    循环进行上述操作，直到nums中所有值为val的元素全部删除完
    时间复杂度:O(N^2)  空间复杂度:O(1)
*/

int removeElement(int* nums, int numsSize, int val) {
    // 
    while (1)
    {
        // 1. 在nums中找val出现的位置
        int pos = 0;
        for (; pos < numsSize; ++pos)
        {
            if (nums[pos] == val)
            {
                break;
            }
        }


        // 2. 检测是否找到
        if (pos == numsSize)
            break;


        // 3. 找到值为value的元素--将其删除
        for (int j = pos + 1; j < numsSize; ++j)
        {
            nums[j - 1] = nums[j];
        }


        numsSize--;
    }


    return numsSize;
}

---

 /*
    方法2：
    解题思路:
        1. 创建一个长度与nums相同的数组temp
        2. 遍历nums，将nums中所有与val不同的元素搬移到temp中
        3. 将temp中所有元素拷贝回nums中
       时间复杂度: O(N)  空间复杂度: O(N)
*/

int removeElement(int* nums, int numsSize, int val) {
    // 1. 申请numSize个元素的新空间
    int* temp = (int*)malloc(sizeof(int) * numsSize);
    if (NULL == temp)
    {
        return 0;
    }


    // 2. 将nums中非value的元素搬移到temp中---尾插到temp中
    int count = 0;
    for (int i = 0; i < numsSize; ++i)
    {
        if (nums[i] != val)
        {
            temp[count] = nums[i];
            ++count;
        }
    }

// 3. 将temp中删除val之后的所有元素拷贝到nums中
    memcpy(nums, temp, sizeof(int) * count);
    free(temp);
    return count;
}

---

/*
    方法2优化：
    解题思路:
    因为题目说了，数组中元素个数最大为100，所以不用动态申请，至二级创建100个元素数组即可
        1. 创建一个100个元素的整形数组temp
        2. 遍历nums，将nums中所有与val不同的元素搬移到temp中
        3. 将temp中所有元素拷贝回nums中
       时间复杂度: O(N)  空间复杂度: O(N)
*/

int removeElement(int* nums, int numsSize, int val) {
    // 1. 申请numSize个元素的新空间
    int temp[100];


    // 2. 将nums中非value的元素搬移到temp中---尾插到temp中
    int count = 0;
    for (int i = 0; i < numsSize; ++i)
    {
        if (nums[i] != val)
        {
            temp[count] = nums[i];
            ++count;
        }
    }


    // 3. 将temp中删除val之后的所有元素拷贝到nums中
    memcpy(nums, temp, sizeof(int) * count);
    return count;
}

---

解题思路:
        1. 设置一个变量count，用来记录nums中值等于val的元素的个数
        2. 遍历nums数组，对于每个元素进行如下操作：
            a. 如果num[i]等于val，说明值为val的元素出现了一次，count++
            b. 如果nums[i]不等于元素，将nums[i]往前搬移count个位置
                因为nums[i]元素之前出现过count个值等于val的元素，已经被删除了
                因此次数需要将nums[i]往前搬移
        3. 返回删除之后新数组中有效元素个数

    时间复杂度：O(N)   空间复杂度：O(1)
 */

int removeElement(int* nums, int numsSize, int val) {
    int count = 0;
    for (int i = 0; i < numsSize; ++i)
    {
        if (nums[i] == val)
        {
            count++;
        }
        else
        {
            nums[i - count] = nums[i];
        }
    }


    return numsSize - count;
}