题解 | 小苯送礼物

#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;

int main() {
    int n, k;
    cin >> n >> k;
    // [
    //     [点赞, 收藏, 粉丝编号]
    //     ....
    // ]
    vector<vector<int>> nums(n + 1, vector<int> (3));

    for (int i = 0; i < n; ++i) {
        cin >> nums[i][0] >> nums[i][1];
        nums[i][2] = i+1;
    }

    sort(nums.begin(), nums.end(),
    [](const auto & a, const auto & b) {

        //收藏点赞相同，按照粉丝编号升序
        if (a[0] == b[0] && a[1] == b[1]) {
            return a[2] < b[2];
        //如不相同，但是支持力度相同，按照收藏降序
        } else if (a[0] + a[1] * 2 == b[0] + b[1] * 2) {
            return a[1] > b[1];
        
        //支持力度也不同，按照支持力度排序
        } else {
            return a[0] + a[1] * 2 > b[0] + b[1] * 2;
        }
        ;
    }
        );

    //取前k个收集进ans数组里，再在ans数组里升序排序输出。
    vector<int> ans(k);
    for (int i = 0; i<k; ++i)
    {
        ans[i] = nums[i][2];
    }
    sort(ans.begin(),ans.end());
    for (int i=0; i<k;++i)
    {
        cout << ans[i] << " ";
    }

}
// 64 位输出请用 printf("%lld")