Advertisement
Josif_tepe

Untitled

Mar 20th, 2024
467
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.96 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <cmath>
  4. #include <set>
  5. using namespace std;
  6.  
  7. long long sumi[100001];
  8. int main() {
  9.     int n, p;
  10.     cin >> n >> p;
  11.     vector<int> v;
  12.     multiset<int> ms;
  13.     long long sum = 0;
  14.     for(int i = 0; i < n; i++) {
  15.         int x;
  16.         cin >> x;
  17.         v.push_back(x);
  18.         ms.insert(x);
  19.         sum += x;
  20.     }
  21.     sumi[0] = sum;
  22.    
  23.     for(int i = 1; i < n; i++) {
  24.         int najmal_element = *ms.begin();
  25.         int najgolem_element = *ms.rbegin();
  26.         int razlika = najgolem_element - najmal_element;
  27.        
  28.         ms.erase(ms.find(najmal_element));
  29.         ms.erase(ms.find(najgolem_element));
  30.         ms.insert(razlika);
  31.        
  32.         sum -= (najmal_element + najgolem_element);
  33.         sum += razlika;
  34.         sumi[i] = sum;
  35.     }
  36.    
  37.     for(int i = 0; i < p; i++) {
  38.         int x;
  39.         cin >> x;
  40.         cout << sumi[x] << endl;
  41.     }
  42.    
  43.     return 0;
  44. }
  45. //  1
  46.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement