I got this question Yesterday in a online Assesment but not Able to get AC. Is there any way to improve this code?

Question

Given an array of size N and a number K. You need to Partition the array into K non-empty partitions such that the sum of (max-min) in each partition is maximized.

Le