天池 在线编程 排名查询

示例：
输入:
scores: [[90, 80, 70], [90, 90, 90], [60, 60, 60]]
k: 2
输出: 0
解释:
总成绩中排名第二的是索引为0的学生

class Solution {
public:
    /**
     * @param scores: two dimensional array
     * @param K: an integer
     * @return: return a integer
     */
    int FindTheRank(vector<vector<int>> &scores, int K) {
        // write your code here
        vector<int> id(scores.size()), allsum(scores.size());
        iota(id.begin(), id.end(), 0);
        for(int i = 0; i < scores.size(); ++i)
        {
            int s = 0;
            for(int j = 0; j < scores[i].size(); ++j)
            {
                s += scores[i][j];
            }
            allsum[i] = s;
        }
        stable_sort(id.begin(), id.end(),[&](int a, int b){
            return allsum[a] > allsum[b];
        });
        return id[K-1];
    }
};