Ankit_132

E

Sep 2nd, 2023
334
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.61 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define ll     long long
  5. #define pb     push_back
  6.  
  7. int main()
  8. {
  9.     int n;
  10.     cin>>n;
  11.  
  12.     vector<int> a(n);
  13.     for(auto &e: a)   cin>>e;
  14.  
  15.     vector<vector<ll int>> pos(n+1);
  16.     for(int i=0; i<n; i++)
  17.         pos[a[i]].pb(i);
  18.  
  19.     ll int ans = 0;
  20.  
  21.     for(int i=1; i<=n; i++)
  22.     {
  23.         if(!pos[i].size())      continue;
  24.  
  25.         ll int prev = pos[i][0];
  26.  
  27.         for(int j=1; j<pos[i].size(); j++)
  28.         {
  29.             ans += j*1ll*pos[i][j] - prev - j*1ll;
  30.             prev += pos[i][j] + j*1ll;
  31.         }
  32.     }
  33.  
  34.     cout<<ans<<"\n";
  35. }
  36.  
Advertisement
Add Comment
Please, Sign In to add comment