SRM 520 Div2 Easy SRMRoomAssignmentPhase

K個の部屋があり,スコアの高い順番に部屋へ割り当てていく.
自分と同じ部屋でスコアの高い人の数を答えよという問題.

ソートして,スコアの高いものから順番に部屋へ格納.
自分のスコアが来たらその時点の部屋のメンバーの数を返す.

class SRMRoomAssignmentPhase
{
public:
  int countCompetitors(vector <int> ratings, int K)
    {
      int me = ratings[0];
      int n = ratings.size();
      sort(ratings.begin(), ratings.end());
      reverse(ratings.begin(), ratings.end());
      vector<vector<int> > room(K);
      int idx = 0;
      for(int i = 0; i < n; i++){
        if (ratings[i] == me) return room[idx].size();
        room[idx].push_back(ratings[i]);
        idx = (idx+1) % K;
      }
      return -1;
    }
};