【牛客编程巅峰赛S1第7场】数组元素交换

题目

给定一个没有重复元素的数组 a，将数组内第 n 大的数字和第 m 大的数（从大到小排序）交换位置后返回数组。

解题思路

因为题目已说明数组 a 没有重复元素，所以可以使用 map 数据结构记录每个元素的下标，同时将数组中的元素按照从小到大排序。
遍历 mp 中的元素，使用 k 表示它是第 k 大的元素，如果 k 与 n 或 m 相等，记录它们的下标，最后交换元素，返回数组 a。

C++代码

class Solution {
public:
    /**
     * 
     * @param a int整型vector 原始数组a
     * @param n int整型 第n大
     * @param m int整型 第m大
     * @return int整型vector
     */
    vector<int> sovle(vector<int>& a, int n, int m) {
        // write code here
        map<int,int> mp;
        for(int i=0; i<a.size(); ++i)
            mp[a[i]] = i;
        int k = a.size();
        int ni=0, mi=0;
        for(auto it=mp.begin(); it!=mp.end(); ++it){
            if(k==n)
                ni = it->second;
            if(k==m)
                mi = it->second;
            --k;
        }
        swap(a[ni], a[mi]);
        return a;
    }
};