leetcode-按既定顺序创建目标数组

 题目是LeetCode第181场周赛的第一题,链接:按既定顺序创建目标数组。具体描述为:给你两个整数数组 nums 和 index。你需要按照以下规则创建目标数组:

  • 目标数组 target 最初为空。
  • 按从左到右的顺序依次读取 nums[i] 和 index[i],在 target 数组中的下标 index[i] 处插入值 nums[i] 。
  • 重复上一步,直到在 nums 和 index 中都没有要读取的元素。

 请你返回目标数组。题目保证数字插入位置总是存在。

 示例1:

输入:nums = [0,1,2,3,4], index = [0,1,2,2,1]
输出:[0,4,1,3,2]
解释:
nums       index     target
0            0        [0]
1            1        [0,1]
2            2        [0,1,2]
3            2        [0,1,3,2]
4            1        [0,4,1,3,2]

 示例2:

输入:nums = [0,1,2,3,4], index = [0,1,2,2,1]
输出:[0,4,1,3,2]
解释:
nums       index     target
0            0        [0]
1            1        [0,1]
2            2        [0,1,2]
3            2        [0,1,3,2]
4            1        [0,4,1,3,2]

 示例3:

输入:nums = [0,1,2,3,4], index = [0,1,2,2,1]
输出:[0,4,1,3,2]
解释:
nums       index     target
0            0        [0]
1            1        [0,1]
2            2        [0,1,2]
3            2        [0,1,3,2]
4            1        [0,4,1,3,2]

 这道题是easy难度的,所以可以直接利用list的add(index, value)操作完成,这么做的时间复杂度是 O ( n 2 ) O(n^{2}) ,因为平均每次插入需要 O ( n ) O(n) ,而一共需要插入n次。

 JAVA版代码如下:

class Solution {
    public int[] createTargetArray(int[] nums, int[] index) {
        int L = index.length;
        List<Integer> lst = new LinkedList<Integer>();
        for (int i = 0; i < L; ++i) {
            lst.add(index[i], nums[i]);
        }
        int[] result = new int[L];
        for (int i = 0; i < L; ++i) {
            result[i] = lst.get(i);
        }
        return result;
    }
}

 提交结果如下:


 结果当然不咋滴,因为还是会频繁涉及到链表的节点插入什么的,所以这里又可以有另一个想法,不使用链表了,直接从index数组出发,得到每个数实际在最终数组中的索引。首先可以给定一个result数组存放最终结果,如果我们从后往前看index数组,会发现对于一个index[i],它最终的真实位置就是在result数组中现在还没有填充数的所有位置中的第index[i]个。有了这个发现,实现起来也就很容易了,当然时间复杂度还是 O ( n 2 ) O(n^{2}) ,但可以省去频繁插入数值的操作,而只需要遍历数组。

 JAVA版代码如下:

class Solution {
    public int[] createTargetArray(int[] nums, int[] index) {
        int L = index.length;
        int[] result = new int[L];
        boolean[] fill = new boolean[L];
        for (int i = L - 1; i >= 0; --i) {
            int count = -1;
            for (int j = 0; j < L; ++j) {
                if (!fill[j]) {
                    if (++count == index[i]) {
                        fill[j] = true;
                        result[j] = nums[i];
                        break;
                    }
                }
            }
        }
        return result;
    }
}

 提交结果如下:


发布了68 篇原创文章 · 获赞 9 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/JR_Chan/article/details/105095600