算法練習題---原地刪除數組元素

https://leetcode-cn.com/problems/remove-element/

題目：給定一個數組 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。

注意這五個元素可爲任意順序。

你不須要考慮數組中超出新長度後面的元素。
說明:

爲何返回數值是整數，但輸出的答案是數組呢?

請注意，輸入數組是以「引用」方式傳遞的，這意味着在函數裏修改輸入數組對於調用者是可見的。

你能夠想象內部操做以下:

// nums 是以「引用」方式傳遞的。也就是說，不對實參做任何拷貝
int len = removeElement(nums, val);

// 在函數裏修改輸入數組對於調用者是可見的。
// 根據你的函數返回的長度, 它會打印出數組中該長度範圍內的全部元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

=================================================================================

解答方案一：

package com.zx.leetcode.removeelement;

/**
 * @Author JAY
 * @Date 2019/6/14 17:25
 * @Description 移除元素
 **/
public class SolutionV2 {

    public static void main(String[] args) {
        int[] nums = new int[]{2,2,2,2,2,2,2,2,2,2,2,0,1,2,2,3,0,4,2};
        int val = 2;
        int len = removeElement(nums, val);
        for (int i = 0; i < len; i++) {
            System.out.println(nums[i]);
        }
    }

    public static int removeElement(int[] nums, int val) {

        int valCount = 0;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == val){
                valCount++;
            }
        }
        int valCount2 = 0;

        for (int i = 0; i < nums.length; i++) {
            if (valCount2 >= valCount){
                break;
            }
            if (nums[i] == val){
                valCount2++;
                for (int j = nums.length - 1; j > i ; j--) {
                    if (nums[j] != val){
                        int a = nums[i];
                        nums[i] = nums[j];
                        nums[j] = a;
                    }
                }
            }
        }

        return nums.length - valCount;
    }
}

解答方案二：

package com.zx.leetcode.removeelement;

import java.util.Arrays;

/**
 * @Author JAY
 * @Date 2019/6/14 17:25
 * @Description 移除元素
 **/
public class Solution {

    public static void main(String[] args) {
        int[] nums = new int[]{0,1,2,2,3,0,4,2};
        int val = 2;
        int len = removeElement(nums, val);
        for (int i = 0; i < len; i++) {
            System.out.println(nums[i]);
        }
    }

    public static int removeElement(int[] nums, int val) {
        int zeroCount = 0;
        int afterZeroCount = 0;
        for (int i = 0; i < nums.length; i++) {
            if (0 == nums[i]){
                zeroCount++;
            }
        }
        for (int i = 0; i < nums.length; i++) {
            if (val == nums[i]){
                nums[i] = 0;
            }
        }

        for (int i = 0; i < nums.length - 1; i++) {
            for (int j = i + 1; j <  nums.length; j++){
              if (nums[i] <= nums[j]){
                  int a = nums[i];
                  nums[i] = nums[j];
                  nums[j] = a;
              }
            }
        }
        for (int i = 0; i < nums.length; i++) {
            if (0 == nums[i]){
                afterZeroCount++;
            }
        }

        int valCount = afterZeroCount - zeroCount;

        return nums.length - valCount;
    }
}