bangoc123 / learn-machine-learning-in-two-months

Những kiến thức cần thiết để học tốt Machine Learning trong vòng 2 tháng. Essential Knowledge for learning Machine Learning in two months.
2.07k stars 662 forks source link

Should change addEdge function in Graph python file - Nên sửa đổi code hàm addEdge #5

Closed moonbka closed 5 years ago

moonbka commented 6 years ago

Hi Bạn, Mình ko thấy file được change code, nên mình tạo issue này để góp ý sửa đổi ạ Mình thấy phần code của : Graph -> addEdge() (phần thêm cạnh), code đang hơi phức tạp về mặt giải thuật. Mình góp ý nên sửa lại như sau, để code vừa đơn giản, vừa hiệu quả hơn;

def addEdge(self, edge):
        if type(edge) is set:
            if edge not in self.edges():
                nodes = self.graph.keys()
                (e_start, e_end) = tuple(edge)
                if e_start not in nodes: #Nếu chưa có node start thì thêm vào, nếu có rồi thì ko cần làm gì
                    self.graph[e_start] = []
                if e_end not in nodes: #Nếu chưa có node end thì thêm vàoì, nếu có rồi thì ko cần làm gì
                    self.graph[e_end] = []
                #Bây giờ cả hai node đều chắc chắn có trong graph
                self.graph[e_start].append(e_end)
                self.graph[e_end].append(e_start)
        else:
            print("Edge type must be set")
bangoc123 commented 6 years ago

@moonbka thanks for your comment. Let me review your code.