# -*- coding: utf-8 -*- # Time : 2023/11/7 17:03 # Author : lirunsheng # User : l'r's # Software: PyCharm # File : X4_1.py # -*- coding: utf-8 -*- # Time : 2023/10/30 15:49 # Author : lirunsheng # User : l'r's # Software: PyCharm # File : X4.py from tkinter import Tk import random import math import time import tkinter as tk from tkinter import * from tkinter import messagebox import numpy as np MODE = 0 BLUE = "#0080FF" BLACK = "#000000" RED = "#FFAAAA" YELLOW = "#FFFACD" LINE = '#c8d2c8' GREY = '#070b19' GREEN = '#5ba585' NODE = '#33a8cd' ZERO = 'gold' MARK_NODE = 0 Specify_Nodes = '' Edges_list = [] NAME = [] class Data: def __init__(self, source: int, num: int): global NAME if num <= 0 or num > 10: raise ValueError("num must be a positive integer") self.edgeinfo = 1 # 是否显示详情 self.nodes_num = num # 节点个数 self.ang = 360 / self.nodes_num # 圆心顶点角度 self.R = 300 # 外接圆半径 self.bc = 2 * self.R * math.sin(math.pi / self.nodes_num) # 节点之间的正多边形距离 self.canvas_len = int(2 * self.R + 80) # 画布边长 self.center = (self.canvas_len // 2, self.canvas_len // 2) # 画布中心点坐标 self.drop = [] self.index = 0 if source == 3 or source == 2: # 指定删除 # name = [str(i) for i in range(1, num + 2)] # print(name) if source == 2: NAME.pop() else: # print(Specify_Nodes) NAME.remove(Specify_Nodes) self.name = NAME self.coordinate = self.coord_creat() # 纯节点坐标 self.Nodes = self.nodes_creat() # 创建节点列表 print(self.Nodes) self.Edges = self.edges_delete(source) # 创建第一条连接 else: if source == 1: name = str(int(NAME[-1])+1) # print(name) NAME.append(name) else: NAME = [str(i) for i in range(1, num + 1)] # print(NAME) self.name = NAME self.coordinate = self.coord_creat() # 纯节点坐标 self.Nodes = self.nodes_creat() # 创建节点列表 print(self.Nodes) self.Edges = self.edges_creat(source) # 创建第一条连接 self.edge_add(2, source) # 创建第2条连接 self.edge_add(3, source) # 创建第3条连接 def nodes_creat(self, n_sum=None): if n_sum == None: n_sum = self.nodes_num nodes = [] # 初始化node表 # 设置画布中心点坐标x0,y0 x0 = self.center[0] y0 = self.center[1] for i in range(n_sum): # 通过几何运算得到多边形各个顶点坐标 rad = math.radians(i * self.ang) # 计算第i个点的弧度 x = int(math.cos(rad) * self.R) # 计算第i个顶点x坐标 y = int(math.sin(rad) * self.R) # 计算第i个顶点y坐标 # name = '' + str(i + 1) # 给第i个顶点命名 name = '' + self.name[i] mark = 0 # 节点为未标记 dominator = 1 # 设置为必经节点 nodes.append([i, name, mark, (x0 + x, y0 + y), dominator]) # 将当前节点加入node中 return nodes def edges_delete(self, source): edges = [] # 初始化所有链接集合 ser = 0 # print(str(int(Specify_Nodes) + 1)) for list_edge in Edges_list: # print(list_edge) edges.append([ser, list_edge[1], list_edge[2],list_edge[3], list_edge[4],list_edge[5], list_edge[6], list_edge[7], list_edge[8]]) # 将连接加入到边的集合中 nox = [int(num) for num in edges[ser][5].split('-')] # print(nox) if len(nox) == 2: ss = edges[ser][5] else: ss = edges[ser][5][:-2] # print(ss) if source == 2: if str(len(self.Nodes)+1) in ss: # print(edges[ser]) edges.pop() ser -= 1 else: if Specify_Nodes in ss: # print(edges[ser]) edges.pop() ser -= 1 ser += 1 return edges def edges_creat(self, source): edges = [] # 初始化所有链接集合 ser = 0 # 初始化链接对象编号 nodes = self.Nodes.copy() # 复制节点对象 nodes1 = self.Nodes.copy() # 复制节点对象 # print(nodes1) self.drop = [] for node in nodes: # 遍历节点对象 if node in nodes1: # 删除nodes1中当期遍历的节点信息 nodes1.remove(node) # print(nodes1) for node1 in nodes1: # 遍历删除后的节点信息 n1 = node[0] # 节点对象编号1 n2 = node1[0] # 节点对象编号2 if self.drop.count(n2) > 3 or self.drop.count(n1) > 3: dist_c = random.randint(101, 120) # 随机生成距离成本 time_c = random.randint(101, 120) # 随机生成链接的时间 else: dist_c = random.randint(10, 120) # 随机生成距离成本 time_c = random.randint(10, 120) # 随机生成链接的时间 seq = 1 # 链接序号 mark = False # 连接是否标记 tag = f"{NAME[n1]}-{NAME[n2]}" # 链接标签 enable = 1 # 连接是否可用 if dist_c >= 100 or time_c >= 100: # 设置某些节点为不显示 enable = 0 else: self.drop.append(n1) self.drop.append(n2) if source == 1: # print(n2, len(self.Nodes)) if n2 == len(self.Nodes)-1: # print('进来了') edges.append([ser, int(NAME[n1])-1, int(NAME[n2])-1, enable, mark, tag, dist_c, time_c, seq]) # 将连接加入到边的集合中 else: # print(self.index) edges.append([ser, Edges_list[self.index][1], Edges_list[self.index][2], Edges_list[self.index][3], Edges_list[self.index][4], Edges_list[self.index][5],Edges_list[self.index][6], Edges_list[self.index][7], Edges_list[self.index][8]]) # 将连接加入到边的集合中 self.index += 1 else: edges.append([ser, int(NAME[n1])-1, int(NAME[n2])-1, enable, mark, tag, dist_c, time_c, seq]) # 将连接加入到边的集合中 ser += 1 # 计数加一 return edges def edge_add(self, no, source): nodes = self.Nodes.copy() # 复制节点对象 nodes1 = self.Nodes.copy() # 复制节点对象 for node in nodes: # 遍历节点对象 if node in nodes1: # 删除nodes1中当期遍历的节点信息 nodes1.remove(node) for node1 in nodes1: # 遍历删除后的节点信息 ser = len(self.Edges) # 链接序号 n1 = node[0] # 节点对象编号1 n2 = node1[0] # 节点对象编号2 show = 0 # 设置为不显示 mark = False # 连接是否标记 tag = f"{NAME[n1]}-{NAME[n2]}-{no}" # 链接标签 dist_c = random.randint(30, 100) # 距离成本 time_c = random.randint(1, 30) # 链接的时间 enable = no # 连接是否可用 if source == 1: if n2 == len(self.Nodes)-1: # print('进来了') self.Edges.append([ser, int(NAME[n1])-1, int(NAME[n2])-1, show, mark, tag, dist_c, time_c, enable]) # 将连接加入到边的集合中 else: # print(no ,self.index) self.Edges.append([ser, Edges_list[self.index][1], Edges_list[self.index][2], Edges_list[self.index][3], Edges_list[self.index][4], Edges_list[self.index][5], Edges_list[self.index][6], Edges_list[self.index][7], Edges_list[self.index][8]]) # 将连接加入到边的集合中 self.index += 1 else: self.Edges.append([ser, int(NAME[n1])-1, int(NAME[n2])-1, show, mark, tag, dist_c, time_c, enable]) def coord_creat(self): '''返回每个节点的坐标''' coordinate = [] x0 = self.center[0] y0 = self.center[1] for i in range(self.nodes_num): rad = math.radians(i * self.ang) x = int(math.cos(rad) * self.R) y = int(math.sin(rad) * self.R) coordinate.append((x0 + x, y0 + y)) return coordinate def node_co(self, key): # 通过节点编号返回编号的坐标 """通过节点编号返回编号的坐标""" nodes = self.Nodes for n in nodes: if n[1] == str(key + 1): return n[3] def node_no(self, key): # 通过节点编号返回编号的坐标 """通过节点编号返回编号的坐标""" nodes = self.Nodes for n in nodes: if n[1] == str(key + 1): return n[0] # G = Graph() # print(1) nodes = 6 # 设置初始节点数量 d = Data(0, nodes) # print(1) root = Tk() # 设置主界面 root.title("旅行商问题") # 设置标题 root.geometry(f"{d.canvas_len + 700}x{d.canvas_len + 100}") # 设置大小 root.resizable(0, 0) # 设置不能调整显示边框 frame = tk.Frame(root, padx=20, pady=20, width=d.canvas_len + 340, height=d.canvas_len + 50) frame.grid() # 绘制显示框 frame_top = tk.Frame(frame, width=d.canvas_len, height=50) frame_top.grid(row=0, column=0) # 绘制信息输出栏 cv = canvas = tk.Canvas(frame, bg='white', bd=2, relief="sunken", width=d.canvas_len + 330, height=d.canvas_len) canvas.grid(row=1, column=0) # 放置绘图Canvas画布 frame_right = tk.Frame(root, bg=RED, width=300, height=d.canvas_len + 50) # 右边栏 frame_right.grid(row=0, column=1) # 右边栏 class Graph: # Graph定义 def edge_info(self, cv: tk.Canvas, info: str, n1: tuple, n2: tuple): # 显示路径信息 if d.edgeinfo == 0: # 若设置为不显示则直接返回 return # print(n1) # print(n2) x = (n1[0] - n2[0]) // 2 + n2[0] # 计算中点x坐标 y = (n1[1] - n2[1]) // 2 + n2[1] # 计算中点y坐标 if (n2[1] - n1[1]) != 0: # 斜率为非0度时的显示 reat = math.atan((n2[0] - n1[0]) / (n2[1] - n1[1])) # 根据斜率计算弧度 ang = round(math.degrees(reat)) + 90 # 根据弧度转角度 if ang > 89: # 斜度大于90度调整坐标 text_x, text_y = x + 15, y else: # 斜度大于90度调整坐标 text_x, text_y = x, y + 15 else: # 斜率为0度时的显示 ang = round(math.degrees(0)) # 根据弧度转角度 text_x, text_y = x - 30, y + 10 # print(info.split(' ')[-1]) cv.create_text(text_x, text_y, text=f'{info.split()[-2]}', fill="black", font=('微软雅黑', 12), angle=ang) # 根据信息显示文字 def draw(self): cv.delete('all') # 清空画布 dis, tim = self.edges_show(cv, d.Edges) # 绘制边 self.lab_show(frame_top, len(d.Nodes), tim, dis) # 显示边信息 self.nodes_show(cv) # 绘制点 cv.update() # 更新画布 def lab_show(self, frame: tk.Frame, nodes: int, time: int, distance: int): labs = ["节点数目", "时间", "距离(成本)"] x0 = d.canvas_len // 3 for i in range(len(labs)): lab1 = tk.Label(frame, text=labs[i], fg=BLACK, font=('微软雅黑', 12)) if i == 0: lab2 = tk.Label(frame, text=f"{nodes}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12)) elif i == 1: lab2 = tk.Label(frame, text=f"{time}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12)) else: lab2 = tk.Label(frame, text=f"{distance}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12)) lab1.place(x=10 + x0 * i, y=10, width=80, height=30) lab2.place(x=10 + x0 * i + 80, y=10, width=80, height=30) lab1 = tk.Label(frame, text='分钟', fg=BLACK, font=('微软雅黑', 12)) lab1.place(x=170 + x0, y=10, width=50, height=30) lab1 = tk.Label(frame, text='千米', fg=BLACK, font=('微软雅黑', 12)) lab1.place(x=170 + x0 * 2, y=10, width=50, height=30) def nodes_show(self, cv: tk.Canvas): COL = NODE # 初始化节点颜色 for i in d.Nodes: # 遍历每一个节点 x, y = i[3] # 记录该节点坐标 if i[2] == 1 or i[2] == 2: # 如果节点被标记，则将节点标为红色 COL = RED elif i[0] == 0: # 如果节点为起始节点，则将节点标记为特殊颜色 COL = ZERO elif i[2] == 5: # 如果节点已经被访问，则将节点标记（用户路径功能） COL = 'gold' else: # 否则则使用初始颜色 COL = NODE if i[4] == 1: # 如果是必经节点，则加黑色边框 w = 1 else: # 如果是非必经节点，则没有边框 w = 0 # 根据坐标，圆的颜色与是否有边框绘制圆圈来表示节点 cv.create_oval(x - 20, y - 20, x + 20, y + 20, fill=COL, width=w, ) # outline='#ee28a3') cv.create_text(x, y, text=f'{i[1]}', fill=GREY, font=('华文楷体', 18)) cv.update() def edges_show(self, cv: tk.Canvas, edges: d.Edges): """ 显示链接 :param cv: tkcanvas对象 :return: """ distance = 0 # 初始化被标记的线的总路径 time1 = 0 # 初始化被标记的线的总时间 for edge in edges: node_1 = d.node_co(edge[1]) # 链接线编号对应的坐标1 node_2 = d.node_co(edge[2]) # 链接线编号对应的坐标2 # print(node_1,node_2) if edge[8] == 2 and edge[3] == 1: # 若该链接为两点间第二条连线 self.more_edge_show(2, edge) elif edge[8] == 3 and edge[3] == 1: # 若为两点间第3条连线 self.more_edge_show(3, edge) if edge[8] == 1 and edge[3] == 1: # 若为两点间第1条连线且该连接存在 self.edge_info(cv, f'{edge[5]} {edge[6]} {edge[7]}', node_1, node_2) point = [node_1, node_2] if edge[4] == 1 and edge[8] == 1 and edge[3] == 1: # 如果路径被标记，则用绿色绘制 if MODE == 1: cv.create_line(point, fill=RED, width=4) elif MODE == 2: cv.create_line(point, fill=BLUE, width=4) else: cv.create_line(point, fill=GREEN, width=4) distance += edge[6] # 更新总路径 time1 += edge[7] # 更新总时间 elif edge[8] == 1 and edge[3] == 1: # 如果是节点之间的第一条线，则用正常颜色标记 cv.create_line(point, fill=LINE, width=2) if d.Nodes[d.node_no(edge[2])][2] + d.Nodes[d.node_no(edge[2])][2] == 3: # 若该线被用户选中，则用蓝色虚线标记 cv.create_line(point, fill=BLUE, width=5, dash=(4, 14)) return (distance, time1) # 返回标记路径的总时间与路程 # def node_edge(self, a, b): # 根据点的编号找边的编号 # if a > b: # x = a # a = b # b = x # return int(a * (2 * len(d.Nodes) - a - 3) / 2 + b - 1) def node_edge(self, a, b): # 根据点的编号找边的编号 for edge in d.Edges: if a == edge[1] and b == edge[2]: return edge[0] if b == edge[1] and a == edge[2]: return edge[0] def way_node_edge(self, a, b): # 根据点的编号找边的编号 way = [] for node in d.Nodes: way.append(int(node[1])) # print(way) # print(a, b) x, y = way[a]-1, way[b]-1 for edge in d.Edges: if x == edge[1] and y == edge[2]: return edge[0] if y == edge[1] and x == edge[2]: return edge[0] def delete_all(self): cv.delete('all') # 清空画布 self.lab_show(frame_top, 0, 0, 0) # 显示边信息 W.user_way = [0] W.way_route_markings() class Edge: def __init__(self): self.mark = [] # 记录已标记的节点数目（连接操作） self.user_cur = 0 # 记录用户所在点 def node_edge(self, no, n1, n2): # 根据节点返回编号 if n1 > n2: n = n1 n1 = n2 n2 = n for e in d.Edges: if e[1] == n1 and e[2] == n2 and e[8] == no: return e[0] return -1 def edge_del_all(self, mode): # mode = 1:优先最短路径 # mode = 2:优先最短时间 # cur1 = node_edge() if mode == 1: for e in d.Edges: if (e[8] == 2 and e[3] == 1) or (e[8] == 3 and e[3] == 1): self.edge_merge_cost(e[1], e[2]) else: for e in d.Edges: if (e[8] == 2 and e[3] == 1) or (e[8] == 3 and e[3] == 1): self.edge_merge_time(e[1], e[2]) G.draw() def edge_merge_cost(self, n1, n2): cur1 = node_edge(n1, n2) # 初始化三条可能的边的编号 cur2 = node_edge(n1, n2, 2) # 初始化三条可能的边的编号 cur3 = node_edge(n1, n2, 3) # 初始化三条可能的边的编号 edges = [(cur1, d.Edges[cur1][6]), (cur2, d.Edges[cur2][6]), (cur3, d.Edges[cur3][6])] # 提出编号与路径花费 valid_edges = [(edge, cost) for edge, cost in edges if d.Edges[edge][3] == 1] # 过滤出存在的边 valid_edges.sort(key=lambda x: x[1]) # 按照 cost 排序，保留 cost 最小的边 for idx, (edge, _) in enumerate(valid_edges): d.Edges[edge][8] = idx + 1 # 设置边的优先级 d.Edges[edge][0] = edge # 确保边的索引正确 for edge, _ in valid_edges[1:]: # 对于非最优边，设置其为不显示 d.Edges[edge][3] = 0 def edge_merge_time(self, n1, n2): cur1 = node_edge(n1, n2) # 初始化三条可能的边的编号 cur2 = node_edge(n1, n2, 2) # 初始化三条可能的边的编号 cur3 = node_edge(n1, n2, 3) # 初始化三条可能的边的编号 edges = [(cur1, d.Edges[cur1][7]), (cur2, d.Edges[cur2][7]), (cur3, d.Edges[cur3][7])] # 提出编号与路径花费 valid_edges = [(edge, cost) for edge, cost in edges if d.Edges[edge][3] == 1] # 过滤出存在的边 valid_edges.sort(key=lambda x: x[1]) # 按照 cost 排序，保留 cost 最小的边 for idx, (edge, _) in enumerate(valid_edges): d.Edges[edge][8] = idx + 1 # 设置边的优先级 d.Edges[edge][0] = edge # 确保边的索引正确 for edge, _ in valid_edges[1:]: # 对于非最优边，设置其为不显示 d.Edges[edge][3] = 0 class Solve: def __init__(self): self.ans = [] # 记录答案 self.mutians = [] # 若有多个最短路，则保存在mutians中 self.way_sum = 0 # 记录遍历的路径数目 self.minn = float("inf") # 初始化最小值为正无穷 self.start = 0 # 起始点 self.end = 0 # 终点 self.mark = [] # 记录该点是否已经经过 self.user_cur = 0 # 记录用户所在点 self.cura = -1 # 记录用户选择的第一个点 self.curb = -1 # 记录用户选择的第二个点 self.con = 0 # 记录必经的节点数目 def node_greedy(self): # 贪心 global MODE, MARK_NODE MARK_NODE = 0 MODE = 2 self.start = time.time() # 记录开始时间 self.ans.clear() # 初始化答案列表 E.edge_del_all(1) # 两点间仅保留距离最短路径 for edge in d.Edges: # 清空标记 edge[4] = False i = 0 # 起始点 flag = [0] * len(d.Nodes) # 初始化已经过点 cur = i # 初始化起始点 flag[cur] = 1 # 将起始点标记为已经过 l = len(d.Nodes) # print(l) distances = [[math.inf for _ in range(l)] for _ in range(l)] ladder_sum = 0 for j in range(1, l + 1): ladder_sum += j - 1 for j in d.Edges: # 每一步选择当前最短的链接加到路径中去 i += 1 # print(j) if i > ladder_sum: break # print(1) else: if j[6] < 99 and j[7] < 99: distances[d.node_no(j[1])][d.node_no(j[2])] = j[6] distances[d.node_no(j[2])][d.node_no(j[1])] = j[6] ways = self.greedy(distances) self.ans = ways.copy() # 如果找到的路径不合法，则表示贪心法无法求解得到合法路径 # print(ways) # for i in distances: # print(i) if len(ways) != l+1 : tk.messagebox.askokcancel(title='结果', message='该图不存在贪心算法路径') print(ways) for way in range(len(ways) - 1): # 标记一条路径 d.Edges[G.way_node_edge(ways[way], ways[way + 1])][4] = True G.draw() self.ans.clear() return -1 for way in range(len(ways) - 1): # 标记一条路径 d.Edges[G.way_node_edge(ways[way], ways[way + 1])][4] = True # print(ways[len(ways) - 1], i) # d.Edges[G.way_node_edge(ways[len(ways) - 1], i)][4] = True # 标记起点终点形成环线 # print(ways) tk.messagebox.askokcancel(title='结果', message='贪心算法路径') self.end = time.time() # 记录结束时间 G.draw() def greedy(self,distances): num_cities = len(distances) visited = [False] * num_cities path = [0] # 起始城市为0 visited[0] = True for _ in range(num_cities - 1): curr_city = path[-1] min_distance = float('inf') next_city = None for city in range(num_cities): if not visited[city] and distances[curr_city][city] < min_distance and distances[curr_city][city] != 0: min_distance = distances[curr_city][city] next_city = city if next_city is not None: path.append(next_city) visited[next_city] = True if distances[path[-1]][0] != np.inf: path.append(0) # 回到起始城市形成闭合路径 # print(path) return path def tsp_backtracking(self,distances): num_cities = len(distances) path = [0] # 起始城市为0 visited = [False] * num_cities visited[0] = True min_distance = float('inf') shortest_path = None def backtrack(curr_city, curr_distance, visited, path): nonlocal min_distance, shortest_path if len(path) == num_cities: # 到达所有城市，更新最短路径 if curr_distance + distances[curr_city][0] < min_distance: min_distance = curr_distance + distances[curr_city][0] shortest_path = path + [0] else: for next_city in range(num_cities): if not visited[next_city]: # 选择下一个未访问的城市 visited[next_city] = True path.append(next_city) new_distance = curr_distance + distances[curr_city][next_city] # 剪枝条件：当前路径已经大于最短路径，不继续搜索 if new_distance < min_distance: backtrack(next_city, new_distance, visited, path) # 回溯 visited[next_city] = False path.pop() backtrack(0, 0, visited, path) return shortest_path def node_dfs(self): # 遍历 深度优先算法 global MODE, MARK_NODE MARK_NODE = 0 MODE = 2 self.start = time.time() # 记录开始时间 self.ans.clear() # 初始化答案列表 E.edge_del_all(1) # 两点间仅保留距离最短路径 for edge in d.Edges: # 清空标记 edge[4] = False i = 0 # 起始点 flag = [0] * len(d.Nodes) # 初始化已经过点 cur = i # 初始化起始点 flag[cur] = 1 # 将起始点标记为已经过 l = len(d.Nodes) # print(l) distances = [[math.inf for _ in range(l)] for _ in range(l)] ladder_sum = 0 for j in range(1, l + 1): ladder_sum += j - 1 for j in d.Edges: # 每一步选择当前最短的链接加到路径中去 i += 1 # print(j) if i > ladder_sum: break # print(1) else: if j[6] < 99 and j[7] < 99: distances[d.node_no(j[1])][d.node_no(j[2])] = j[6] distances[d.node_no(j[2])][d.node_no(j[1])] = j[6] cost, ways = self.dfs(distances) self.ans = ways.copy() if len(ways) != l + 1: tk.messagebox.askokcancel(title='结果', message='该图不存在dfs路径') print(ways) for way in range(len(ways) - 1): # 标记一条路径 d.Edges[G.way_node_edge(ways[way], ways[way + 1])][4] = True G.draw() self.ans.clear() return -1 for way in range(len(ways) - 1): # 标记一条路径 d.Edges[G.way_node_edge(ways[way], ways[way + 1])][4] = True tk.messagebox.askokcancel(title='结果', message='dfs算法路径') self.end = time.time() # 记录结束时间 G.draw() def dfs_tsp(self,graph, start, current, path, visited, cost, min_cost, min_path): if len(path) == len(graph) and graph[current][start] != np.inf: path.append(start) cost += graph[current][start] if cost < min_cost[0]: min_cost[0] = cost min_path[0] = path.copy() path.pop() cost -= graph[current][start] return for next_node in range(len(graph)): if graph[current][next_node] != np.inf and not visited[next_node]: visited[next_node] = True path.append(next_node) cost += graph[current][next_node] self.dfs_tsp(graph, start, next_node, path, visited, cost, min_cost, min_path) visited[next_node] = False path.pop() cost -= graph[current][next_node] def dfs(self,graph): n = len(graph) min_cost = [float('inf')] min_path = [[]] for start_node in range(n): visited = [False] * n path = [start_node] cost = 0 visited[start_node] = True self.dfs_tsp(graph, start_node, start_node, path, visited, cost, min_cost, min_path) return min_cost[0], min_path[0] def graph_check(self, a, b): # 检查A点B点是否互通 find = [] # 使用并查集来检测a，b点是否互通 for i in range(len(d.Nodes)): # 初始化并查集 find.append(i) # 使得find[i]=i for edge in d.Edges: # 根据已有的边来更新并查集 if (edge[8] == 1 and edge[3] == 0): continue x = edge[1] y = edge[2] while not x == find[x]: x = find[x] while not y == find[y]: y = find[y] if x < y: find[y] = x else: find[x] = y x = a y = b while not x == find[x]: # 查询a点所在集合 x = find[x] while not y == find[y]: # 查询b点所在集合 y = find[y] if find[x] == find[y]: # 输出检查结果 messagebox.askokcancel(title='结果', message='节点 ' + str(a + 1) + '与节点 ' + str(b + 1) + '互通') else: messagebox.askokcancel(title='结果', message='节点 ' + str(a + 1) + '与节点 ' + str(b + 1) + '不互通') def check(self): # 检查是否存在最短路 global MARK_NODE MARK_NODE = 0 find = [] for i in range(len(d.Nodes)): find.append(i) for edge in d.Edges: if edge[8] == 1 and edge[3] == 0: continue x = edge[1] y = edge[2] while not x == find[x]: x = find[x] while not y == find[y]: y = find[y] if x < y: find[y] = x else: find[x] = y flag = 1 for i in range(len(d.Nodes)): if not find[i] == find[0]: flag = 0 if flag: tk.messagebox.askokcancel(title='结果', message='该图存在最短路') else: tk.messagebox.askokcancel(title='结果', message='该图不能完全互通') def node_edge(n1, n2, no=1): # 根据点的编号找边的编号 if n1 > n2: # 为了防止充分编号，让n1小于n2 n1, n2 = n2, n1 for e in d.Edges: # 遍历所有路径，找到符合要求的返回 if e[1] == n1 and e[2] == n2 and e[8] == no: return e[0] return -1 # 若找不到，则返回-1 class Ways (): def __init__(self): self.user_way = [0] # 记录用户路径 self.user_cur = 0 # 记录用户所在点 self.user_flag = set() # 记录用户已经过的点 def auto_ways(self): # 标记用户路径 for w in d.Edges: # 清空路径标记 w[4] = 0 for N in d.Nodes: # 清空节点标记 N[2] = 0 for w in self.user_way: # 标记用户已选的点 d.Nodes[w][2] = 5 # 标记用户经过的路径 for w in range(len(self.user_way) - 1): d.Edges[G.node_edge(self.user_way[w],self.user_way[w + 1])][4] = 1 # 如果已经访问所有节点，自动回到初始节点 if len(set(self.user_way)) == len(d.Nodes): d.Edges[G.node_edge(self.user_way[len(self.user_way) - 1], 0)][4] = 1 G.draw() # 按照标记重新绘制图形 def check_user_best(self): # 检查用户路径是不是最佳路径 # global user_way # global ans s.node_dfs() # 长度不一样肯定不匹配 if not len(self.user_way) == len(s.ans): return -1,0 dis_user = 0 dis_ans = 0 # print(user_way) # print(ans) num = len(self.user_way) for i in range(num - 1): dis_user += d.Edges[G.node_edge(self.user_way[i]+1, self.user_way[i+1] + 1)][6] dis_ans += d.Edges[G.way_node_edge(s.ans[i], s.ans[i + 1])][6] return dis_ans, dis_user # if dis_ans == dis_user: # 匹配成功，用户路径是最短路 # return 1 # # 匹配失败，不是最短路 # return 0 def show_best_ans(self, top, best, user): lab1 = tk.Label(top, text="最短路径:", fg=BLACK, font=('微软雅黑', 20)) b_way = "" for a in s.ans: b_way += d.Nodes[a][1] b_way += "->" b_way += d.Nodes[0][1] y0 = 0.05 lab2 = tk.Label(top, text=f"{b_way}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab1.place(relx=0.1, rely=0.50 + y0, width=150, height=30) lab2.place(relx=0.4, rely=0.45 + y0, width=200, height=100) lab2 = tk.Label(top, text=f"最佳路径总成本：{best}\n用户路径总成本：{user}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab2.place(relx=0.4, rely=0.7 + y0, width=200, height=100) def show_BEST(self): best, user = self.check_user_best() # 计算最优路径与用户路径的总花费 if best == user: # 如果最优路径与用户路径的总花费相等，则说明用户路径是最短路径 lab1 = tk.Label(frame_right, text="最短路径:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) lab1.place(relx=0.1, rely=0.8, width=80, height=30) else: # 否则说明用户路径不是最短路径 self.servey_path_display() def servey_path_display(self): lab1 = tk.Label(frame_right, text="最短路径:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) lab1.place(relx=0.1, rely=0.75, width=80, height=30) # self.auto_ways() b_way = "" for a in s.ans: b_way += d.Nodes[a][1] b_way += "->" b_way += d.Nodes[0][1] lab2 = tk.Label(frame_right, text=f"{b_way}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab2.place(relx=0.1, rely=0.8, width=200, height=100) def check_user_greey(self): # 检查用户路径是不是贪心路径,不是的话返回错误点 s.node_greedy()# 计算贪心路径 if not len(self.user_way) == len(s.ans):# 路径长度不一样肯定不匹配，直接返回 self.auto_ways() return num = len(self.user_way) for i in range(num):#逐个节点进行匹配 if not self.user_way[i] == s.ans[i]:# 若第i位匹配失败则返回i return i return -1 #-1表示是贪心路径 def show_greddy_ans(self, top): lab1 = tk.Label(top, text="贪心路径:", fg=BLACK, font=('微软雅黑', 20)) b_way = "" for a in s.ans: b_way += d.Nodes[a][1] b_way += "->" b_way += d.Nodes[0][1] y0 = 0.05 lab2 = tk.Label(top, text=f"{b_way}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab1.place(relx=0.1, rely=0.50 + y0, width=150, height=30) lab2.place(relx=0.4, rely=0.45 + y0, width=200, height=100) def show_GREDDY(self): top = Toplevel(root, width = 500, height = 500) top.title('判断路径是否为贪心路径')# 新建一个窗口来显示判断路径是否为贪心路径的结果 lab1 = tk.Label(top, text="用户路径:", fg=BLACK, font=('微软雅黑', 20)) u_way = ""# 显示用户路径 for a in self.user_way: u_way += d.Nodes[a][1] u_way += "->" u_way += d.Nodes[0][1] # print(u_way) y0 = 0.05 lab2 = tk.Label(top, text=f"{u_way}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab1.place(relx=0.1, rely=0.10 + y0, width=150, height=30) lab2.place(relx=0.4, rely=0.05 + y0, width=200, height=100) # 逐个节点比较用户路径是否为贪心路径 g = self.check_user_greey() print(g) print(self.user_way) if g == -1:#如果用户路径是贪心路径则显示 result = '用户路径是贪心路径' elif g: result = '用户路径不是贪心路径' result2 = '用户路径: ' + u_way if len(s.ans) == 7: g_way = '' for a in s.ans: g_way += d.Nodes[a][1] g_way += "->" g_way += d.Nodes[0][1] result2 += '\n贪心路径：' + g_way else: result2 += '\n贪心路径不存在' lab = tk.Label(top, text=result2, bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab.place(relx=0.1, rely=0.5 + y0, width=250, height=100) else:# 若用户路径不是贪心路径，则显示在第几步时选择出错，并将该次选择的正确选择与用户实际选择显示出来 result = '用户路径不是贪心路径' result2 = '用户路径：'+u_way if len(s.ans) == 7: g_way = '' for a in s.ans: g_way += d.Nodes[a][1] g_way += "->" g_way += d.Nodes[0][1] result2 += '\n贪心路径：' + g_way else: result2 += '\n贪心路径不存在' lab = tk.Label(top, text=result2, bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab.place(relx=0.1, rely=0.5 + y0, width=250, height=100) lab = tk.Label(top, text=result, bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=250, justify='left') lab.place(relx=0.1, rely=0.3 + y0, width=200, height=50) self.auto_ways()#将结果输出到输出框 # 对用户路径进行标记与记录 def way_route_markings(self): global MARK_NODE MARK_NODE = 0 # E.newedge_del_all(1) for w in frame_right.winfo_children(): w.destroy() # 清除右侧组件 self.user_way = [0] lab = tk.Label(frame_right, text="路径操作", bg=RED, fg=BLACK, font=('微软雅黑', 20)) lab.place(relx=0.18, rely=0.02, width=200, height=30) y0 = 0.1 global entry1,entry2,default_value # 定义默认值变量 default_value = tk.StringVar() default_value.set(self.user_way[-1]+1) laba1 = tk.Label(frame_right, text="当前节点:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) laba1.place(relx=0.00, rely=0.02 + y0, width=80, height=30) # 创建第一个输入框 entry1 = tk.Entry(frame_right, textvariable=default_value) entry1.place(relx=0.3, rely=0.02 + y0, width=50, height=30) labb1 = tk.Label(frame_right, text="当前下一节点:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) labb1.place(relx=0.48, rely=0.02 + y0, width=100, height=30) # 创建第二个输入框 entry2 = tk.Entry(frame_right) entry2.place(relx=0.82, rely=0.02 + y0, width=50, height=30) labb3 = tk.Label(frame_right, text="是否进行连接:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) labb3.place(relx=0.2, rely=0.08 + y0, width=120, height=30) button = tk.Button(frame_right, text="确认", command=lambda: self.extract_values()) button.place(relx=0.7, rely=0.08 + y0, width=80, height=30) button2 = tk.Button(frame_right, text="撤销", command=lambda: self.revoke()) button2.place(relx=0.01, rely=0.18 + y0, width=140, height=30) button3 = tk.Button(frame_right, text="清除路径", command=lambda: self.clear_all_paths()) button3.place(relx=0.52, rely=0.18 + y0, width=140, height=30) button4 = tk.Button(frame_right, text="判断路径是否为贪心路径", command=lambda: self.greedy_path()) button4.place(relx=0.01, rely=0.26 + y0, width=290, height=30) button5 = tk.Button(frame_right, text="判断路径是否为最短路径", command=lambda: self.shortest_path()) button5.place(relx=0.01, rely=0.33 + y0, width=290, height=30) self.user_path_display() def extract_values(self): global entry1, entry2, default_value value1 = entry1.get() # 获取第一个输入框的值 value2 = entry2.get() # 获取第二个输入框的值 try: value1 = int(value1) # 将值转换为整数 value2 = int(value2) # 将值转换为整数 print("值1:", value1) print("值2:", value2) # 判断两点是否连接 if abs(value2) > 10 and abs(value1) > 10: tk.messagebox.askokcancel(title='错误', message='请重新输入正确的编号！') return else: E = 0 if value1 < value2: str_route = f'{value1}-{value2}' else: str_route = f'{value2}-{value1}' for item in d.Edges: if str_route == item[5]: if item[6] < 100 and item[7] < 100: if d.Edges[G.node_edge(value1-1, value2-1)][4]: E = 2 tk.messagebox.askokcancel(title='错误', message='城市已连接！') else: if self.user_way[-1] == (value1-1): E = 1 global MODE MODE = 1 self.user_way.append(value2 - 1) default_value.set(value2) d.Edges[G.node_edge(value1-1, value2-1)][4] = True entry1.config(textvariable=default_value) # 更新entry1的默认值 entry2.delete(0, tk.END) else: E = 3 tk.messagebox.askokcancel(title='错误', message='未连贯！') break print(E) if E == 1: G.draw() self.user_path_display() elif E == 0: tk.messagebox.askokcancel(title='错误', message='城市未连接！') except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的城市编号！') def extract_ligature(self, n1, n2): no = G.way_node_edge(n1, n2) if d.Edges[no][4]: tk.messagebox.askokcancel(title='错误', message='城市已连接！') else: if self.user_way[-1] == int(d.Nodes[n1][1])-1: global MODE MODE = 1 self.user_way.append(int(d.Nodes[n2][1]) - 1) d.Edges[no][4] = True else: tk.messagebox.askokcancel(title='错误', message='未连贯！') G.draw() self.user_path_display() def revoke(self): try: if len(self.user_way) > 1: value1 = int(self.user_way[-1]) # 将值转换为整数 value2 = int(self.user_way[-2]) # 将值转换为整数 if value1 < value2: str_route = f'{value1+1}-{value2+1}' else: str_route = f'{value2+1}-{value1+1}' # print(str_route) for item in d.Edges: # print(item) if str_route == item[5]: # print(item) global MODE global entry1, default_value MODE = 1 self.user_way.pop() print(self.user_way) d.Edges[G.node_edge(value1, value2)][4] = False default_value.set(value2+1) entry1.config(textvariable=default_value) # 更新entry1的默认值 break G.draw() self.user_path_display() else: tk.messagebox.askokcancel(title='提示', message='未进行路径建设！') except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的城市编号！') def clear_all_paths(self): try: if len(self.user_way) > 1: for i in range(len(self.user_way)-1): value1 = int(self.user_way[-1]) # 将值转换为整数 value2 = int(self.user_way[-2]) # 将值转换为整数 if value1 < value2: str_route = f'{value1 + 1}-{value2 + 1}' else: str_route = f'{value2 + 1}-{value1 + 1}' # print(str_route) for item in d.Edges: # print(item) if str_route == item[5]: # print(item) global MODE global entry1, default_value MODE = 3 self.user_way.pop() print(self.user_way) d.Edges[G.node_edge(value1, value2)][4] = False default_value.set(value2 + 1) entry1.config(textvariable=default_value) # 更新entry1的默认值 break else: tk.messagebox.askokcancel(title='提示', message='未进行路径建设！') for i in range(len(d.Edges)): d.Edges[i][4] = False G.draw() self.user_path_display() except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的城市编号！') def greedy_path(self): g = self.check_user_greey() # print(g) # print(self.user_way) if g == -1: # 如果用户路径是贪心路径则显示 lab1 = tk.Label(frame_right, text="用户路径是贪心路径", bg=RED, fg=BLACK, font=('微软雅黑', 12)) lab1.place(relx=0.1, rely=0.8, width=180, height=30) elif g: self.servey_greedy_path() def servey_greedy_path(self): lab1 = tk.Label(frame_right, text="最短路径:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) lab1.place(relx=0.1, rely=0.75, width=80, height=30) # self.auto_ways() g_way = '' for a in s.ans: g_way += d.Nodes[a][1] g_way += "->" g_way += d.Nodes[0][1] lab2 = tk.Label(frame_right, text=f"{g_way}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab2.place(relx=0.1, rely=0.8, width=200, height=100) def shortest_path(self): best, user = self.check_user_best() # 计算最优路径与用户路径的总花费 if best == user: # 如果最优路径与用户路径的总花费相等，则说明用户路径是最短路径 lab1 = tk.Label(frame_right, text="用户路径是最短路径", bg=RED, fg=BLACK, font=('微软雅黑', 12)) lab1.place(relx=0.1, rely=0.8, width=180, height=30) else: # 否则说明用户路径不是最短路径 self.servey_path_display() # 用户路径展示 def user_path_display(self): lab1 = tk.Label(frame_right, text="用户路径:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) lab1.place(relx=0.1, rely=0.55, width=80, height=30) # self.auto_ways() way = "" # d_way = [] # for i in d.Nodes: # d_way.append(i[1]) for i in range(len(self.user_way)): if len(self.user_way)-i > 1: way += str(self.user_way[i]+1) way += "->" else: way += str(self.user_way[i]+1) lab2 = tk.Label(frame_right, text=f"{way}", bg=BLUE, fg=BLACK, font=('微软雅黑', 12), wraplength=200, justify='left') lab2.place(relx=0.1, rely=0.6, width=200, height=100) def way_operate(self): global MARK_NODE MARK_NODE = 0 # E.newedge_del_all(1) for w in frame_right.winfo_children(): w.destroy() # 清除右侧组件 self.user_way = [0] lab = tk.Label(frame_right, text="路径增加与删除", bg=RED, fg=BLACK, font=('微软雅黑', 20)) lab.place(relx=0.18, rely=0.02, width=200, height=30) y0 = 0.1 label1 = tk.Label(frame_right, text="节点1:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) label1.place(relx=0.00, rely=0.02 + y0, width=80, height=30) global entry3, entry4 # 创建第一个输入框 entry3 = tk.Entry(frame_right) entry3.place(relx=0.3, rely=0.02 + y0, width=50, height=30) label2 = tk.Label(frame_right, text="节点2:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) label2.place(relx=0.48, rely=0.02 + y0, width=100, height=30) # 创建第二个输入框 entry4 = tk.Entry(frame_right) entry4.place(relx=0.82, rely=0.02 + y0, width=50, height=30) label3 = tk.Label(frame_right, text="是否进行两节点之间操作:", bg=RED, fg=BLACK, font=('微软雅黑', 14)) label3.place(relx=0.1, rely=0.1 + y0, width=240, height=30) button = tk.Button(frame_right, text="线路增加", command=lambda: self.way_add()) button.place(relx=0.25, rely=0.18 + y0, width=160, height=30) button2 = tk.Button(frame_right, text="线路修改", command=lambda: self.way_change()) button2.place(relx=0.25, rely=0.28 + y0, width=160, height=30) button3 = tk.Button(frame_right, text="线路删除", command=lambda: self.way_delete()) button3.place(relx=0.25, rely=0.38 + y0, width=160, height=30) def way_add(self): global entry3, entry4 value1 = entry3.get() # 获取第一个输入框的值 value2 = entry4.get() # 获取第二个输入框的值 try: value1 = int(value1) # 将值转换为整数 value2 = int(value2) # 将值转换为整数 print("值1:", value1) print("值2:", value2) ways_list = [int(node[1]) for node in d.Nodes] # 判断两点是否连接 if abs(value2) > 10 and abs(value1) > 10: tk.messagebox.askokcancel(title='错误', message='请重新输入正确的编号！') return else: if value1 < value2: str_route = f'{value1}-{value2}' else: str_route = f'{value2}-{value1}' if d.drop.count(ways_list.index(value1)) > 4 or d.drop.count(ways_list.index(value2)) > 4: print(d.drop) tk.messagebox.askokcancel(title='提醒', message='当前节点已经到达最大连接数！') return no = 0 for item in d.Edges: if str_route == item[5]: no = G.node_edge(value1 - 1, value2 - 1) if d.Edges[no][3] == 1: tk.messagebox.askokcancel(title='提醒', message='当前路线已存在！') return top = creat_window(str_route+'路径参数配置') # 创建弹出窗口 distance_entry = create_input_box(top, "路程/公里：（1-99）", 101) # 创建一个输入框，获取卷积核大小 time_entry = create_input_box(top, "时间/h：（1-99）", 101) # 创建一个输入框，获取卷积步长 def get_input(): # 创建 # 获取输入框的内容 try: result1 = int(distance_entry.get()) result2 = int(time_entry.get()) print(result1,result2) if 100 > result1 > 0 and 0 < result2 < 100: d.Edges[no][3] = 1 d.Edges[no][6] = result1 d.Edges[no][7] = result2 top.destroy() # 关闭窗口 G.draw() # for i in d.Edges: # print(i) else: tk.messagebox.askokcancel(title='错误', message='路程与时间范围错误！') except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的路程与时间！') button = tk.Button(top, text="获取信息", command=get_input) button.pack() except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的城市编号！') def way_change(self): global entry3, entry4 value1 = entry3.get() # 获取第一个输入框的值 value2 = entry4.get() # 获取第二个输入框的值 try: value1 = int(value1) # 将值转换为整数 value2 = int(value2) # 将值转换为整数 print("值1:", value1) print("值2:", value2) ways_list = [int(node[1]) for node in d.Nodes] # 判断两点是否连接 if abs(value2) > 10 and abs(value1) > 10: tk.messagebox.askokcancel(title='错误', message='请重新输入正确的编号！') return else: if value1 < value2: str_route = f'{value1}-{value2}' else: str_route = f'{value2}-{value1}' no = 0 for item in d.Edges: if str_route == item[5]: no = G.node_edge(value1 - 1, value2 - 1) if d.Edges[no][3] == 0: tk.messagebox.askokcancel(title='提醒', message='当前路线不存在！') return top = creat_window(str_route + '路径参数修改') # 创建弹出窗口 distance_entry = create_input_box(top, "路程/公里：（1-99）", d.Edges[no][6]) # 创建一个输入框，获取卷积核大小 time_entry = create_input_box(top, "时间/h：（1-99）", d.Edges[no][7]) # 创建一个输入框，获取卷积步长 def get_input(): # 创建 # 获取输入框的内容 try: result1 = int(distance_entry.get()) result2 = int(time_entry.get()) print(result1, result2) if 100 > result1 > 0 and 0 < result2 < 100: d.Edges[no][6] = result1 d.Edges[no][7] = result2 top.destroy() # 关闭窗口 G.draw() # for i in d.Edges: # print(i) else: tk.messagebox.askokcancel(title='错误', message='路程与时间范围错误！') except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的路程与时间！') button = tk.Button(top, text="获取信息", command=get_input) button.pack() except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的城市编号！') def way_delete(self): global entry3, entry4 value1 = entry3.get() # 获取第一个输入框的值 value2 = entry4.get() # 获取第二个输入框的值 try: value1 = int(value1) # 将值转换为整数 value2 = int(value2) # 将值转换为整数 print("值1:", value1) print("值2:", value2) ways_list = [int(node[1]) for node in d.Nodes] # 判断两点是否连接 if abs(value2) > 10 and abs(value1) > 10: tk.messagebox.askokcancel(title='错误', message='请重新输入正确的编号！') return else: if value1 < value2: str_route = f'{value1}-{value2}' else: str_route = f'{value2}-{value1}' no = 0 for item in d.Edges: if str_route == item[5]: no = G.node_edge(value1 - 1, value2 - 1) if d.Edges[no][3] == 0: tk.messagebox.askokcancel(title='提醒', message='当前路线不存在！') return d.Edges[no][3] = 0 d.Edges[no][6] = 111 d.Edges[no][7] = 111 G.draw() except ValueError: tk.messagebox.askokcancel(title='错误', message='请输入正确的城市编号！') class Node: # 标记节点按钮 def node_mark_display(self): global MARK_NODE,Specify_Nodes W.user_way = [0] MARK_NODE = 1 for w in frame_right.winfo_children(): w.destroy() # 清除右侧组件 curx = -1 y0 = 0.1 for i in d.Nodes: if i[2] == 1: curx = i[0] lab = tk.Label(frame_right, text="节点操作", bg=RED, fg=BLACK, font=('微软雅黑', 20)) lab.place(relx=0.18, rely=0.02, width=200, height=30) lab1 = tk.Label(frame_right, text=f"当前共有{len(d.Nodes)}个节点", bg=RED, fg=BLACK, font=('微软雅黑', 15)) lab1.place(relx=0.1, rely=0.0 + y0, width=200, height=30) lab1 = tk.Label(frame_right, text="当前节点:", bg=RED, fg=BLACK, font=('微软雅黑', 12)) if curx == -1: text = "未选择" else: text = f"{d.Nodes[curx][1]}" Specify_Nodes = str(d.Nodes[curx][1]) lab2 = tk.Label(frame_right, text=text, bg=BLUE, fg=BLACK, font=('微软雅黑', 12)) lab1.place(relx=0.1, rely=0.05 + y0, width=80, height=30) lab2.place(relx=0.4, rely=0.05 + y0, width=80, height=30) B = tk.Button(frame_right, text="删除当前节点", command=lambda: self.node_del_exact()) B.place(relx=0.1, rely=0.1 + y0, width=150, height=30) B = tk.Button(frame_right, text="增加一个节点", command=lambda: self.node_add()) B.place(relx=0.1, rely=0.25 + y0, width=150, height=30) B = tk.Button(frame_right, text="减少一个节点", command=lambda: self.node_del()) B.place(relx=0.1, rely=0.3 + y0, width=150, height=30) def node_add(self): global d,Edges_list for i in range(len(d.Edges)): d.Edges[i][4] = False Edges_list = [] Edges_list = d.Edges # print(len(Edges_list)) # 增加一个节点 nodes = len(d.Nodes) + 1 # 如果节点数目大于五，则将连接详细信息改为不显示 if nodes > 5: d.edgeinfo = 0 if nodes > 10: tk.messagebox.askokcancel(title='错误', message='已达到最大节点') return d = Data(1, nodes) # for i in d.Edges: # print(i) G.draw() self.node_mark_display() def node_del(self): global d,Edges_list for i in range(len(d.Edges)): d.Edges[i][4] = False Edges_list = [] Edges_list = d.Edges # print(len(Edges_list)) if d.nodes_num <= 2: tk.messagebox.askokcancel(title='错误', message='节点过少') return nodes = len(d.Nodes) - 1 if nodes < 6: d.edgeinfo = 1 d = Data(2, nodes) # for i in d.Edges: # print(i) G.draw() self.node_mark_display() def node_del_exact(self): global d,Edges_list for i in range(len(d.Edges)): d.Edges[i][4] = False Edges_list = d.Edges if len(Specify_Nodes) == 3: tk.messagebox.askokcancel(title='错误', message='未选择节点') return if d.nodes_num <= 2:# 若删除节点以后节点过少，则直接返回并警告 tk.messagebox.askokcancel(title='错误', message='节点过少') return nodes = len(d.Nodes) - 1 if nodes < 6: d.edgeinfo = 1 d = Data(3, nodes) # for i in d.Edges: # print(i) G.draw() self.node_mark_display() def refresh(): node = random.randint(4, 10) global d, MODE,MARK_NODE MARK_NODE = 0 MODE = 0 d = Data(0, node) print(node) W.user_way=[0] G.draw() W.way_route_markings() def graph_init(): global MODE,MARK_NODE MARK_NODE = 0 MODE = 0 G.draw() W.way_route_markings() def empty(): global MARK_NODE MARK_NODE = 0 G.delete_all() def auto_opera_mu(menu: tk.Menu): """TSP问题自动生成与自动求解相关""" menu_node = Menu(menu, tearoff=False) menu_node.add_command(label="自动随机产生一个TSP问题", command=lambda: graph_init()) menu_node.add_command(label="重新生成", command=lambda: refresh()) menu_node.add_command(label="清空", command=lambda: empty()) menu_node.add_command(label="自动求解最优路径-dfs遍历", command=lambda: s.node_dfs()) menu_node.add_command(label="自动求解优化路径-贪心", command=lambda: s.node_greedy()) menu_node.add_command(label="检查是否存在路径", command=lambda: s.check()) # 在主目录菜单上新增"文件"选项，并通过menu参数与下拉菜单绑定 menu.add_cascade(label="TSP问题自动生成与自动求解相关", menu=menu_node) # 创建弹出窗口 def creat_window(title): top = tk.Toplevel(root) top.geometry("300x350") top.title(title) return top # 输入框 def create_input_box(top, text, value): box_label = tk.Label(top, text=text) box_label.pack(padx=10, pady=10) box_size = tk.IntVar(top, value=value) # 创建一个IntVar对象，并设置默认值为3 box_size_entry = tk.Entry(top, textvariable=box_size) # 关联IntVar对象 box_size_entry.pack(padx=20, pady=20) return box_size_entry def dir2(x, y): # 计算两点间距离 return (x[0] - y[0]) * (x[0] - y[0]) + (x[1] - y[1]) * (x[1] - y[1]) def left1(event): # 查找鼠标左键按下时位置是否在某个节点内 n = -1 for node in d.Nodes: if dir2([event.x, event.y], [node[3][0], node[3][1]]) < 20 * 20: n = node[0] # n为点击的节点，若没有则为-1 m = mode.get() # 查看现在操作类型 if m == 1 and not n == -1: # 节点操作 # N.node_mark(n) if n >= len(d.Nodes): # 弹出对话框 tk.messagebox.askokcancel(title='错误', message='该节点不存在') return for node in d.Nodes: if node[0] == n: node[2] = 1 else: node[2] = 0 G.draw() if MARK_NODE == 1: N.node_mark_display() elif MARK_NODE == 2: pass print(n) def node_opera_mu(menu: tk.Menu): """节点操作""" menu_node = Menu(menu, tearoff=False) menu_node.add_command(label="添加一个节点", command=lambda: N.node_add()) menu_node.add_command(label="删除一个节点", command=lambda: N.node_del()) menu_node.add_command(label="选择一个节点", command=lambda: N.node_mark_display()) # 在主目录菜单上新增"文件"选项，并通过menu参数与下拉菜单绑定 menu.add_cascade(label="节点操作", menu=menu_node) def path_opera_mu(menu: tk.Menu): """路径操作""" menu_node = Menu(menu, tearoff=False) menu_node.add_command(label="路径选择", command=lambda: W.way_route_markings()) menu_node.add_command(label="路径修改", command=lambda: W.way_operate()) # 在主目录菜单上新增"文件"选项，并通过menu参数与下拉菜单绑定 menu.add_cascade(label="路径操作", menu=menu_node, ) if __name__ == '__main__': G = Graph() mode = tk.IntVar(value=1) main_menu = tk.Menu(root) root.config(menu=main_menu) cv.bind('', left1) E = Edge() W = Ways() s = Solve() N = Node() auto_opera_mu(main_menu) node_opera_mu(main_menu) path_opera_mu(main_menu) W.way_route_markings() root.mainloop()