Leetcode 1331. 数组序号转换

• 2022 年 7 月 16 日
• 筆記
• C/C++

给你一个整数数组 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
• -10^ 9 <= arr[i] <= 10^9

Code:
常规超时思路

class Solution {
public:
    vector<int> arrayRankTransform(vector<int>& arr) {
        if(arr.size()==0)
            return arr;
        vector<int>arrsrc=arr;
        sort(arr.begin(),arr.end());
        
        
        typedef struct
        {
            int num;
            int index;
        }param;
        
        vector<param>vec;
        vector<int>res;
        
        int cnt=1;
        param p;
        p.num=arr[0];
        p.index=cnt;
        vec.push_back(p);
        
        
        map<int,param>mymap;
        
        
        
        for(int i=1;i<arr.size();i++)
        {
            
            param p;
            p.num=arr[i];
            if(arr[i]==arr[i-1])
            {
                p.index=cnt;
            }
            else
                p.index=++cnt;
            vec.push_back(p);
            
        }
        
        
        
        for(int i=0;i<arrsrc.size();i++)
        {
            
            for(int j=0;j<vec.size();j++)
            {
                if(vec[j].num==arrsrc[i])
                {
                    res.push_back(vec[j].index);
                    break;
                }
            }
            //       res.push_back();
        }
        return res;
        
    }   
};

终极大法：利用map解决时间问题

Code:

class Solution {
public:
    vector<int> arrayRankTransform(vector<int>& arr) {
        vector<int>arrsrc=arr;
        sort(arr.begin(),arr.end());
        
        
        typedef struct
        {
            int num;
            int index;
        }param;
        
        vector<param>vec;
        vector<int>res;
        
        if(arr.size()==0)
            return res;
        
        int cnt=1;
        param p;
        p.num=arr[0];
        p.index=cnt;
        vec.push_back(p);
        map<int,param>mymap;
        mymap[arr[0]]=p;
        for(int i=1;i<arr.size();i++)
        {
            
            param p;
            p.num=arr[i];
            if(arr[i]==arr[i-1])
            {
                p.index=cnt;
            }
            else
                p.index=++cnt;
            vec.push_back(p);
            
            mymap[arr[i]]=p;
            
        }
 
        for(int i=0;i<arrsrc.size();i++)
        {
            res.push_back(mymap[arrsrc[i]].index);
        }
        return res;
        
    }   
};