DIY: Find K Closest Elements

Solve the interview question "Find K Closest Elements" in this lesson.

Problem statement

Given a sorted integer array arr, and two integers k and x, return the k closest integers to x in this array. You must ensure that the result is sorted in ascending order.

An ID a is closer to num than an ID b, if |a - num| <= |b - num| and a < b.

Input

The input will be an array of integers and two integers. The following are examples of input to the function:

# Sample Input 1
arr = [1, 2, 3, 4, 5], k = 4, x = 3

# Sample Input 2
arr = [1, 2, 3, 4, 5], k = 4, x = -1

# Sample Input 3
arr = [-29, -11, -3, 0, 5, 10, 50, 63, 198], k = 6, x = 8

Output

The output will be a list of integers containing k closest integers to x. The following are examples of the outputs:

# Sample Output 1
[1, 2, 3, 4] 

# Sample Output 2
[1, 2, 3, 4]

# Sample Output 3
[-29, -11, -3, 0, 5, 10]

Level up your interview prep. Join Educative to access 80+ hands-on prep courses.