■ 题目描述
【跳格子】
地上共有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')
标签:格子,示例,nums,开启,dic,面试,算法,steps
From: https://blog.csdn.net/YW2019/article/details/137197753