From 332f13fefb21cf47361562db42bac73ef608ec0d Mon Sep 17 00:00:00 2001
From: JonZhao <1044264932@qq.com>
Date: Wed, 5 Jun 2019 15:45:16 +0800
Subject: use limited vehicle num to initiate path in acs-vei

---
 vrptw_base.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'vrptw_base.py')

diff --git a/vrptw_base.py b/vrptw_base.py
index 52cdafa..30c20de 100644
--- a/vrptw_base.py
+++ b/vrptw_base.py
@@ -96,14 +96,18 @@ class VrptwGraph:
             self.pheromone_mat[current_ind][next_ind] += self.rho/best_path_distance
             current_ind = next_ind
 
-    def nearest_neighbor_heuristic(self):
+    def nearest_neighbor_heuristic(self, max_vehicle_num=None):
         index_to_visit = list(range(1, self.node_num))
         current_index = 0
         current_load = 0
         current_time = 0
         travel_distance = 0
         travel_path = [0]
-        while len(index_to_visit) > 0:
+
+        if max_vehicle_num is None:
+            max_vehicle_num = self.node_num
+
+        while len(index_to_visit) > 0 and max_vehicle_num > 0:
             nearest_next_index = self._cal_nearest_next_index(index_to_visit, current_index, current_load, current_time)
 
             if nearest_next_index is None:
@@ -113,6 +117,8 @@ class VrptwGraph:
                 current_time = 0
                 travel_path.append(0)
                 current_index = 0
+
+                max_vehicle_num -= 1
             else:
                 current_load += self.nodes[nearest_next_index].demand
 
-- 
cgit v1.2.3