Given an array of integers nums and an integer target, return the indices of the two distinct elements whose values add up to target. Return the answer as a list of two indices in any order. You may assume exactly one valid pair exists.
def two_sum(nums, target):