路径总和(C++)

112. 路径总和

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。

解题思路:看到这道题,首先觉得与二叉树的所有路径非常像,完全可以用二叉树的所有路径思路来解题。需要做修改的地方就是在获取路径的时候进行求和,最后所有路径和与目标和做比较。

代码:

class Solution {	//112. 路径总和
public:
	bool hasPathSum(TreeNode* root, int targetSum) {
		if (root == nullptr) return false;
		vector<int> allPathSum;

		_getPathSum(root, 0, allPathSum);
		for (int i = 0; i < allPathSum.size(); ++i) {
			if (allPathSum[i] == targetSum) {
				return true;
			}
		}
		return false;
	}
private:
	void _getPathSum(TreeNode* cur, int sum, vector<int>& allPathSum) {
		sum += cur->val;
		if (cur->left == nullptr && cur->right == nullptr) {
			allPathSum.push_back(sum);
			return;
		}
		if (cur->left) {
			_getPathSum(cur->left, sum, allPathSum);
		}
		if (cur->right) {
			_getPathSum(cur->right, sum, allPathSum);
		}
	}
};

进阶方法:当我看了代码随想录的解题方法,发现了我之前写的方法的问题。其实当遍历时,找到与目标和相同的路径值,就应该直接返回,我之前的方法是对所有路径都进行了遍历。

代码:

class Solution {	//112. 路径总和  进阶
public:
	bool hasPathSum(TreeNode* root, int targetSum) {
		if (root == nullptr) return false;
		if (root->left == nullptr && root->right == nullptr && targetSum == root->val) return true;
		return hasPathSum(root->left, targetSum - root->val) || hasPathSum(root->right, targetSum - root->val);
	}
};
对于最大路径分割问题,一个完整的C++代码通常包括以下步骤:构建图、找到最小生成树(如使用Kruskal或Prim算法)、遍历生成树并移除边以得到最大路径分割。下面是一个使用Kruskal算法的简化版本,假设我们已经有了一个图的边和权重集合: ```cpp #include <iostream> #include <vector> #include <set> #include <utility> // 边的结构体 struct Edge { int src, dest, weight; }; bool compareEdges(Edge a, Edge b) { return a.weight < b.weight; } int maxPathSplit(std::vector<Edge> &edges) { // 将所有边按照权重排序 std::sort(edges.begin(), edges.end(), compareEdges); std::vector<int> mstSet(edges.size(), 0); // 记录每个边是否属于最小生成树 std::set<int> graph; // 存储已经加入树的顶点 int totalWeight = 0; // 最小生成树的总重量 int mstSum = 0; // 当前路径分割的总重量 // Kruskal算法构建最小生成树 for (Edge e : edges) { if (graph.find(e.src) == graph.end() && graph.find(e.dest) == graph.end()) { mstSet[e.src] = mstSet[e.dest] = 1; // 双向插入 graph.insert(e.src); graph.insert(e.dest); totalWeight += e.weight; mstSum += e.weight; } } // 遍历最小生成树,移除边以得到最大路径分割 for (int i = 0; i < edges.size(); ++i) { if (mstSet[edges[i].src] && mstSet[edges[i].dest]) { mstSum -= edges[i].weight * 2; // 双向删除 } else if (mstSet[edges[i].src]) { mstSum -= edges[i].weight; // 单向删除 } } return mstSum; // 返回最大路径分割后的节点总和 } int main() { std::vector<Edge> edges = {{0, 1, 10}, {0, 2, 6}, {0, 3, 5}, {1, 4, 9}, {2, 4, 7}, {3, 4, 8}}; int pathSplitSum = maxPathSplit(edges); std::cout << "Maximum path split sum is: " << pathSplitSum << std::endl; return 0; } ``` 这段代码首先使用Kruskal算法找到最小生成树,然后遍历该树,移除边以得到最大路径分割。需要注意的是,此代码假定输入的边集合构成了一棵树,且无自环和交叉边。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

海螺蜜

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值