力扣练习——43 路径总和

43 路径总和

1.问题描述
给定一个二叉树和一个目标和，判断该树中是否存在根节点到叶子节点的路径，这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

示例:

给定如下二叉树，以及目标和 sum = 22，

          5

         / \

        4   8

       /    /  \

     11  13  4

    /  \      \

   7    2      1

返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

可使用以下main函数：

#include

#include

#include

#include

using namespace std;

struct TreeNode

{

int val;

TreeNode *left;

TreeNode *right;

TreeNode() : val(0), left(NULL), right(NULL) {}

TreeNode(int x) : val(x), left(NULL), right(NULL) {}

TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}

};

TreeNode* inputTree()

{

int n,count=0;

char item[100];

cin>>n;

if (n==0)

    return NULL;

cin>>item;

TreeNode* root = new TreeNode(atoi(item));

count++;

queue<TreeNode*> nodeQueue;

nodeQueue.push(root);

while (count<n)

{

    TreeNode* node = nodeQueue.front();

    nodeQueue.pop();

    cin>>item;

    count++;

    if (strcmp(item,"null")!=0)

    {

        int leftNumber = atoi(item);

        node->left = new TreeNode(leftNumber);

        nodeQueue.push(node->left);

    }

    if (count==n)

        break;

    cin>>item;

    count++;

    if (strcmp(item,"null")!=0)

    {

        int rightNumber = atoi(item);

        node->right = new TreeNode(rightNumber);

        nodeQueue.push(node->right);

    }

}

return root;

}

int main()

{

TreeNode* root;

root=inputTree();

int sum;

cin>>sum;

bool res=Solution().hasPathSum(root,sum);

cout<<(res?"true":"false");

}
2.输入说明
首先输入结点的数目n（注意，这里的结点包括题中的null空结点）

然后输入n个结点的数据，需要填充为空的结点，输入null。

最后输入一个整数sum。
3.输出说明
输出true或false
4.范例
输入
13
5 4 8 11 null 13 4 7 2 null null null 1
22
输出
true
5.代码

#include <iostream>

#include <queue>

#include <cstdlib>

#include <cstring>

using namespace std;

struct TreeNode

{

	int val;

	TreeNode *left;

	TreeNode *right;

	TreeNode() : val(0), left(NULL), right(NULL) {}

	TreeNode(int x) : val(x), left(NULL), right(NULL) {}

	TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}

};

TreeNode* inputTree()

{

	int n, count = 0;

	char item[100];

	cin >> n;

	if (n == 0)

		return NULL;

	cin >> item;

	TreeNode* root = new TreeNode(atoi(item));

	count++;

	queue<TreeNode*> nodeQueue;

	nodeQueue.push(root);

	while (count < n)

	{

		TreeNode* node = nodeQueue.front();

		nodeQueue.pop();

		cin >> item;

		count++;

		if (strcmp(item, "null") != 0)

		{

			int leftNumber = atoi(item);

			node->left = new TreeNode(leftNumber);

			nodeQueue.push(node->left);

		}

		if (count == n)

			break;

		cin >> item;

		count++;

		if (strcmp(item, "null") != 0)

		{

			int rightNumber = atoi(item);

			node->right = new TreeNode(rightNumber);

			nodeQueue.push(node->right);

		}

	}

	return root;

}

bool hasPathSum(TreeNode* root, int sum)
{
	if (!root)return false;
	//记录结点
	queue<TreeNode*>q;
	//记录已经遍历得到结点相加值
	queue<int>q_val;
    //接下来根节点入队，别忘记
	q.push(root);
	q_val.push(root->val);
	//遍历
	while (!q.empty())
	{
		TreeNode *node = q.front();//队头元素
		q.pop();//出队
		int node_val = q_val.front();
		q_val.pop();
		if (node->left == NULL && node->right == NULL)//叶子结点
		{
			if (node_val == sum)
				return true;
			continue;//若不满足,跳过
		}
		//左子树非空
		if (node->left != NULL)
		{
			q.push(node->left);//左节点入队
			q_val.push(node->left->val + node_val);//更新队列元素值
		}
		//右子树非空
		if (node->right != NULL)
		{
			q.push(node->right);//左节点入队
			q_val.push(node->right->val + node_val);//更新队列元素值
		}

	}
	return false;
}

int main()

{

	TreeNode* root;

	root = inputTree();

	int sum;

	cin >> sum;

	bool res = hasPathSum(root, sum);

	cout << (res ? "true" : "false");

}