类型一:邻接表

题目一:员工的重要性

给定一个保存员工信息的数据结构,它包含了员工唯一的id重要度直系下属的id

比如,员工1是员工2的领导,员工2是员工3的领导。他们相应的重要度为15, 10, 5。那么员工1的数据结构是[1, 15, [2]],员工2的数据结构是[2, 10, [3]],员工3的数据结构是[3, 5, []]。注意虽然员工3也是员工1的一个下属,但是由于并不是直系下属,因此没有体现在员工1的数据结构中。

现在输入一个公司的所有员工信息,以及单个员工id,返回这个员工和他所有下属的重要度之和。

示例 1:

输入: [[1, 5, [2, 3]], [2, 3, []], [3, 3, []]], 1
输出: 11
解释:
员工1自身的重要度是5,他有两个直系下属2和3,而且2和3的重要度均为3。因此员工1的总重要度是 5 + 3 + 3 = 11。

注意:

  1. 一个员工最多有一个直系领导,但是可以有多个直系下属
  2. 员工数量不超过2000。

  【注意点:应使用局部变量(weight)记录结果,不能使用全局变量】

"""
# Employee info
class Employee(object):
def __init__(self, id, importance, subordinates):
self.id = id
self.importance = importance
self.subordinates = subordinates
"""
class Solution(object):
def getImportance(self, employees, id):
"""
:type employees: Employee
:type id: int
:rtype: int
"""
if not employees:
return 0
##局部变量
weight = 0
for item in employees:
if item.id == id:
weight += item.importance
for j in item.subordinates:
weight += self.getImportance(employees, j)
return weight

题目二:钥匙和房间

N 个房间,开始时你位于 0 号房间。每个房间有不同的号码:0,1,2,...,N-1,并且房间里可能有一些钥匙能使你进入下一个房间。

在形式上,对于每个房间 i 都有一个钥匙列表 rooms[i],每个钥匙 rooms[i][j][0,1,...,N-1] 中的一个整数表示,其中 N = rooms.length。 钥匙 rooms[i][j] = v 可以打开编号为 v 的房间。

最初,除 0 号房间外的其余所有房间都被锁住。

你可以自由地在房间之间来回走动。

如果能进入每个房间返回 true,否则返回 false

示例 1:

输入: [[1],[2],[3],[]]
输出: true
解释:
我们从 0 号房间开始,拿到钥匙 1。
之后我们去 1 号房间,拿到钥匙 2。
然后我们去 2 号房间,拿到钥匙 3。
最后我们去了 3 号房间。
由于我们能够进入每个房间,我们返回 true。

示例 2:

输入:[[1,3],[3,0,1],[2],[0]]
输出:false
解释:我们不能进入 2 号房间。

提示:

  1. 1 <= rooms.length <= 1000
  2. 0 <= rooms[i].length <= 1000
  3. 所有房间中的钥匙数量总计不超过 3000

每个房间找钥匙,找到则到下一个钥匙对应的房间DFS。采用一个visited列表存储是否到达过这个房间,最后如果所有房间都达到过则返回True。

def canVisitAllRooms(rooms):
if not rooms:
return True
n = len(rooms)
visited = [False] * n
visited[0] = True
def dfs(rooms,keys,visited):
for key in keys:
if not visited[key]:
visited[key] = True
dfs(rooms,rooms[key],visited)
dfs(rooms,rooms[0],visited)
return all(visited)

类型二:DAG拓扑排序

题目一:课程安排207

题目二:课程安排210

现在你总共有 n 门课需要选,记为 0n-1

在选修某些课程之前需要一些先修课程。 例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示他们: [0,1]

给定课程总量以及它们的先决条件,返回你为了学完所有课程所安排的学习顺序。

可能会有多个正确的顺序,你只要返回一种就可以了。如果不可能完成所有课程,返回一个空数组。

示例 1:

