判断有向图是否有环（python）

2020英语流利说算法笔试题

没做出来，结束了学习一下。

思路是记下每个节点的入和出的数量，不断删去只出不入（或只入不出）的边，直到不能继续删除，再观察此时是否还剩有有度的节点。

因为考完才写的，没有测试过能过多少。

tip:一开始if mat_i[i] == 0 and mat_o[i]!= 0:漏了后面一半条件，进入死循环。

 

import sys
''' 给定一个有向图，矩阵维数为顶点数，(i，j)表示i->j有边，求是否存在环
例：[[0, 1, 0], [0, 0, 1], [1, 0, 0]]——print(1)
[[0, 0, 0, 1, 0], [1, 0, 0, 0, 0], [0, 0, 0, 1, 1], 
[0, 0, 0, 0, 0], [0, 1, 0, 0, 0]]——print(0)
'''
for line in sys.stdin:
    lines = eval(line)
    if len(lines)<2:
        print(0)
    mat_i = [0]*len(lines)
    mat_o = [0]*len(lines)
    for i in range(len(lines)):
        for j in range(len(lines)):
            if lines[i][j]:
                mat_i[j] += 1
                mat_o[i] += 1

    while(1):
        flag = 0
        for i in range(len(lines)):
            if mat_i[i] == 0 and mat_o[i]!= 0:
                for j in range(len(lines)):
                    if lines[i][j]:
                        mat_i[j] -= 1
                mat_o[i] = 0
                flag = 1
        if flag == 0:
            break
    if any(mat_i):
        print(1)
    else:
        print(0)