SRM408 Div2 Medium

500

読むだけ

view plainprint?

1. class OlympicCandles {  
2. public:  
3.   int numberOfNights(vector <int> C)  
4.   {  
5.     int night = 1;  
6.   
7.     while (true) {  
8.       sort(C.begin(), C.end());  
9.       reverse(C.begin(), C.end());  
10.       int cnt = 0;  
11.       for (int i = 0; i < (int)min((int)C.size(), night); ++i) {  
12.         if (0 < C[i]) ++cnt;  
13.         --C[i];  
14.       }  
15.       if (cnt == night) ++night;  
16.       else break;  
17.     }  
18.   
19.     return night - 1;  
20.   }  
21. };  

class OlympicCandles {
public:
  int numberOfNights(vector <int> C)
  {
    int night = 1;

    while (true) {
      sort(C.begin(), C.end());
      reverse(C.begin(), C.end());
      int cnt = 0;
      for (int i = 0; i < (int)min((int)C.size(), night); ++i) {
        if (0 < C[i]) ++cnt;
        --C[i];
      }
      if (cnt == night) ++night;
      else break;
    }

    return night - 1;
  }
};