LeetCode#1024求根到叶子节点数字之和
题目链接
一道水题,主要是记录一下string居然也可以使用push_back() & pop_back()菜,以及stoi、atoi函数(似乎并不是标准库中的函数,功能是将字符串转换成整数),stoi可以直接传入string类型的对象,而atoi只能传入char类型的数组。
附代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
|
class Solution { public: void dfs(TreeNode* node,int &ans,string &temp) { if(node -> left == nullptr && node -> right == nullptr) { temp.push_back(node->val+'0'); int num = stoi(temp); ans+=num; return; } temp.push_back(node->val+'0'); if(node->left!=nullptr) { dfs(node->left,ans,temp); temp.pop_back(); } if(node->right!=nullptr) { dfs(node->right,ans,temp); temp.pop_back(); } } int sumNumbers(TreeNode* root) { if(root == nullptr) return 0; int ans = 0; string temp; dfs(root,ans,temp); return ans; } };
|