huffnode.h

Huffman编码及译码 1．将给定字符文件编码：生成编码

////////////////////////////////////////////////
//基类结点
#ifndef HUFFNODE_H
#define HUFFNODE_H

class HuffNode
{   
public:
    virtual int weight()=0;
    virtual bool isLeaf()=0;
    virtual HuffNode* left()=0;
    virtual HuffNode* right()=0;
    virtual void setLeft(HuffNode*)=0;
    virtual void setRight(HuffNode*)=0;
	

};


////////////////////////////////////////////////
//叶结点

class LeafNode:public HuffNode
{
private:
	FreqPair * it;
public:
	LeafNode(char& val,int freq)
	{
       it=new FreqPair(val,freq);
	}
	int weight (){return it->weight();}
	char val(){return it->val();}
	bool isLeaf(){return true;}

    HuffNode* left(){
		    return NULL;
	}
    HuffNode* right(){
			return NULL;
	}
    void setLeft(HuffNode*){}
    void setRight(HuffNode*){}
};

///////////////////////////////////////////////
//枝结点

class IntlNode:public HuffNode{
private:
	HuffNode * lc;
    HuffNode * rc;
	int wgt;
public:
	IntlNode(HuffNode* l,HuffNode* r){
		wgt=l->weight()+r->weight();
		lc=l;
		rc=r;	
	}
	int weight(){return wgt;}
        bool isLeaf(){return false;}
	HuffNode* left(){return lc;}        
	HuffNode* right(){return rc;}
	void setLeft(HuffNode *b){lc=(HuffNode*)b;}
	void setRight(HuffNode*b){rc=(HuffNode*)b;}
}; 
#endif