Create DistanceMetric.py

FreddieLi · FreddieLi · commit 88fcde42faa2 · 2023-09-29T16:52:19.000+08:00
diff --git a/01_base/练习/DistanceMetric.py b/01_base/练习/DistanceMetric.py
@@ -0,0 +1,38 @@
+# Python3 program for the above approach
+from collections import defaultdict
+
+# Function to find sum of differences
+# of indices of occurrences of each
+# unique array element
+def sum_i(arr, n):
+	# Stores indices of each array element
+	mp = defaultdict(lambda : [])
+	# Store the indices
+	for i in range(n):
+		mp[arr[i]].append(i)
+	# Stores the sums
+	ans = [0] * n
+
+	# Traverse the array
+	for i in range(n):
+		# Find sum for each element
+		sum = 0
+		# Iterate over the Map
+		for it in mp[arr[i]]:
+			# Calculate sum of occurrences of arr[i]
+			sum += abs(it - i)
+			# Store sum for current element
+			ans[i] = sum
+	# Print answer for each element
+	for i in range(n):
+		print(ans[i], end = " ")
+
+# Driver code
+if __name__ == '__main__':
+	# Given array
+	arr = [ 1, 3, 1, 1, 2 ]
+	# Given size
+	n = len(arr)
+	# Function Call
+	sum_i(arr, n)
+# This code is contributed by Shivam Singh
