Advertisement
Guest User

task A

a guest
Feb 21st, 2020
115
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.81 KB | None | 0 0
  1. #include <iostream>
  2. #include <fstream>
  3. #include <vector>
  4. #include <algorithm>
  5.  
  6. using namespace std;
  7.  
  8. int compare(pair<int, int> p1, pair<int, int> p2)  
  9. {
  10.     return (p1.first > p2.first);
  11. }
  12.  
  13. int main()
  14. {
  15.     int n, m, k;
  16.     cin >> n >> m >> k;
  17.     vector<pair<int, int>> input(n);
  18.    
  19.     for (int i = 0; i < n; i++) {
  20.         cin >> input[i].first;
  21.         input[i].second = i + 1;
  22.     }
  23.  
  24.     sort(input.begin(), input.end(), compare);
  25.  
  26.     vector<vector<int>> output(k, vector<int>(0));
  27.     int i = 0;
  28.     while (input.size() > 0) {
  29.         int element = input.back().second;
  30.         input.pop_back();
  31.         output[i].push_back(element);
  32.         i += 1;
  33.         if (i == k) i = 0;
  34.     }
  35.  
  36.     for (int i = 0; i < k; i++) {
  37.         cout << output[i].size() << " ";
  38.         for (int j = 0; j < output[i].size(); j++) {
  39.             cout << output[i][j] << " ";
  40.         }
  41.         cout << endl;
  42.     }
  43. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement