更新0097_增加Python3版本

problems/kamacoder/0097.小明逛公园.md

@@ -339,7 +339,7 @@ int main() {
     }
 }
 
-``` 
+```
 
 ## 空间优化
 
@@ -426,6 +426,68 @@ floyd算法的时间复杂度相对较高，适合 稠密图且源点较多的
 
 ### Python
 
+基于三维数组的Floyd
+
+```python
+if __name__ == '__main__':
+    max_int = 10005  # 设置最大路径，因为边最大距离为10^4
+
+    n, m = map(int, input().split())
+
+    grid = [[[max_int] * (n+1) for _ in range(n+1)] for _ in range(n+1)]  # 初始化三维dp数组
+
+    for _ in range(m):
+        p1, p2, w = map(int, input().split())
+        grid[p1][p2][0] = w
+        grid[p2][p1][0] = w
+
+    # 开始floyd
+    for k in range(1, n+1):
+        for i in range(1, n+1):
+            for j in range(1, n+1):
+                grid[i][j][k] = min(grid[i][j][k-1], grid[i][k][k-1] + grid[k][j][k-1])
+
+    # 输出结果
+    z = int(input())
+    for _ in range(z):
+        start, end = map(int, input().split())
+        if grid[start][end][n] == max_int:
+            print(-1)
+        else:
+            print(grid[start][end][n])
+```
+
+基于二维数组的Floyd
+
+```python
+if __name__ == '__main__':
+    max_int = 10005  # 设置最大路径，因为边最大距离为10^4
+
+    n, m = map(int, input().split())
+
+    grid = [[max_int]*(n+1) for _ in range(n+1)]  # 初始化二维dp数组
+
+    for _ in range(m):
+        p1, p2, val = map(int, input().split())
+        grid[p1][p2] = val
+        grid[p2][p1] = val
+
+    # 开始floyd
+    for k in range(1, n+1):
+        for i in range(1, n+1):
+            for j in range(1, n+1):
+                grid[i][j] = min(grid[i][j], grid[i][k] + grid[k][j])
+
+    # 输出结果
+    z = int(input())
+    for _ in range(z):
+        start, end = map(int, input().split())
+        if grid[start][end] == max_int:
+            print(-1)
+        else:
+            print(grid[start][end])
+```
+
 ### Go
 
 ### Rust