【层次遍历】序列化、反序列化二叉树

最新推荐文章于 2022-07-31 09:05:17 发布

CodeHunter_qcy

最新推荐文章于 2022-07-31 09:05:17 发布

阅读量808

点赞数 1

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_43390235/article/details/106029934

版权

LeetCode 专栏收录该内容

45 篇文章 2 订阅

订阅专栏

在这里插入图片描述

借助辅助队列，层次操作结果做处理

public class Codec {

   String res;

	// 序列化
	public String serialize(TreeNode root) {
		if (root == null) {
			return "[]";
		}

		Queue<TreeNode> queue = new LinkedList<>();
		queue.add(root);
		res = "[" + root.val;
		while (!queue.isEmpty()) {
			TreeNode node = queue.poll();
			if (node.left != null) {
				queue.add(node.left);
				res += "," + node.left.val;
			} else {
				res += ",null";
			}
			if (node.right != null) {
				queue.add(node.right);
				res += "," + node.right.val;
			} else {
				res += ",null";
			}
		}
		res += "]";
		return res;
	}

	// 反序列化.
	public TreeNode deserialize(String data) {
		if (data.length() == 2) {
			return null;
		}
		// 去括号
		data = data.substring(1, data.length() - 1);
		// 逗号分割
		String str[] = data.split(",");
		Queue<TreeNode> queue = new LinkedList<>();
		TreeNode root = new TreeNode(Integer.valueOf(str[0]));
		queue.add(root);
		int i = 1;
		while (!queue.isEmpty()) {
			TreeNode node = queue.poll();
			if (str[i].equals("null")) {
				node.left = null;
			} else {
				node.left = new TreeNode(Integer.valueOf(str[i]));
				queue.add(node.left);
			}
			i++;
			if (str[i].equals("null")) {
				node.right = null;
			} else {
				node.right = new TreeNode(Integer.valueOf(str[i]));
				queue.add(node.right);
			}
			i++;
		}
		return root;
	}
}