隊列安排【洛谷】

在這裏插入圖片描述
思路:這個就是一道模擬二叉樹的題。反正我是這麼理解的。
首先建立一個結構體數組,包含左孩子,右孩子,以及一個變量用來標記後面他是否被刪除了。
然後就是插入的時候先判斷p(根據題目),判斷完p之後還要判斷k的左右孩子。
舉個栗子:比如說i爲2,k爲3,p爲1.那麼現在2要往3的右邊插隊,你就首先要判斷3原先右邊有沒有數,沒有就直接插,有的話就是吧3的右孩子賦給2的右孩子,然後把3的右孩子變成2.
到這裏,整個二叉樹就算是模擬完了。然後就是刪除結點,刪一個把結構體裏面標記變量變一下就可以了;
最後就是遞歸輸出,先搜它左邊,搜到頭後開始輸出,緊接着搜他右邊。
代碼如下:

#include<bits/stdc++.h>
using namespace std;
int n,m,k,x;
struct node{
	int l,r,v;
}d[100010];
void dfs(int i)		//一個我自己都不知道自己在哪的遞歸
{
	if(i==-1)	return ;
	dfs(d[i].l);
	if(d[i].v==0)	cout<<i<<" ";
	dfs(d[i].r);
}
int main()
{
	cin>>n;
	d[1].l=d[1].r=-1,d[1].v=0;
	for(int i=2;i<=n;i++)
	{
		d[i].r=d[i].l=-1;d[i].v=0;
		cin>>k>>x;
		if(x)
			if(d[k].r)	{d[i].r=d[k].r;d[k].r=i;}
			else	d[k].r=i;
		else
			if(d[k].l)	{d[i].l=d[k].l;d[k].l=i;}
			else	d[k].l=i;
	}
	cin>>m;
	while(m--)
	{
		cin>>k;
		d[k].v=1;
	}
	dfs(1);
}
		
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章