来源:力扣(LeetCode)
描述:
给你一个整数数组 arr
,请你将数组中的每个元素替换为它们排序后的序号。
序号代表了一个元素有多大。序号编号的规则如下:
- 序号从 1 开始编号。
- 一个元素越大,那么序号越大。如果两个元素相等,那么它们的序号相同。
- 每个数字的序号都应该尽可能地小。
示例 1:
输入:arr = [40,10,20,30]
输出:[4,1,2,3]
解释:40 是最大的元素。 10 是最小的元素。 20 是第二小的数字。 30 是第三小的数字。
示例 2:
输入:arr = [100,100,100]
输出:[1,1,1]
解释:所有元素有相同的序号。
示例 3:
输入:arr = [37,12,28,9,100,56,80,5,12]
输出:[5,3,4,2,8,6,7,1,3]
提示:
- 0 <= arr.length <= 105
- -109 <= arr[i] <= 109
方法:排序 + 哈希表
思路
首先用一个数组保存排序完的原数组,然后用一个哈希表保存各元素的序号,最后将原属组的元素替换为序号后返回。
代码:
class Solution {
public:
vector<int> arrayRankTransform(vector<int>& arr) {
vector<int> sortedArr = arr;
sort(sortedArr.begin(), sortedArr.end());
unordered_map<int, int> ranks;
vector<int> ans(arr.size());
for (auto &a : sortedArr) {
if (!ranks.count(a)) {
ranks[a] = ranks.size() + 1;
}
}
for (int i = 0; i < arr.size(); i++) {
ans[i] = ranks[arr[i]];
}
return ans;
}
};
执行用时:100 ms, 在所有 C++ 提交中击败了39.83%的用户
内存消耗:38.3 MB,在所有 C++ 提交中击败了46.48%的用户
复杂度分析
时间复杂度:O(n × logn),其中 n 是输入数组 arr 的长度,排序消耗O(n × logn) 时间。
空间复杂度:O(n)。有序数组和哈希表各消耗O(n) 空间。
arthor:LeetCode-Solution