A.数组元素交换

题目描述

牛牛有一个没有重复元素的数组a，他想要将数组内第n大的数字和第m大的数(从大到小排序)交换位置你能帮帮他吗。给定一个数组a，求交换第n大和第m大元素后的数组。

示例1
输入
[1,2],1,2
输出
[2,1]

备注

题解

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
        int index1, index2;
        vector<int> t(a);
        sort(t.begin(), t.end());

        int x = t[t.size()-n];
        int y = t[t.size()-m];
        for(int i = 0; i < a.size(); i++){
            if(a[i] == x) index1 = i;
            if(a[i] == y) index2 = i;
        }
        swap(a[index1], a[index2]);
        return a;
    }
};