Advertisement
Josif_tepe

Untitled

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