trietreecell.h

我们一个小组

#pragma once

#include "FrequencyHeap.h"

class TrieTreeNode ;
class TrieTree ;

class TrieTreeCell
{
	friend class TrieTreeNode ;
	friend class TrieTree ;

public:

	TrieTreeCell():m_pNextNode(0)
	{

	}
	
	void InsertWord(double frequency, int id)
	{
		m_frequencyHeap.Insert(frequency, id) ;
	}

	void GetWord(vector<int>& vecIdTable)
	{
		m_frequencyHeap.GetWord(vecIdTable) ;
	}

	bool IsEmpty()
	{
		return m_frequencyHeap.GetCount() == 0 ;
	}

	bool IncreaseFrequency(int id)
	{
		return m_frequencyHeap.IncreaseFrequency(id) ;
	}

private:
	TrieTreeNode *m_pNextNode ;
	FrequencyHeap m_frequencyHeap ;
};