常见面试算法题-跳格子

■ 题目描述

【跳格子】

地上共有N个格子，你需要跳完地上所有的格子，但是格子间是有强依赖关系的，跳完前一个格子后，后续的格子才会被开启，格子间的依赖关系由多组steps数组给出，steps[0]表示前一个格子，steps[1]表示steps[0]可以开启的格子:

比如[0,1]表示从跳完第0个格子以后第1个格子就开启了，比如[2,1]，[2,3]表示跳完第2个格子后第1个格子和第3个格子就被开启了。

请你计算是否能由给出的steps数组跳完所有的格子，如果可以输出yes，否则输出no。

说明：

1.你可以从一个格子跳到任意一个开启的格子

2.没有前置依赖条件的格子默认就是开启的

3.如果总数是N，则所有的格子编号为[0,1,2,3…N-1]连续的数组

输入描述:

输入一个整数N表示总共有多少个格子，接着输入多组二维数组steps表示所有格子之间的依赖关系。

输出描述:

如果能按照steps给定的依赖顺序跳完所有的格子输出yes，

否则输出no。

输入

3

0 1

0 2

输出

yes

说明

总共有三个格子[0,1,2]，跳完0个格子后第1个格子就开启了，跳到第0个格子后第2个格子也被开启了，按照0->1->2或者0->2->1的顺序都可以跳完所有的格子

示例2   输入输出示例仅供调试，后台判题数据一般不包含示例

输入

2

1 0

0 1

输出

no

说明

总共有2个格子，第1个格子可以开启第0格子，但是第1个格子又需要第0个格子才能开启，相互依赖，因此无法完成

示例3   输入输出示例仅供调试，后台判题数据一般不包含示例

输入

6

0 1

0 2

0 3

0 4

0 5

输出

yes

说明

总共有6个格子，第0个格子可以开启第1,2,3,4,5个格子，所以跳完第0个格子之后其他格子都被开启了，之后按任何顺序可以跳完剩余的格子

示例4   输入输出示例仅供调试，后台判题数据一般不包含示例

输入

5

4 3

0 4

2 1

3 2

输出

yes

说明

跳完第0个格子可以开启格子4，跳完格子4可以开启格子3，跳完格子3可以开启格子2，跳完格子2可以开启格子1，按照0->4->3->2->1这样就跳完所有的格子

以下代码为本人原创，可以供大家参考，若有不足之处，感谢指出！！！！

n = int(input())
dic = dict()
while True:
    nums = list(map(int, input().split()))
    if len(nums) != 0:
        if nums[0] in dic:
            dic[nums[0]].append(nums[1])
        else:
            dic[nums[0]] = [nums[1]]
    else:
        break
cur = []
for i in dic.values():
    cur += i
cur = set(cur)
x = []
for i in range(n):
    if i not in cur:
        x.append(i)
if len(x) == 0:
    print('NO')
else:
    stack = []
    ans = []
    for i in x:
        ans.append(i)
        stack += dic[i]
        stack = list(set(stack))
    while stack:
        stack1 = []
        for j in stack:
            if j not in ans:
                stack1 += dic.get(j, [])
                ans.append(j)
        stack1 = list(set(stack1))
        stack = stack1
    if len(ans) == n:
        print('YES')
    else:
        print('NO')