如何实现三态选择树——重点是树的遍历

有时候我们经常需要实现树的多态选择，本文就介绍一种三态选择树的具体实现。
步骤一：生成一个对话框工程。
步骤二：添加树控件，设置所需的属性。
步骤三：做三态选择框图标，且在OnInitDialog()中初始化树，使树控件与选择框图标所构成的图像列表控件联系，具体实现看源代码。
步骤四：实现树的左击消息，实现三态选择。

void CMutiTreeDemoDlg::OnClickTree1(NMHDR* pNMHDR, LRESULT* pResult) 
{
	UINT uFlags=0;
	HTREEITEM hPItem;
	CPoint point = GetCurrentMessage()->pt;
	ScreenToClient(&point);
	point.Offset(-8,-16);
	HTREEITEM hItem =m_TripleTree.HitTest(point, &uFlags);
	if ( (uFlags&TVHT_ONITEM) )
	{
		hPItem=m_TripleTree.GetParentItem(hItem);
		int m,n,m1,n1;
		m_TripleTree.GetItemImage(hItem,m,n);
		if(m!=1)
		{
			m_TripleTree.SetItemImage(hItem,1,1);
			m_TripleTree.GetItemImage(hItem,m1,n1);
			TravelChild(hItem, m1);
			if(hPItem!=NULL)
			TravelSiblingAndParent(hItem,m1);
		}
		else if(m==1)
		{
			m_TripleTree.SetItemImage(hItem,0,0);
			m_TripleTree.GetItemImage(hItem,m1,n1);
			TravelChild(hItem, m1);
			if(hPItem!=NULL)
				TravelSiblingAndParent(hItem,m1);
		}
			//	MessageBox("xxxx");
	}
	*pResult = 0;	
}

这其中，最难实现的是树的遍历，我用递归
TravelChild(HTREEITEM hItem, int m1);
TravelSiblingAndParent(HTREEITEM hItem,int m1); 来实现。

代码如下：

//递归
void CMutiTreeDemoDlg::TravelChild(HTREEITEM hItem,int m)
{
	HTREEITEM hChildItem,hBrotherItem;
	hChildItem=m_TripleTree.GetChildItem(hItem);
	m_TripleTree.SetItemImage(hChildItem,m,m);
	if(hChildItem!=NULL)
	TravelChild(hChildItem, m);
	hBrotherItem=m_TripleTree.GetNextSiblingItem(hChildItem);
	m_TripleTree.SetItemImage(hBrotherItem,m,m);
	if(hBrotherItem!=NULL)
		TravelChild(hBrotherItem, m);
}

void CMutiTreeDemoDlg::TravelSiblingAndParent(HTREEITEM hItem, int m)
{
	int m1,n1;
	HTREEITEM hNextSiblingItem,hPrevSiblingItem,hParentItem;
	m1=m;
	hNextSiblingItem=m_TripleTree.GetNextSiblingItem(hItem);
	while(hNextSiblingItem!=NULL)
	{
		m_TripleTree.GetItemImage(hNextSiblingItem,m1,n1);
		if(m1!=m)
			break;
		else
		{
			hNextSiblingItem=m_TripleTree.GetNextSiblingItem(hNextSiblingItem);
			if(hNextSiblingItem!=NULL)
				m_TripleTree.GetItemImage(hNextSiblingItem,m1,n1);
		}
	}
	
	hPrevSiblingItem=m_TripleTree.GetPrevSiblingItem(hItem);
	while(hPrevSiblingItem!=NULL)
	{
		m_TripleTree.GetItemImage(hPrevSiblingItem,m1,n1);
		if(m1!=m)
			break;
		else
		{
			hPrevSiblingItem=m_TripleTree.GetPrevSiblingItem(hPrevSiblingItem);
			if(hPrevSiblingItem!=NULL)
				m_TripleTree.GetItemImage(hPrevSiblingItem,m1,n1);
		}
	}
	if(m1==m)
	{
		hParentItem=m_TripleTree.GetParentItem(hItem);
		if(hParentItem!=NULL)
		{
			m_TripleTree.SetItemImage(hParentItem,m1,m1);
			TravelSiblingAndParent(hParentItem,m1);
		}
	}	
	
	if(m1!=m)
	{
		
		hParentItem=m_TripleTree.GetParentItem(hItem);
		while(hParentItem!=NULL)
		{
			m_TripleTree.SetItemImage(hParentItem,2,2);
			hParentItem=m_TripleTree.GetParentItem(hParentItem);
		}
	}
	
}