classSolution: deftrap(self, height: List[int]) -> int: height_len = len(height) if height_len < 3: return0 max_value = 0 max_index = 0 for key,value inenumerate(height): if value > max_value: max_value = value max_index = key # 面积 area = 0 # left -> max part_max = height[0] for left inrange(max_index): if height[left] > part_max: part_max = height[left] else: area += part_max - height[left] # right -> max part_max = height[height_len-1] for right inrange(height_len-1,max_index,-1): if height[right] > part_max: part_max = height[right] else: area += part_max - height[right] return area
班上有 N 名学生。其中有些人是朋友,有些则不是。他们的友谊具有是传递性。如果已知 A 是 B 的朋友,B 是 C 的朋友,那么我们可以认为 A 也是 C 的朋友。所谓的朋友圈,是指所有朋友的集合。给定一个 N * N 的矩阵 M,表示班级中学生之间的朋友关系。如果M[i][j] = 1,表示已知第 i 个和 j 个学生互为朋友关系,否则为不知道。你必须输出所有学生中的已知的朋友圈总数。
classSolution: deffindCircleNum(self, M: List[List[int]]) -> int: len_m = len(M) self.parent = [i for i inrange(len_m)] # init deffind_set(i): if i == self.parent[i]: return i return find_set(self.parent[i]) defunion(i,j): find_i, find_j = find_set(i),find_set(j) self.parent[find_j] = find_i for i inrange(len_m): for j inrange(len_m): if M[i][j]: union(i,j) returnlen([i for i inrange(len_m) if find_set(i) == i])