使用另一个数组对数组进行排序

Question

我正在尝试解决这个 LeetCode 问题 - https://leetcode.com/problems/create-target-array-in-the-given-order/description/其中说明包含 -

给定两个 arrays 的整数 nums 和 index。 您的任务是根据以下规则创建目标数组：

最初目标数组是空的。 从左到右读取 nums[i] 和 index[i]，在索引 index[i] 处插入目标数组中的值 nums[i]。 重复上一步，直到nums和index中没有要读取的元素。 返回目标数组。

保证插入操作是有效的。

我尝试了这种方法，但它没有按预期工作。

public class LeetCode1389 {
    public static void main(String[] args) {
        System.out.println(Arrays.toString(createTargetArray(new int[]{0,1,2,3,4}, new int[]{0,1,2,2,1})));
    }

    static int[] createTargetArray(int[] nums, int[] index) {
        int[] target = new int[nums.length];
        for (int i = 0; i < nums.length; i++){
            for (int j = 0; j < index.length; j++){
                target[index[i]] = nums[i];
            }
        }
        return target;
    }
}

Answer 1

请改用List ，它提供了在特定索引处插入的add方法。 最后将List转换为 int 数组。

static int[] createTargetArray(int[] nums, int[] index) {
    List<Integer> target = new ArrayList<>(nums.length);
    for (int i = 0; i < nums.length; i++) target.add(index[i], nums[i]);
    return target.stream().mapToInt(i -> i).toArray();
}