题目描述
给你一个整数数组 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) {
//(1)特殊情况处理
vector<int> res;
int size = arr.size();
if (size == 0)
return res;
//(2)排序
vector<int> arr_temp = arr;
sort(arr_temp.begin(), arr_temp.end());
//(3)建立值与序号的映射
unordered_map<int, int> mymap;
mymap[arr_temp[0]] = 1;
for (int i = 1,j=1; i < size; i++)
{
if (arr_temp[i] == arr_temp[i - 1])
continue;
j++;
mymap[arr_temp[i]] = j;
}
//(4)遍历arr,读取对应序号
for (int i = 0; i < size; i++)
res.push_back(mymap[arr[i]]);
return res;
}
};
思路
(1)特殊情况处理,输入数组为空时,返回空数组;
(2)复制arr给arr_temp,对arr_temp排序;
(3)将arr_temp的值作为哈希表mymap的键,将序号作为mymap的值;
(4)遍历arr,将arr中的值作为mymap的键,读取对应的序号,存于res中。
示例带入
比如,示例为arr=[20,40,10,10]
步骤(1)
:跳过
步骤(2)
:arr_temp=[10,10,20,40]
步骤(3)
:mymap[10]=1,mymap[20]=2,mymap[40]=3
步骤(4)
:res=[2,3,1,1]
执行结果
执行用时:108 ms, 在所有 C++ 提交中击败了94.58%的用户
内存消耗:38.3 MB, 在所有 C++ 提交中击败了60.79%的用户