用python写一个图的算法

能不能用python3写一下，还得需要注意一下那个N 的范围和一些runtime error