[golang] 27. Remove Element

// [url]https://leetcode.com/studyplan/top-interview-150/[/url]
 
package topInterview
 
// removeElement
// 27. Remove element
// Given an integer array nums and an integer val, remove all occurrences of val in nums in-place. The order of the elements may be changed. Then return the number of elements in nums which are not equal to val.
//
// Consider the number of elements in nums which are not equal to val be k, to get accepted, you need to do the following things:
//
// Change the array nums such that the first k elements of nums contain the elements which are not equal to val. The remaining elements of nums are not important as well as the size of nums.
// Return k.
// Custom Judge:
//
// The judge will test your solution with the following code:
//
// int[] nums = [...]; // Input array
// int val = ...; // Value to remove
// int[] expectedNums = [...]; // The expected answer with correct length.
//
//  // It is sorted with no values equaling val.
//
// int k = removeElement(nums, val); // Calls your implementation
//
// assert k == expectedNums.length;
// sort(nums, 0, k); // Sort the first k elements of nums
//
//  for (int i = 0; i < actualLength; i++) {
//      assert nums[i] == expectedNums[i];
//  }
//
// If all assertions pass, then your solution will be accepted.
//
// Example 1:
//
// Input: nums = [3,2,2,3], val = 3
// Output: 2, nums = [2,2,_,_]
// Explanation: Your function should return k = 2, with the first two elements of nums being 2.
// It does not matter what you leave beyond the returned k (hence they are underscores).
// Example 2:
//
// Input: nums = [0,1,2,2,3,0,4,2], val = 2
// Output: 5, nums = [0,1,4,0,3,_,_,_]
// Explanation: Your function should return k = 5, with the first five elements of nums containing 0, 0, 1, 3, and 4.
// Note that the five elements can be returned in any order.
// It does not matter what you leave beyond the returned k (hence they are underscores).
//
// Constraints:
//
// 0 <= nums.length <= 100
// 0 <= nums[i] <= 50
// 0 <= val <= 100
// Тут задача избавится от всех элементов в слайсе nums которые равны переменной val
// изменить данные в слайсе и вернуть количество элементов не равных val
func removeElement(nums []int, val int) int {
 
    // Объявить переменную в которой будет индекс крайнего вставленного элемента
    insertIndex := 0
 
    // Инициировать цикл for range по слайсу
    for _, num := range nums {
        // если текущий элемент слайса nums не равен проверяемому значению
        if num != val {
            // заменить элемент в слайсе nums c индексом крайнего вставленного элемента.
            nums[insertIndex] = num
            // Увеличить индекс вставленного элемента на один
            insertIndex++
        }
    }
 
    // Вернуть индекс
    return insertIndex
}

func TestRemoveElement(t *testing.T) {
 
    data := []struct {
        nums     []int
        val      int
        expected struct {
            result    int
            numsAfter []int
        }
    }{
        {
            []int{3, 2, 2, 3},
            3,
            struct {
                result    int
                numsAfter []int
            }{
                2,
                []int{2, 2},
            },
        },
        {
            []int{0, 1, 2, 2, 3, 0, 4, 2},
            2,
            struct {
                result    int
                numsAfter []int
            }{
                5,
                []int{0, 1, 3, 0, 4},
            },
        },
    }
 
    for i, datum := range data {
        result := removeElement(datum.nums, datum.val)
 
        if datum.expected.result != result || !reflect.DeepEqual(datum.nums[:len(datum.expected.numsAfter)], datum.expected.numsAfter) {
            t.Errorf("unexpected result for test index %d expected %+v got [%+v, %+v ]", i, datum.expected, result, datum.nums[:len(datum.expected.numsAfter)])
        }
    }
}