hello-algo/codes/csharp/chapter_backtracking/preorder_find_constrained_paths.cs

61 lines
1.5 KiB
C#
Raw Normal View History

/**
* File: preorder_find_constrained_paths.cs
* Created Time: 2023-04-17
* Author: hpstory (hpstory1024@163.com)
*/
using hello_algo.include;
using NUnit.Framework;
namespace hello_algo.chapter_backtracking;
public class preorder_find_constrained_paths
{
static List<TreeNode> path;
static List<List<TreeNode>> res;
/* 前序遍历 */
static void preOrder(TreeNode root)
{
// 剪枝
if (root == null || root.val == 3)
{
return;
}
// 尝试
path.Add(root);
if (root.val == 7)
{
// 记录解
res.Add(new List<TreeNode>(path));
}
preOrder(root.left);
preOrder(root.right);
// 回退
path.RemoveAt(path.Count - 1);
}
[Test]
public void Test()
{
TreeNode root = TreeNode.ArrToTree(new int?[] { 1, 7, 3, 4, 5, 6, 7 });
Console.WriteLine("\n初始化二叉树");
PrintUtil.PrintTree(root);
// 前序遍历
path = new List<TreeNode>();
res = new List<List<TreeNode>>();
preOrder(root);
Console.WriteLine("\n输出所有根节点到节点 7 的路径,且路径中不包含值为 3 的节点");
foreach (List<TreeNode> path in res)
{
List<int> vals = new List<int>();
foreach (TreeNode node in path)
{
vals.Add(node.val);
}
Console.WriteLine(string.Join(" ", vals));
}
}
}