add diagonal_difference code in Udemy

Udemy/diagonal_difference.py

@@ -0,0 +1,20 @@
+def diagonalDifference(arr):
+    primary_diagonal = 0
+    secondary_diagonal = 0
+    n = len(arr)
+
+    for i in range(n):
+        primary_diagonal += arr[i][i]
+        secondary_diagonal += arr[i][n - i - 1]
+
+    return abs(primary_diagonal - secondary_diagonal)
+
+# Input handling
+n = int(input())
+arr = []
+
+for _ in range(n):
+    arr.append(list(map(int, input().split())))
+
+# Function call and output
+print(diagonalDifference(arr))

Udemy/triplet_comparison.py

@@ -0,0 +1,17 @@
+def compareTriplets(a, b):
+    alice_points = 0
+    bob_points = 0
+    for i in range(3):
+        if a[i] > b[i]:
+            alice_points += 1
+        elif a[i] < b[i]:
+            bob_points += 1
+    return (alice_points, bob_points)
+
+# Input handling
+a = list(map(int, input().split()))
+b = list(map(int, input().split()))
+
+# Output the result
+result = compareTriplets(a, b)
+print(result[0], result[1])