Answer basic embedded questions and LeetCode easy problems.
Implement the requested function for each test case using standard Python. Input and output shapes are defined in the test cases, and solutions should use simple, efficient algorithms appropriate for easy interview problems.
def solve_problem(data):