snsd0805/NTU-AI-HW1
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: return when isGoal

Browse Source
This commit is contained in:
snsd0805 2024-03-26 22:10:59 +08:00
parent 7a31882480
commit f0c5c6d84a
Signed by: snsd0805
GPG Key ID: 569349933C77A854
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
search.py
View File

@ -100,6 +100,8 @@ def depthFirstSearch(problem: SearchProblem):
path = path + [direction] path = path + [direction]
visited.append(new_state) visited.append(new_state)
isGoal = problem.isGoalState(new_state) isGoal = problem.isGoalState(new_state)
if isGoal:
break
for (_state, _direction, _cost) in problem.getSuccessors(new_state): # (state, direction, cost) for (_state, _direction, _cost) in problem.getSuccessors(new_state): # (state, direction, cost)
stack.push(((_state, _direction, _cost), path)) stack.push(((_state, _direction, _cost), path))
path = path[1:] path = path[1:]
@ -130,6 +132,8 @@ def breadthFirstSearch(problem: SearchProblem):
path = path + [direction] path = path + [direction]
visited.append(new_state) visited.append(new_state)
isGoal = problem.isGoalState(new_state) isGoal = problem.isGoalState(new_state)
if isGoal:
break
for (_state, _direction, _cost) in problem.getSuccessors(new_state): # (state, direction, cost) for (_state, _direction, _cost) in problem.getSuccessors(new_state): # (state, direction, cost)
queue.push(((_state, _direction, _cost), path)) queue.push(((_state, _direction, _cost), path))
path = path[1:] path = path[1:]
@ -161,6 +165,8 @@ def uniformCostSearch(problem: SearchProblem):
costs = costs + cost costs = costs + cost
visited.append(new_state) visited.append(new_state)
isGoal = problem.isGoalState(new_state) isGoal = problem.isGoalState(new_state)
if isGoal:
break
for (_state, _direction, _cost) in problem.getSuccessors(new_state): # (state, direction, cost) for (_state, _direction, _cost) in problem.getSuccessors(new_state): # (state, direction, cost)
queue.push(((_state, _direction, _cost), path, costs), costs+_cost) queue.push(((_state, _direction, _cost), path, costs), costs+_cost)
path = path[1:] path = path[1:]