PIPI OJ 1364: 删除给定值的叶子节点

菜鸟生成记(80)

1364: 删除给定值的叶子节点

这一题有点考验数据结构基础功底,删除几个节点就把我难住了,指针总是出错;最终从学长那里学了一个逻辑删除(没有真的删除节点,只是逻辑上删除节点,物理上节点还在);这玩意儿不用删除节点操作，操作安全

#include<stdio.h>
#include<iostream>
using namespace std;
typedef struct node{
	int data;//数据域 
	int flag;//删除标记 
	struct node *l,*r,*pre;//指针域 
	node()//初始化 
	{l=r=pre=NULL;flag=0;}
}*tree;
int n=0;
tree t[100];//节点指针数组 
void create(tree &T,tree f)//先序建树 
{//创建二叉树 
	int x;
	cin>>x;
	if(x==-1)
	T=NULL;//空节点 
	else
	{//节点分配内存 
		T=new node;
		T->data=x;//数据域赋值 
		T->pre=f;//父亲赋值 
		create(T->l,T);//左子树 
		create(T->r,T);//右子树 
	}
}
void inp(tree T)
{//先序遍历 输出 
	if(T)
	{
		if(T->flag==0)//未被标记的节点输出值 
		cout<<T->data<<" ";
		inp(T->l);
		inp(T->r);
	}
}
void dfs(tree &T,int x)
{
	if(T)
	{
		if(T->l==NULL&&T->r==NULL&&T->data==x)
		{//找到所有值为x的叶子节点 
			t[n++]=T;
			return;
		}
		dfs(T->l,x);
		dfs(T->r,x);
	}
}
/*
1 2 2 -1 -1 -1 3 2 -1 -1 4 -1 -1
2
*/
int main()
{
	int x;
	tree T=NULL,p;
	create(T,T);//建树 
	cin>>x;
	dfs(T,x);//查找所有的值为x的叶子节点 
	for(int i=0;i<n;i++)
	{//自下而上遍历 
		p=t[i];
		while(p)
		{
			//p->flag==1是删除标记,逻辑删除(并未从二叉树中删除,遍历时不输出值) 
			//逻辑删除的几种情况 
			if(p->l==NULL&&p->r==NULL&&p->data==x)
			p->flag=1; 
			if(p->l!=NULL&&p->r!=NULL&&p->l->flag==1&&p->r->flag==1&&p->data==x)
			p->flag=1;
			if(p->l!=NULL&&p->l->flag==1&&p->r==NULL&&p->data==x)
			p->flag=1;
			if(p->r!=NULL&&p->l==NULL&&p->r->flag==1&&p->data==x)
			p->flag=1;
			p=p->pre;//前往父亲节点 
		}
	}
	inp(T);//遍历二叉树 
	return 0;
}