Optimize the Space of Code in Coding Screen

You are given a task to optimize the space complexity of an existing code while ensuring that all test cases pass. Provide a code implementation and explain the difference in space complexity before and after your changes. Example Input 5 5 3 8 6 2