offer2-75


给定两个数组,arr1arr2

  • 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]

提示:

  • 1 <= arr1.length, arr2.length <= 1000
  • 0 <= arr1[i], arr2[i] <= 1000
  • arr2 中的元素 arr2[i] 各不相同
  • arr2 中的每个元素 arr2[i] 都出现在 arr1
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        List<Integer> arr1List = new ArrayList<>();
        for (int v : arr1) {
            arr1List.add(v);
        }

        arr1List.sort(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                int index1 = Integer.MAX_VALUE / 2;
                int index2 = Integer.MAX_VALUE / 2;
                for (int i = 0; i < arr2.length; i++) {
                    if (arr2[i] == o1) {
                        index1 = i;
                        break;
                    }
                }
                for (int i = 0; i < arr2.length; i++) {
                    if (arr2[i] == o2) {
                        index2 = i;
                        break;
                    }
                }

                if (index1 == index2) {
                    return o1 - o2;
                } else {
                    return index1 - index2;
                }
            }
        });

        int[] result = new int[arr1List.size()];
        int i = 0;
        for (Integer v : arr1List) {
            result[i++] = v;
        }
        return result;
    }
}
//leetcode submit region end(Prohibit modification and deletion)

文章作者: 倪春恩
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 倪春恩 !