拓扑排序判断图是否有环(Python)

class Stack:
    def __init__(self):
        self.arr = []

    def push(self, item):
        self.arr.append(item)

    def pop(self):
        return self.arr.pop(-1)

    def getTop(self):
        return self.arr[-1]

    def __len__(self):
        return len(self.arr)


def topologicalSort(matrix
                    # : list[list[int]]
                    ) -> [0, 1]:
    N = len(matrix)
    ind = []
    stack = Stack()
    count = 0
    for i in range(N):
        ind.append( sum(row[i] for row in matrix))
        if ind[i] == 0:
            stack.push(i)
    while len(stack) > 0:
        i = stack.pop()
        count += 1
        for j in range(N):
            if matrix[i][j] == 1:
                ind[j] -= 1
                if ind[j] == 0:
                    stack.push(j)
    if count == N:
        return 0
    else:
        return 1


try:
    t = int(input())
    matrixs = []
    while (t > 0):
        matrix = []
        t -= 1
        n = int(input())
        while (n > 0):
            n -= 1
            snums = input().split()
            nums = []
            for snum in snums:
                nums.append(int(snum))
            matrix.append(nums)
        matrixs.append(matrix)
    for matrix in matrixs:
        print(topologicalSort(matrix))

except:
    pass

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值