116. Populating Next Right Pointers in Each Node

题意:

将一颗二叉树的节点使用每个节点中的next指针将每行从左向右连接起来。

思路:层次遍历

每次加入链表同时加入对应的行号,若前一节点与当前节点行号不同则前一节点指向NULL。

struct Node {
	TreeLinkNode *TLN;
	int L;
	Node(TreeLinkNode *T, int i) :TLN(T), L(i) {};
};

void connect(TreeLinkNode *root) {
	if (!root)
		return;
	root->next = NULL;
	list<Node> s;
	Node p(root, 1);
	if (root->left)
		s.push_back(Node(root->left, 2));
	if (root->right)
		s.push_back(Node(root->right, 2));

	Node back = p;
	while (!s.empty()) {
		p = s.front();
		s.pop_front();
		if (p.L == back.L)
			back.TLN->next = p.TLN;
		else 
			back.TLN->next = NULL;


		if (p.TLN->left)
			s.push_back(Node(p.TLN->left, p.L + 1));
		if (p.TLN->right)
			s.push_back(Node(p.TLN->right, p.L + 1));
		back = p;
	}
}
全部评论

相关推荐

昨天 18:54
门头沟学院 Java
点赞 评论 收藏
分享
10-09 00:50
已编辑
长江大学 算法工程师
不期而遇的夏天:1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
牛客279957775号:铁暗恋
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务