1857E - Power of Points - CodeForces Solution

#include<bits/stdc++.h>
using namespace std;
#define ll long long

int main(){
   ll n,m,a,b,x,z,p,y,c,d;
   cin>>p;
   string S;
   for(ll t=0;t<p;t++){
      cin>>n;
      vector<ll> vect;
      int arr[n];
      map<ll,ll> mp1;
      ll s=0;
      for(ll k=0;k<n;k++){
          cin>>a;
          vect.push_back(a);
          s+=a;
          mp1[a]++;
          arr[k]=a;
      }
      m=vect.size();
      sort(vect.begin(),vect.end());
      mp1.clear();
      map<ll,ll> mp;
      y=0;
      for(ll k=0;k<m;k++) {mp1[vect[k]]=2*k-m+1;mp[vect[k]]=y;y+=vect[k];}
      vector<ll> ans;
      for(ll k=0;k<n;k++){
          //cout<<arr[k]<<" "<<mp1[arr[k]]<<" "<<s-arr[k]<<" "<<mp[arr[k]]<<endl;
          ans.push_back(mp1[arr[k]]*arr[k]+s-arr[k]-2*mp[arr[k]]+n);
          cout<<ans[k]<<" ";
      }
      cout<<endl;
   }
}