maggicl/AICup
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

solver

Browse source
This commit is contained in:
UmbertoJr 2019-10-31 11:57:26 +01:00
parent 6dde40e207
commit 48fcd8738b
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
code/TSP_solver.py
View file

@ -33,7 +33,17 @@ class Solver_TSP:
return self.solution return self.solution
def nn(self, instance_, starting_node=0): def nn(self, instance_, starting_node=0):
pass dist_matrix = np.copy(instance_.dist_matrix)
n = int(instance_.nPoints)
dist_matrix[np.arange(n), np.arange(n)] = 1000
node = np.argmin([starting_node])
tour = [node]
for _ in range(n- 2):
node = np.argmin([node])
tour.append(node)
self.solution = np.array(tour)
self.solved = True
return self.solution
def plot_solution(self): def plot_solution(self):
assert self.solved, "You can't plot the solution, you need to solve it first!" assert self.solved, "You can't plot the solution, you need to solve it first!"