Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <bits/stdc++.h>
- using namespace std;
- #define ll long long
- #define pb push_back
- int main()
- {
- int n;
- cin>>n;
- vector<int> a(n);
- for(auto &e: a) cin>>e;
- vector<vector<ll int>> pos(n+1);
- for(int i=0; i<n; i++)
- pos[a[i]].pb(i);
- ll int ans = 0;
- for(int i=1; i<=n; i++)
- {
- if(!pos[i].size()) continue;
- ll int prev = pos[i][0];
- for(int j=1; j<pos[i].size(); j++)
- {
- ans += j*1ll*pos[i][j] - prev - j*1ll;
- prev += pos[i][j] + j*1ll;
- }
- }
- cout<<ans<<"\n";
- }
Advertisement
Add Comment
Please, Sign In to add comment