牛客编程巅峰赛S1第7场 - 青铜&白银 A.模拟

链接:https://ac.nowcoder.com/acm/contest/6630/A
来源:牛客网

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

示例1
输入
复制
[1,2],1,2
输出
复制
[2,1]
备注:
,1≤n,m≤∣a∣,1≤ai≤1e3

 bool cmp(int a,int b){
        return a>b;
    }
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
        vector<int>b=a;
        int p1,p2;
        sort(b.begin(),b.end(),cmp);
        for(int i=0;i<a.size();i++){
             if(a[i]==b[n-1]){
                 p1=i;
             }
            if(a[i]==b[m-1]){
                p2=i;
            }
        }
        swap(a[p1],a[p2]);
        return a;
    }
};

你可能感兴趣的:(牛客)