Ankit_132

E

Aug 7th, 2023
469
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.92 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. #define ll     long long
  6. #define _test   int _TEST; cin>>_TEST; while(_TEST--)
  7. #define pb     push_back
  8. #define ff     first
  9. #define ss     second
  10.  
  11. int main()
  12. {
  13.     _test
  14.     {
  15.         ll int n;
  16.         cin>>n;
  17.  
  18.         vector<ll int> x(n);
  19.         for(auto &e: x)
  20.             cin>>e;
  21.  
  22.         vector<pair<ll int, int>> vp;
  23.  
  24.         for(int i=0; i<n; i++)
  25.             vp.pb({x[i], i});
  26.  
  27.         sort(vp.begin(), vp.end());
  28.  
  29.         ll int val = 0;
  30.  
  31.         for(int i=0; i<n; i++)
  32.             val += (vp[i].ff - vp[0].ff + 1);
  33.  
  34.         vector<ll int> ans(n);
  35.         ans[vp[0].ss] = val;
  36.  
  37.         for(ll int i=1; i<n; i++)
  38.         {
  39.             val -= (n-i)*(vp[i].ff-vp[i-1].ff);
  40.             val += i*(vp[i].ff-vp[i-1].ff);
  41.  
  42.             ans[vp[i].ss] = val;
  43.         }
  44.  
  45.         for(auto e: ans)
  46.             cout<<e<<" ";
  47.         cout<<"\n";
  48.     }
  49. }
Advertisement
Add Comment
Please, Sign In to add comment