python判断无向图环是否存在的示例

暂时是一个手动设置无向图中的边，用一个二维数组表示，后面会改进为用户自己定义无向图的边。

学习python的新手，若大佬有解决的办法，希望不吝赐教

#无向图判断环是否存在
def dfs(u,fa):
 for i in range(v):
   n=g[u][i]#n为图中的顶点数
   # print(u,n,fa,i,'')
   if n in vertex:#判断n是否属于图的顶点
     if n==fa:
       continue
     if visit[n]==0:
       visit[n]=1
       if dfs(n,u)==1:
         return 1
     else:
       return 1
 return 0
global flag,visit,v,g,e,vertex
vertex=[]
flag=0
v=4;e=4#v是顶点，e是边数
#v,e=map(int,input().split())
vertex=[1,2,3,4]#无向图的顶点集合
g=[[0 for i in range(20)]for i in range(20)]
visit=[0 for i in range(20)]
g[1][0]=2;g[2][0]=1;g[2][1]=3;g[3][0]=2;g[3][1]=4;g[4][0]=3;g[1][1]=3;g[3][2]=1
for i in range(1,v+1):
 if visit[i]==0:
   visit[i]=1
   if dfs(i,-1):
     flag=1
     break
if flag==1:
 print('Yes, there is A loop in this network')
else:
 print('No loop')

来源：https://blog.csdn.net/qq_37163152/article/details/84980004