1122. 数组的相对排序

2020-01-07  本文已影响0人  周英杰Anita

题目描述:

给你两个数组,arr1 和 arr2,

arr2 中的元素各不相同
arr2 中的每个元素都出现在 arr1 中
对 arr1 中的元素进行排序,使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末尾。

示例:

输入:arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
输出:[2,2,2,1,4,3,3,9,6,7,19]
 

提示:

arr1.length, arr2.length <= 1000
0 <= arr1[i], arr2[i] <= 1000
arr2 中的元素 arr2[i] 各不相同
arr2 中的每个元素 arr2[i] 都出现在 arr1 中

思路:

描述有点混乱,先这样吧
1. 先创建一个大数组temp,和最终返回数组ans,记录ans数组索引值;
2. 第一步,遍历数组1,以数组元素为temp的索引值,该索引的值代表数组元素出现的次数。
3. 第二步,遍历数组2,以数组元素的值为索引,获取该索引对应的元素数量,并设置ans的值。
4. 第三步,遍历数组temp,取出数组中所有值不为零的元素,根据索引值设置ans的值。
5. 最后,返回数组ans;

Java解法:

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int len1 = arr1.length;
        int len2 = arr2.length;
        int[] temp = new int[1001];
        int[] ans = new int[len1];
        int index = 0;
        for(int i = 0; i < len1; i++)
        {
            temp[arr1[i]]++;
        }
        for(int i = 0; i < len2; i++)
        {
            while(temp[arr2[i]] > 0)
            {
                ans[index++] = arr2[i];
                temp[arr2[i]]--;
            }
        }
        for(int i = 0; i < 1001; i++)
        {
            while(temp[i] > 0)
            {
                ans[index++] = i;
                temp[i]--;
            }
        }
        return ans;
    }
}

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers

上一篇下一篇

猜你喜欢

热点阅读