Find K Closest Elements - Leetcode 658 - Python