Code Fights - extractEachKth

README.md

Given array of integers, remove each kth element from it.

Example

For inputArray = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] and k = 3, the output should be
extractEachKth(inputArray, k) = [1, 2, 4, 5, 7, 8, 10].

Input/Output

• [time limit] 4000ms (py3)

• [input] array.integer inputArray

  Guaranteed constraints:

  5 ≤ inputArray.length ≤ 15,
  -20 ≤ inputArray[i] ≤ 20.
  

• [input] integer k

  Guaranteed constraints:

  1 ≤ k ≤ 10.
  

• [output] array.integer

  inputArray without elements k - 1, 2k - 1, 3k - 1 etc.

extract_each_kth.py

def extract_each_kth(input_array: list, k: int):
    result = []
    for i, e in enumerate(input_array):
        if (i + 1) % k != 0:
            result.append(e)
    return result

test_extract_each_kth.py

import unittest
from extract_each_kth import extract_each_kth


class TestExtractEachKth(unittest.TestCase):
    def test_1(self):
        input_array = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
        k = 3
        solution = [1, 2, 4, 5, 7, 8, 10]
        result = extract_each_kth(input_array, k)
        self.assertEqual(result, solution)

    def test_2(self):
        input_array = [1, 1, 1, 1, 1]
        k = 1
        solution = []
        result = extract_each_kth(input_array, k)
        self.assertEqual(result, solution)

    def test_3(self):
        input_array = [1, 2, 1, 2, 1, 2, 1, 2]
        k = 2
        solution = [1, 1, 1, 1]
        result = extract_each_kth(input_array, k)
        self.assertEqual(result, solution)


if __name__ == '__main__':
    unittest.main()