import bean.TreeNode;
import java.util.*;
public class TreeTest {
public static final int PRE_ORDER = 1;
public static final int MID_ORDER = 2;
public static final int AFTER_ORDER = 3;
static List<Integer> preOrder = new ArrayList<>();
static List<Integer> midOrder = new ArrayList<>();
static List<Integer> afterOrder = new ArrayList<>();
public static void main(String[] args) {
TreeNode root = new TreeNode(3);
root.left = new TreeNode(6);
root.right = new TreeNode(5);
root.left.right = new TreeNode(9);
root.right.left = new TreeNode(10);
root.right.right = new TreeNode(20);
new TreeTest().traversal(root);
int[] preArray = new int[preOrder.size()];
for (int i = 0; i < preOrder.size(); i++) {
preArray[i] = preOrder.get(i);
System.out.print(" " + preOrder.get(i));
}
System.out.println();
int[] midArray = new int[midOrder.size()];
for (int i = 0; i < midOrder.size(); i++) {
midArray[i] = midOrder.get(i);
System.out.print(" " + midOrder.get(i));
}
System.out.println();
for (int i = 0; i < afterOrder.size(); i++) {
System.out.print(" " + afterOrder.get(i));
}
// 根據(jù)先序和中序重建樹
TreeNode rebuild = new TreeNode(preArray[0]);
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < midArray.length; i++) {
map.put(midArray[i], i);
}
Stack<TreeNode> stack = new Stack<>();
stack.push(rebuild);
for (int i = 1; i < preArray.length; i++) {
TreeNode top = stack.peek();
// 棧頂值在mid中的位置
int p = map.get(top.val);
//插入值在mid中的位置
int j = map.get(preArray[i]);
if (j < p) {
top.left = new TreeNode(preArray[i]);
stack.push(top.left);
} else {
TreeNode q = null;
while (!stack.isEmpty() && map.get(stack.peek().val) < j) {
q = stack.pop();
}
q.right = new TreeNode(preArray[i]);
stack.push(q.right);
}
}
System.out.println(isSameTree(root, rebuild));
}
// 判斷兩棵樹是否相同
public static boolean isSameTree(TreeNode a, TreeNode b) {
if (a == null && b == null) {
return true;
}
if (a == null || b == null) {
return false;
}
return a.val == b.val && isSameTree(a.left, b.left) && isSameTree(a.right, b.right);
}
// 樹的遍歷
public void traversal(TreeNode root) {
if (root == null) {
return;
}
preOrder.add(root.val);
traversal(root.left);
midOrder.add(root.val);
traversal(root.right);
afterOrder.add(root.val);
}
}
重建樹
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
- 文/潘曉璐 我一進(jìn)店門亭罪,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人歼秽,你說(shuō)我怎么就攤上這事应役。” “怎么了燥筷?”我有些...
- 文/不壞的土叔 我叫張陵箩祥,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我肆氓,道長(zhǎng)袍祖,這世上最難降的妖魔是什么? 我笑而不...
- 正文 為了忘掉前任谢揪,我火速辦了婚禮盲泛,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘键耕。我一直安慰自己寺滚,他們只是感情好,可當(dāng)我...
- 文/花漫 我一把揭開白布屈雄。 她就那樣靜靜地躺著村视,像睡著了一般。 火紅的嫁衣襯著肌膚如雪酒奶。 梳的紋絲不亂的頭發(fā)上蚁孔,一...
- 那天,我揣著相機(jī)與錄音惋嚎,去河邊找鬼杠氢。 笑死,一個(gè)胖子當(dāng)著我的面吹牛另伍,可吹牛的內(nèi)容都是我干的鼻百。 我是一名探鬼主播,決...
- 文/蒼蘭香墨 我猛地睜開眼摆尝,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼温艇!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起堕汞,我...
- 序言:老撾萬(wàn)榮一對(duì)情侶失蹤勺爱,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后讯检,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體琐鲁,經(jīng)...
- 正文 獨(dú)居荒郊野嶺守林人離奇死亡卫旱,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
- 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了围段。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片顾翼。...
- 正文 年R本政府宣布取逾,位于F島的核電站,受9級(jí)特大地震影響苹支,放射性物質(zhì)發(fā)生泄漏砾隅。R本人自食惡果不足惜,卻給世界環(huán)境...
- 文/蒙蒙 一债蜜、第九天 我趴在偏房一處隱蔽的房頂上張望晴埂。 院中可真熱鬧,春花似錦寻定、人聲如沸儒洛。這莊子的主人今日做“春日...
- 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)琅锻。三九已至,卻和暖如春向胡,著一層夾襖步出監(jiān)牢的瞬間恼蓬,已是汗流浹背。 一陣腳步聲響...
- 正文 我出身青樓拇派,卻偏偏與公主長(zhǎng)得像荷辕,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子攀痊,可洞房花燭夜當(dāng)晚...