输入: 2, [[1,0]]
输出: [0,1]
解释: 总共有 2 门课程。要学习课程 1,你需要先完成课程 0。因此,正确的课程顺序为 [0,1] 。

示例 2:

输入: 4, [[1,0],[2,0],[3,1],[3,2]]
输出: [0,1,2,3] or [0,2,1,3]
解释: 总共有 4 门课程。要学习课程 3,你应该先完成课程 1 和课程 2。并且课程 1 和课程 2 都应该排在课程 0 之后。
因此,一个正确的课程顺序是 [0,1,2,3] 。另一个正确的排序是 [0,2,1,3] 。

说明:

  1. 输入的先决条件是由边缘列表表示的图形,而不是邻接矩阵。详情请参见图的表示法
  2. 你可以假定输入的先决条件中没有重复的边。

提示:

  1. 这个问题相当于查找一个循环是否存在于有向图中。如果存在循环,则不存在拓扑排序,因此不可能选取所有课程进行学习。
  2. 通过 DFS 进行拓扑排序 - 一个关于Coursera的精彩视频教程(21分钟),介绍拓扑排序的基本概念。
  3. 拓扑排序也可以通过 BFS 完成。

  1、建立图

  2、循环n次,每次是遍历一个节点是否已经visited且合法地加入path中了,如果False不合法则直接返回【】。

  3、遍历一个节点时会将其后面的所有子节点都处理掉。

from collections import defaultdict
def findPath(n,arr):
if n == 0:
return []
graph = defaultdict(list)
for u , v in arr:
graph[v].append(u)
# 0为Unkown,1为visiting,2为visited
path = []
visited = [0] * n
for i in range(n):
if not DFS(graph,visited,path,i):
return []
return path[::-1]
def DFS(graph,visited,path,i):
####i节点:其正在遍历,但它的子节点的子节点也是它,表示产生了有环,则return FALSE
if visited[i] == 1: return False
####i节点 :已经遍历过,后面已经没有节点了,return true
elif visited[i] == 2:return True
####表示正在遍历i节点
visited[i] = 1
for j in graph[i]:
if not DFS(graph,visited,path,j):
return False
path.append(i)
visited[i] = 2
return True n = 5
arr = [[1,0],[2,0],[3,1],[3,2],[4,0]]
print(findPath(n,arr))

题目三:解题报告,连除----399

已经给出了某些变量的比值,求新的变量的比值。如果这个变量没有出现过,或者不可到达,那么返回-1.

 

题目中给了顶点和顶点之间的关系,其实就是制定了这个图的样子。然后要求的新的比值其实就是从一个顶点到达另外一个顶点的路径,并且把这条路径上所有的权重相乘。

注意,如果a/b=3,那么从a到b是3,那么从b到a是1/3.

既然是从一个顶点出发到达另外一个顶点,所以应该是dfs解决的问题。

原文:https://blog.csdn.net/fuxuemingzhu/article/details/82591165

  1、建立图  {a:{b : 2.0} 、b:{a:1 /2.0,c:3.0}、c:{b:1/3.0}}

  2、不在图中则返回-1  

  3、在图中,x == y,返回1,x != y,返回x到y的拓扑排序的权重相乘值。



  

from collections import defaultdict
def solveque(arr ,values , que):
if not arr:
return [-1] * len(que)
if not que:
return []
graph = defaultdict(dict)
for (x,y) , value in zip(arr,values):
graph[x][y] = value
graph[y][x] = 1/value if value else 0
for x,y in que:
if x in graph and y in graph:
return dfs(graph,x,y,set())
else:
return -1.0
def dfs(graph,x,y,visited):
if x == y:
return 1.0
visited.add(x)
for k in graph[x]:
if k in visited:continue
visited.add(k)
d = dfs(graph,k,y,visited)
if d > 0:
return d*graph[x][k]
return -1.0
arr = [['a','b'],['b','c']]
values = [2.0,3.0]
que = [['a','c'],['a','v']]
print(solveque(arr ,values , que))
05-11 15:54