目錄
547. 朋友圈
班上有 N 名學生。其中有些人是朋友,有些則不是。他們的友誼具有是傳遞性。如果已知 A 是 B 的朋友,B 是 C 的朋友,那麼我們可以認爲 A 也是 C 的朋友。所謂的朋友圈,是指所有朋友的集合。
給定一個 N * N 的矩陣 M,表示班級中學生之間的朋友關係。如果M[i][j] = 1,表示已知第 i 個和 j 個學生互爲朋友關係,否則爲不知道。你必須輸出所有學生中的已知的朋友圈總數。
示例 1:
輸入:
[[1,1,0],
[1,1,0],
[0,0,1]]
輸出: 2
說明:已知學生0和學生1互爲朋友,他們在一個朋友圈。
第2個學生自己在一個朋友圈。所以返回2。
示例 2:
輸入:
[[1,1,0],
[1,1,1],
[0,1,1]]
輸出: 1
說明:已知學生0和學生1互爲朋友,學生1和學生2互爲朋友,所以學生0和學生2也是朋友,所以他們三個在一個朋友圈,返回1。
注意:
N 在[1,200]的範圍內。
對於所有學生,有M[i][i] = 1。
如果有M[i][j] = 1,則有M[j][i] = 1。
class Solution {
private void dfs(int[][] a, boolean[] visited, int i) {
//對 i 遍歷所有的 a[i][j],
//遞歸出口是全部 j 都不滿足if條件
for(int j = 0; j < a.length; j++) {
if(a[i][j] == 1 && !visited[j]) {
//標記
visited[j] = true;
//遞歸接力
dfs(a, visited, j);
}
}
}
public int findCircleNum(int[][] M) {
//記錄訪問過的元素
boolean[] visited = new boolean[M.length];
int res = 0;
for(int i = 0; i < M.length; i++) {
if(!visited[i]) {
//一個dfs生成一個朋友圈,計數一次
dfs(M, visited, i);
res++;
}
}
return res;
}
}
662. 二叉樹最大寬度
給定一個二叉樹,編寫一個函數來獲取這個樹的最大寬度。樹的寬度是所有層中的最大寬度。這個二叉樹與滿二叉樹(full binary tree)結構相同,但一些節點爲空。
每一層的寬度被定義爲兩個端點(該層最左和最右的非空節點,兩端點間的null節點也計入長度)之間的長度。
示例 1:
輸入:
1
/ \
3 2
/ \ \
5 3 9
輸出: 4
解釋: 最大值出現在樹的第 3 層,寬度爲 4 (5,3,null,9)。
示例 2:
輸入:
1
/
3
/ \
5 3
輸出: 2
解釋: 最大值出現在樹的第 3 層,寬度爲 2 (5,3)。
示例 3:
輸入:
1
/ \
3 2
/
5
輸出: 2
解釋: 最大值出現在樹的第 2 層,寬度爲 2 (3,2)。
示例 4:
輸入:
1
/ \
3 2
/ \
5 9
/ \
6 7
輸出: 8
解釋: 最大值出現在樹的第 4 層,寬度爲 8 (6,null,null,null,null,null,null,7)。
注意: 答案在32位有符號整數的表示範圍內。
class Solution {
//想到帶下標的層次遍歷
public int widthOfBinaryTree(TreeNode root) {
if(root == null) {
return 0;
}
//節點序號,左子節點2 * num - 1,右子節點2 * num
int num = 1;
int depth = 1;
//保存每一層的最左序號和最右序號
ArrayList<Integer> left = new ArrayList<Integer>();
left.add(1);
ArrayList<Integer> right = new ArrayList<Integer>();
right.add(1);
preOrder(root, num, depth, left, right);
//求每層的左右最大差值即可
int maxWidth = 1;
for(int i = 0; i < left.size(); i++) {
maxWidth = Math.max(maxWidth, right.get(i) - left.get(i) + 1);
}
return maxWidth;
}
//這題應該用先序纔對,這樣可以先把list長度擴到樹深度,不會丟失節點
public void preOrder(TreeNode root, int num, int depth,
ArrayList<Integer> left, ArrayList<Integer> right) {
if(root == null) {
return;
}
//list還沒擴到當前樹深度,就先add新節點
if(left.size() < depth) {
left.add(num);
right.add(num);
}
//否則比較更新即可
else {
if(num < left.get(depth - 1)) {
left.set(depth - 1, num);
}
if(num > right.get(depth - 1)) {
right.set(depth - 1, num);
}
}
preOrder(root.left, 2 * num - 1, depth + 1, left, right);
preOrder(root.right, 2 * num, depth + 1, left, right);
}
}