Insertion Sort
Difficulty: Easy, Duration: 10 minutes, Languages: JS, TS, Author: Kiran Dash (opens in a new tab)
Watch the live Stream on YouTube (opens in a new tab)
Question
Implement the insertion sort algorithm to sort an array of numbers in ascending order.
Example
example.ts
insertionSort([3, 1, 2]); // [1, 2, 3]
insertionSort([12, 16, 14, 1, 6, 9]); // [1, 6, 9, 12, 14, 16]
Launch exercise in editor (opens in a new tab)
Solution
Questions to ask the interviewer:
- Should I sort the array in ascending or descending order?
- Should the data be sorted in-place or should I return a new array by using additional data structures?
- What kind of data will be in the array? Will it be numbers, strings, or a mix of both?
- Will the array contain negative numbers? If yes, should I consider them while sorting?
- Does the array contain duplicate elements? If yes, should I consider them while sorting?
Explanation:
Please check the images below to understand the insertion sort algorithm step by step.
- Start from the second element (index 1) and compare it with the previous elements.
- If the current element is smaller than the previous element, swap them.
- Continue this process until the current element is greater than the previous element.
solution.ts
function insertionSort(
arr: Array<Number>,
order: "ASC" | "DESC" = "ASC"
): Array<Number> {
// Iterate through the array starting from second element
for (let i = 1; i < arr.length; i++) {
// Store the current value so that it can be swapped later
let currentValue = arr[i];
// Create another loop starting from i - 1
let j = i - 1; // pointer that points to index of element before the currentValue
// progress in backward direction till the currentValue is smaller than the previous element and swap
while (
(order === "ASC" && j >= 0 && currentValue < arr[j]) ||
(order === "DESC" && j >= 0 && currentValue > arr[j])
) {
// Move the previous element to one position right
arr[j + 1] = arr[j];
// decrement the pointer to one step further in left
j--;
}
// Complete the swap by setting the currentValue to the final position which is one position next to the pointer we created earlier
arr[j + 1] = currentValue;
}
return arr;
}
Time Complexity:
- Best Case: O(n) when the array is already sorted. In this case, we only need to compare each element with its previous element. No swaps are required.
- Average Case: O(n^2): In the average case, we need to compare each element with all the previous elements. The number of comparisons will be n(n-1)/2. The number of swaps will be n(n-1)/2. So the time complexity will be O(n^2).
- Worst Case: O(n^2): In the worst case, the array will be sorted in descending order. In this case, we need to compare each element with all the previous elements. The number of comparisons will be n(n-1)/2. The number of swaps will be n(n-1)/2. So the time complexity will be O(n^2).
Space Complexity:
- O(1): because we are sorting the array in-place. No additional data structures are used. If we return a new array, the space complexity will be O(n).
Edge Cases:
- If the array is empty, return an empty array.
- If the array input is large, consider using a different sorting algorithm like quicksort or mergesort. Because the time complexity of insertion sort is O(n^2).
- If input is not an array, throw a TypeError.
- If input array has non-integers, throw a TypeError.