task2

.gitignore

@@ -0,0 +1,10 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv

.python-version

@@ -0,0 +1 @@
+3.13

main.py

@@ -0,0 +1,30 @@
+import numpy as np
+from scipy.stats import wasserstein_distance
+
+def calculate_emd(vector1, vector2):
+    vec1 = np.array(vector1)
+    vec2 = np.array(vector2)
+    
+    positions = np.arange(len(vec1))
+    emd = wasserstein_distance(positions, positions, vec1, vec2)
+    
+    total_weight = np.sum(vec1)  # 权重和
+    emd = emd * total_weight
+    
+    return emd
+
+def main():
+    # input_str = input().strip()
+    line1 = input("输入第一行：").strip()
+    line2 = input("输入第二行：").strip()
+
+    vector1 = list(map(float, line1.split()))
+    vector2 = list(map(float, line2.split()))
+
+    # 计算EMD距离
+    emd_value = calculate_emd(vector1, vector2)
+
+    print(int(round(emd_value)))
+
+if __name__ == '__main__':
+    main()

pyproject.toml

@@ -0,0 +1,7 @@
+[project]
+name = "poker-task2"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = []