Your question is Search Insert in Sorted Arrays. Take a moment with it on the right.
Talk me through your thinking if you like. When you're confident, submit your answer and I'll grade it like a real screen (7/10 or better passes).
Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be inserted in order.