Advertisement
Josif_tepe

Untitled

Feb 5th, 2023
887
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.06 KB | None | 0 0
  1. #include <iostream>
  2. #include <algorithm>
  3. #include <vector>
  4. #include <cmath>
  5. #include <set>
  6. #include <array>
  7. using namespace std;
  8. const int maxn = 2e5 + 10;
  9.  
  10. multiset<int> smaller_values, greater_values;
  11. int k;
  12. void insert_value(int x) {
  13.     int max_in_smaller = *smaller_values.rbegin();
  14.    
  15.     if(max_in_smaller < x) {
  16.         greater_values.insert(x);
  17.        
  18.         if(greater_values.size() > k / 2) {
  19.             int min_in_greater = *greater_values.begin();
  20.             smaller_values.insert(min_in_greater);
  21.             greater_values.erase(greater_values.find(min_in_greater));
  22.         }
  23.     }
  24.     else {
  25.         smaller_values.insert(x);
  26.        
  27.         if(smaller_values.size() > (k + 1) / 2) {
  28.             int tmp_max_in_smaller = *smaller_values.rbegin();
  29.             greater_values.insert(tmp_max_in_smaller);
  30.             smaller_values.erase(smaller_values.find(tmp_max_in_smaller));
  31.         }
  32.     }
  33.    
  34. }
  35. void remove_value(int x) {
  36.     if(greater_values.find(x) != greater_values.end()) {
  37.         greater_values.erase(greater_values.find(x));
  38.     }
  39.     else if(smaller_values.find(x) != smaller_values.end()) {
  40.         smaller_values.erase(smaller_values.find(x));
  41.     }
  42.    
  43.     if(smaller_values.empty()) {
  44.         smaller_values.insert(*greater_values.begin());
  45.         greater_values.erase(greater_values.begin());
  46.     }
  47.    
  48. }
  49. int main() {
  50.     ios_base::sync_with_stdio(false);
  51.  
  52.     int n;
  53.     cin >> n >> k;
  54.    
  55.     vector<int> v(n);
  56.     for(int i = 0; i < n; i++) {
  57.         cin >> v[i];
  58.     }
  59.    
  60.     smaller_values.insert(v[0]);
  61.    
  62.     for(int i = 1; i < k; i++) {
  63.         insert_value(v[i]);
  64.     }
  65.    
  66.     cout << *smaller_values.rbegin() << " ";
  67.    
  68.     for(int i = k; i < n; i++) {
  69.         if(k > 1) {
  70.             remove_value(v[i - k]);
  71.             insert_value(v[i]);
  72.         }
  73.         else {
  74.             insert_value(v[i]);
  75.             remove_value(v[i - 1]);
  76.         }
  77.        
  78.         cout << *smaller_values.rbegin() << " ";
  79.     }
  80.    
  81.     return 0;
  82. }
  83. // 5, 1, 2, 3
  84. // 1, 2, 3, 5
  85.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement