C++(clang++11) 解法, 执行用时: 102ms, 内存消耗: 68600K, 提交时间: 2021-04-10 10:47:59

#include<bits/stdc++.h>
using namespace std;
int a[1000005];
queue<int>que[100005];
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		que[a[i]].push(i);
	}
	long long int ans=0;
	for(int i=n;i>=1;i--){
		ans+=abs(que[a[i]].front()+i-n-1);
		que[a[i]].pop();
	}
	cout<<ans/2;
} 

Python3(3.9) 解法, 执行用时: 222ms, 内存消耗: 20832K, 提交时间: 2021-02-02 10:42:15

n = int(input())
a = list(map(int, input().split()))
pos = [list() for _ in range(n + 1)]
for i in range(n):
    pos[a[i]].append(i)
ans = 0
for i in pos:
    j = [n - 1 - k for k in i][::-1]
    for k in range(len(i)):
        ans += abs(i[k] - j[k])
print(ans // 2)