今天还是二叉树,全都是用递归实现的,感觉第二道题还是挺有难度的,回溯是第一次接触,感觉还是不太熟。
110.平衡二叉树
这一题要处理只有一个根节点的特殊情况。还有,如果left和right出现-1的话就要立刻返回-1,而不能等到后面相减处理,否则程序会认为left和right仍然没有结束。
class Solution {
public:
bool isBalanced(TreeNode* root) {
if(root == NULL) return true;
int height = getHeight(root);
return height > 0 ? true : false;
}
int getHeight(TreeNode* root){
if(root == NULL) return 0;
int left = getHeight(root -> left);
if(left == -1) return -1;
int right = getHeight(root -> right);
if(right == -1) return -1;
if(abs(left - right) > 1) return -1;
else return 1 + max(left, right);
}
};
257. 二叉树的所有路径
这道题感觉还是挺难的,我做的时候没啥思路,看讲解去了,这里有一个要注意的点:在弹出vector中的整数时,并非弹出当前节点存储的值,而是当前节点的孩子节点存储的值,这样才能保证最后留下根节点的值,然后继续遍历另一边。
class Solution {
public:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> result;
vector<int> path;
getPath(result, path, root);
return result;
}
void getPath(vector<string> &result, vector<int> &path, TreeNode* node){
path.push_back(node -> val); //将当前节点的值存入数组
if(node -> left == NULL && node -> right == NULL){ //当前节点已经是叶子节点
result.push_back(convert(path));
return ;
}
if(node -> left){ //左节点不为空
getPath(result, path, node -> left);
path.pop_back(); //这里弹出的是其叶子节点的值
}
if(node -> right){ //右节点不为空
getPath(result, path, node -> right);
path.pop_back(); //这里弹出的是其叶子节点的值
}
}
string convert(vector<int> &path){
string result;
for(int i : path){
result += to_string(i);
result += "->";
}
result.resize(result.size() - 2); //去除末尾的->
return result;
}
};
404.左叶子之和
说实话这个递归看的还是有点抽象,看着看着就把自己绕进去了,这题要注意的点就是对左叶子节点要额外加一个if判断语句,否则返回的是0,但对右叶子节点不需要加额外的if语句。
class Solution {
public:
int sumOfLeftLeaves(TreeNode* root) {
if(root == NULL) return 0;
if(root -> left == NULL && root -> right == NULL)
return 0;
int left = sumOfLeftLeaves(root -> left);
if(root -> left && root -> left -> left == NULL && root -> left -> right == NULL) //左叶子节点
left = root -> left -> val;
int right = sumOfLeftLeaves(root -> right);
return left + right;
}
};
222.完全二叉树的节点个数
这个简单,也是用递归做,如果节点是空姐点则返回0,如果不是空节点,就返回该节点的左子树中节点的个数+右子树中节点的个数。
class Solution {
public:
int countNodes(TreeNode* root) {
if(root == NULL) return 0;
int left = countNodes(root -> left);
int right = countNodes(root -> right);
return left + right + 1;
}
};