[mypy] fix type annotations for graphs/a_star.py #4052

DIRECTORY.md

@@ -274,6 +274,7 @@
     * [Test Send File](https://github.com/TheAlgorithms/Python/blob/master/file_transfer/tests/test_send_file.py)
 
 ## Fractals
+  * [Julia Sets](https://github.com/TheAlgorithms/Python/blob/master/fractals/julia_sets.py)
   * [Koch Snowflake](https://github.com/TheAlgorithms/Python/blob/master/fractals/koch_snowflake.py)
   * [Mandelbrot](https://github.com/TheAlgorithms/Python/blob/master/fractals/mandelbrot.py)
   * [Sierpinski Triangle](https://github.com/TheAlgorithms/Python/blob/master/fractals/sierpinski_triangle.py)
@@ -424,10 +425,12 @@
   * [Binomial Distribution](https://github.com/TheAlgorithms/Python/blob/master/maths/binomial_distribution.py)
   * [Bisection](https://github.com/TheAlgorithms/Python/blob/master/maths/bisection.py)
   * [Ceil](https://github.com/TheAlgorithms/Python/blob/master/maths/ceil.py)
+  * [Check Polygon](https://github.com/TheAlgorithms/Python/blob/master/maths/check_polygon.py)
   * [Chudnovsky Algorithm](https://github.com/TheAlgorithms/Python/blob/master/maths/chudnovsky_algorithm.py)
   * [Collatz Sequence](https://github.com/TheAlgorithms/Python/blob/master/maths/collatz_sequence.py)
   * [Combinations](https://github.com/TheAlgorithms/Python/blob/master/maths/combinations.py)
   * [Decimal Isolate](https://github.com/TheAlgorithms/Python/blob/master/maths/decimal_isolate.py)
+  * [Double Factorial Iterative](https://github.com/TheAlgorithms/Python/blob/master/maths/double_factorial_iterative.py)
   * [Double Factorial Recursive](https://github.com/TheAlgorithms/Python/blob/master/maths/double_factorial_recursive.py)
   * [Entropy](https://github.com/TheAlgorithms/Python/blob/master/maths/entropy.py)
   * [Euclidean Distance](https://github.com/TheAlgorithms/Python/blob/master/maths/euclidean_distance.py)
@@ -511,6 +514,7 @@
   * [Sum Of Arithmetic Series](https://github.com/TheAlgorithms/Python/blob/master/maths/sum_of_arithmetic_series.py)
   * [Sum Of Digits](https://github.com/TheAlgorithms/Python/blob/master/maths/sum_of_digits.py)
   * [Sum Of Geometric Progression](https://github.com/TheAlgorithms/Python/blob/master/maths/sum_of_geometric_progression.py)
+  * [Sylvester Sequence](https://github.com/TheAlgorithms/Python/blob/master/maths/sylvester_sequence.py)
   * [Test Prime Check](https://github.com/TheAlgorithms/Python/blob/master/maths/test_prime_check.py)
   * [Trapezoidal Rule](https://github.com/TheAlgorithms/Python/blob/master/maths/trapezoidal_rule.py)
   * [Triplet Sum](https://github.com/TheAlgorithms/Python/blob/master/maths/triplet_sum.py)
@@ -928,7 +932,6 @@
   * [Reverse Letters](https://github.com/TheAlgorithms/Python/blob/master/strings/reverse_letters.py)
   * [Reverse Words](https://github.com/TheAlgorithms/Python/blob/master/strings/reverse_words.py)
   * [Split](https://github.com/TheAlgorithms/Python/blob/master/strings/split.py)
-  * [Swap Case](https://github.com/TheAlgorithms/Python/blob/master/strings/swap_case.py)
   * [Upper](https://github.com/TheAlgorithms/Python/blob/master/strings/upper.py)
   * [Word Occurrence](https://github.com/TheAlgorithms/Python/blob/master/strings/word_occurrence.py)
   * [Word Patterns](https://github.com/TheAlgorithms/Python/blob/master/strings/word_patterns.py)

graphs/a_star.py

@@ -1,37 +1,21 @@
-grid = [
-    [0, 1, 0, 0, 0, 0],
-    [0, 1, 0, 0, 0, 0],  # 0 are free path whereas 1's are obstacles
-    [0, 1, 0, 0, 0, 0],
-    [0, 1, 0, 0, 1, 0],
-    [0, 0, 0, 0, 1, 0],
-]
-
-"""
-heuristic = [[9, 8, 7, 6, 5, 4],
-             [8, 7, 6, 5, 4, 3],
-             [7, 6, 5, 4, 3, 2],
-             [6, 5, 4, 3, 2, 1],
-             [5, 4, 3, 2, 1, 0]]"""
-
-init = [0, 0]
-goal = [len(grid) - 1, len(grid[0]) - 1]  # all coordinates are given in format [y,x]
-cost = 1
-
-# the cost map which pushes the path closer to the goal
-heuristic = [[0 for row in range(len(grid[0]))] for col in range(len(grid))]
-for i in range(len(grid)):
-    for j in range(len(grid[0])):
-        heuristic[i][j] = abs(i - goal[0]) + abs(j - goal[1])
-        if grid[i][j] == 1:
-            heuristic[i][j] = 99  # added extra penalty in the heuristic map
-
+from __future__ import annotations
 
-# the actions we can take
-delta = [[-1, 0], [0, -1], [1, 0], [0, 1]]  # go up  # go left  # go down  # go right
+DIRECTIONS = [
+    [-1, 0],  # left
+    [0, -1],  # down
+    [1, 0],  # right
+    [0, 1],  # up
+]
 
 
 # function to search the path
-def search(grid, init, goal, cost, heuristic):
+def search(
+    grid: list[list[int]],
+    init: list[int],
+    goal: list[int],
+    cost: int,
+    heuristic: list[list[int]],
+) -> tuple[list[list[int]], list[list[int]]]:
 
     closed = [
         [0 for col in range(len(grid[0]))] for row in range(len(grid))
@@ -52,7 +36,7 @@ def search(grid, init, goal, cost, heuristic):
 
     while not found and not resign:
         if len(cell) == 0:
-            return "FAIL"
+            raise ValueError("Algorithm is unable to find solution")
         else:  # to choose the least costliest action so as to move closer to the goal
             cell.sort()
             cell.reverse()
@@ -64,9 +48,9 @@ def search(grid, init, goal, cost, heuristic):
             if x == goal[0] and y == goal[1]:
                 found = True
             else:
-                for i in range(len(delta)):  # to try out different valid actions
-                    x2 = x + delta[i][0]
-                    y2 = y + delta[i][1]
+                for i in range(len(DIRECTIONS)):  # to try out different valid actions
+                    x2 = x + DIRECTIONS[i][0]
+                    y2 = y + DIRECTIONS[i][1]
                     if x2 >= 0 and x2 < len(grid) and y2 >= 0 and y2 < len(grid[0]):
                         if closed[x2][y2] == 0 and grid[x2][y2] == 0:
                             g2 = g + cost
@@ -79,22 +63,46 @@ def search(grid, init, goal, cost, heuristic):
     y = goal[1]
     invpath.append([x, y])  # we get the reverse path from here
     while x != init[0] or y != init[1]:
-        x2 = x - delta[action[x][y]][0]
-        y2 = y - delta[action[x][y]][1]
+        x2 = x - DIRECTIONS[action[x][y]][0]
+        y2 = y - DIRECTIONS[action[x][y]][1]
         x = x2
         y = y2
         invpath.append([x, y])
 
     path = []
     for i in range(len(invpath)):
         path.append(invpath[len(invpath) - 1 - i])
+    return path, action
+
+
+if __name__ == "__main__":
+    grid = [
+        [0, 1, 0, 0, 0, 0],
+        [0, 1, 0, 0, 0, 0],  # 0 are free path whereas 1's are obstacles
+        [0, 1, 0, 0, 0, 0],
+        [0, 1, 0, 0, 1, 0],
+        [0, 0, 0, 0, 1, 0],
+    ]
+
+    init = [0, 0]
+    # all coordinates are given in format [y,x]
+    goal = [len(grid) - 1, len(grid[0]) - 1]
+    cost = 1
+
+    # the cost map which pushes the path closer to the goal
+    heuristic = [[0 for row in range(len(grid[0]))] for col in range(len(grid))]
+    for i in range(len(grid)):
+        for j in range(len(grid[0])):
+            heuristic[i][j] = abs(i - goal[0]) + abs(j - goal[1])
+            if grid[i][j] == 1:
+                # added extra penalty in the heuristic map
+                heuristic[i][j] = 99
+
+    path, action = search(grid, init, goal, cost, heuristic)
+
     print("ACTION MAP")
     for i in range(len(action)):
         print(action[i])
 
-    return path
-
-
-a = search(grid, init, goal, cost, heuristic)
-for i in range(len(a)):
-    print(a[i])
+    for i in range(len(path)):
+        print(path[i])