Amazon Robotics Interview Question

A variation of the leetcode question two sum