write dfs strategy

This commit is contained in:
2024-06-25 15:13:54 +00:00
parent 3caff0910b
commit 7d08ea3480

View File

@ -86,8 +86,27 @@ def depthFirstSearch(problem: SearchProblem):
print("Is the start a goal?", problem.isGoalState(problem.getStartState())) print("Is the start a goal?", problem.isGoalState(problem.getStartState()))
print("Start's successors:", problem.getSuccessors(problem.getStartState())) print("Start's successors:", problem.getSuccessors(problem.getStartState()))
""" """
"*** YOUR CODE HERE ***" print("Start:", problem.getStartState())
util.raiseNotDefined() print("Is the start a goal?", problem.isGoalState(problem.getStartState()))
print("Start's successors:", problem.getSuccessors(problem.getStartState()))
action_list=[]
# action_list=["South", "South", "West", "South", "West", "West", "South", "West"]
vis_stk=util.Stack()
has_visited={}
vis_stk.push((problem.getStartState(),[]))
has_visited[problem.getStartState()]=True
while True:
if vis_stk.isEmpty():
break
cur_state,cur_actions=vis_stk.pop()
if problem.isGoalState(cur_state):
action_list=cur_actions
break
for next_state,action,_ in problem.getSuccessors(cur_state):
if next_state not in has_visited:
vis_stk.push((next_state,cur_actions+[action]))
has_visited[next_state]=True
return action_list
def breadthFirstSearch(problem: SearchProblem): def breadthFirstSearch(problem: SearchProblem):
"""Search the shallowest nodes in the search tree first.""" """Search the shallowest nodes in the search tree first."""