第K极值（python）

解题思路:

注意事项:

参考代码:

def is_prime(num):
   if num <= 1:
       return False

   for i in range(2, int(num ** 0.5) + 1):
       if num % i == 0:
           return False

   return True


n, k = map(int, input('n, k:').split())
nums = list(map(int, input('nums:').split()))
nums.sort()
num = nums[n - k] - nums[k - 1]

if is_prime(num):
   print('YES')
   print(num)
else:
   print('NO')
   print(num)