Après avoir écrit une telle question sur Leetcode, je sens que ma réflexion est plutôt bonne, partagez-la:
Idée: Tout d'abord, ajoutez chaque valeur de nœud non nulle à un ensemble via une recherche en profondeur d'abord, puis parcourez l'ensemble pour trouver la valeur minimale de la différence.
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private int min = Integer.MAX_VALUE;
public int minDiffInBST(TreeNode root) {
List<Integer> valueList = new ArrayList<Integer>();
dfs(root,valueList);
for(int i = 0;i < valueList.size();i++){
for(int j = i + 1;j < valueList.size();j++){
min = Math.min(min,Math.abs(valueList.get(i)-valueList.get(j)));
}
}
return min;
}
public void dfs(TreeNode node,List<Integer> list){
if(node != null){
int a = node.val;
list.add(a);
dfs(node.left,list);
dfs(node.right,list);
}
}
}
Mais c'est évident: la boucle double for augmente considérablement la complexité temporelle!
Ensuite, nous devons réfléchir à la façon d'optimiser:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private int min = Integer.MAX_VALUE;
public int minDiffInBST(TreeNode root) {
List<Integer> valueList = new ArrayList<Integer>();
dfs(root,valueList);
/*for(int i = 0;i < valueList.size();i++){
for(int j = i + 1;j < valueList.size();j++){
min = Math.min(min,Math.abs(valueList.get(i)-valueList.get(j)));
}
}*/
Collections.sort(valueList);
for(int i = 0; i < valueList.size() - 1;i++){
min = Math.min(min,valueList.get(i+1)-valueList.get(i));
}
return min;
}
public void dfs(TreeNode node,List<Integer> list){
if(node != null){
int a = node.val;
list.add(a);
dfs(node.left,list);
dfs(node.right,list);
}
}
}
Jetons un coup d'œil à l'effet: