8221 演唱会 排序

#include <iostream>
#include <algorithm>
using namespace std;
struct Song{
    int num;//歌曲编号
    int happy;//欢乐值
};
bool cmp(Song a,Song b){
    return a.happy>b.happy;//按欢乐值从大到小排序
}
int main(){
    int n,m,b;
    cin>>n>>m>>b;
    Song songs[n];
    for (int i=0;i<n;i++){
        cin>>songs[i].happy;
        songs[i].num=i+1;
    }
    sort(songs,songs+n,cmp);//按欢乐值排序
    // 将小童最喜欢的歌曲加入歌单
    bool a=false;
    for(int i=0;i<m-1;i++){
        if (songs[i].num==b){
            a=true;
            break;
        }
    }
    if(a){
        cout<<b<<" ";
    } else {
        cout<<songs[n-1].num<<" ";
    }
    // 输出其他选中的歌曲
    for(int i=0;i<m -1; i++){
        if (songs[i].index!=b) {
            cout<<songs[i].index<<" ";
        }
    }
    return 0;
}

千千蓝鲸