trie实现&&字符编码&&graphviz

很菜的编码。效率，时间，空间，伤心的问题。小试一把c++模板，非常强大。函数对象，比函数指针强多了。tree数据结构用的孩子表示法。一个vector。。。见过高手用递归map表示的，很强！因为整个trie就是一个递归的数据结构。trie后面还有suffix trie，suffix tree。更多trie资料见wikipedia.

加入了编码(TCHAR)的东西，支持中文。

生成了graphviz用的dot图文件。这是个好东东。可以画各种图。配合其他工具，可以画函数调用图。

windows下装好graphviz，在cmd下，dot -O -Tpng -Efontname="宋体" trie.dot，就可以生成图了。命令行用法见官网。哦，生成的原始文件时MBCS编码的，需要转换成UTF-8编码(ultraedit就可以啦），在运行dot命令。(请确认bin目录在搜索路径中+_+）我的提示有warning，不过貌似没关系的！

graphviz支持中文也挺简单的，首先文件中中文要是utf-8编码，再指定字体就可以了。命令行或者文件中指定都可以。

编码方面的东东。这个程序要是在linux下运行如果输出，文件是啥编码？……utf-8，哦，首先linux本身要是utf-8得。⊙﹏⊙b汗。。因为ofstream设置locale了~~设置成跟系统locale一样的了。参数是空字符串。。在输出时，c/c++库到操作系统之间应该做了一层转换。。。。

windows下locale不能设置成UTF-8，见MSDN。The set of available languages, country/region codes, and code pages includes all those supported by the Win32 NLS API except code pages that require more than two bytes per character, such as UTF-7 and UTF-8.

能设置的话，就直接utf-8输出到文件了。

 

我也没在linux试过，+_+。本程序在vs2008下调试通过。

vs2008默认是unicode的，其实是ucs2。也就是wchar_t。

 

#include <tchar.h> #include <iostream> #include <vector> #include <fstream> #include <locale> using namespace std; typedef basic_string<TCHAR, char_traits<TCHAR>,allocator<TCHAR> > tstring; typedef basic_ofstream<TCHAR, char_traits<TCHAR> > tofstream; class Operation; template <typename EDGETYPE> class Trie { friend class Operation; private: struct Node { int nIndex;//for graph EDGETYPE edgeValue; vector<Node*> children; Node(EDGETYPE v):edgeValue(v) { } Node(){} }; class Operation { public: Operation(tofstream & out):of(out) { } void operator()(Node* p) { vector<Node*>::iterator it=p->children.begin(); for(;it!=p->children.end();++it) { of.imbue(locale("")); of<<p->nIndex<<_T("->")<<(*it)->nIndex<<_T("[label=/"")<<(*it)->edgeValue<<_T("/"]")<<endl; } } private: tofstream &of; }; Node *pHead; int nCount;//for graph void Tranverse(Node*,Operation&); void Destroy(Node*); public: ~Trie(); Trie(); void Insert(tstring& ); void CreateDotFile(tstring strPathName=_T("trie.dot")); }; template < typename EDGETYPE > Trie<EDGETYPE>::Trie() { pHead=new Node(0); nCount=0;//for graph pHead->nIndex=nCount++;//for graph } template < typename EDGETYPE > Trie<EDGETYPE>::~Trie() { if (pHead) { Destroy(pHead); } } template < typename EDGETYPE > void Trie<EDGETYPE>::Destroy(typename Trie<EDGETYPE>::Node* p) { if (p->children.size()==0) { delete p; } vector<Node*>::iterator iter=p->children.begin(); for (;iter!=p->children.end();++iter) { Destroy(*iter); } } template < typename EDGETYPE > void Trie<EDGETYPE>::Insert(tstring& str) { tstring::iterator iter=str.begin(); Node* pNode=pHead; while (pNode) { vector<Node*>::iterator iterVector=pNode->children.begin(); for (;iterVector!=pNode->children.end();++iterVector) { if ((*iterVector)->edgeValue==*iter) { break; } } if(iterVector==pNode->children.end())break;//break while pNode=*iterVector; ++iter; } //now pNode is the node we will insert after. for (;iter!=str.end();++iter) { Node* p=new Node(*iter); p->nIndex=nCount++; pNode->children.push_back(p); pNode=p; } } template < typename EDGETYPE > void Trie<EDGETYPE>::CreateDotFile(tstring strPathName) { tofstream of(strPathName.c_str()); of<<_T("digraph trie{")<<endl; Operation op(of); Tranverse(pHead,op); of<<_T("}")<<endl; of.close(); } template < typename EDGETYPE > void Trie<EDGETYPE>::Tranverse(Node* p,Operation& op) { if (p->children.size()==0) { return; } op(p); vector<Node*>::iterator it=p->children.begin(); for (;it!=p->children.end();++it) { Tranverse(*it,op); } } int _tmain(int argc, _TCHAR* argv[]) { Trie<TCHAR> t; //t.Insert(tstring(_T("inn"))); //t.Insert(tstring(_T("int"))); //t.Insert(tstring(_T("ate"))); //t.Insert(tstring(_T("age"))); //t.Insert(tstring(_T("adv"))); t.Insert(tstring(_T("湖南"))); t.Insert(tstring(_T("湖北"))); t.Insert(tstring(_T("山东"))); t.Insert(tstring(_T("山西"))); t.CreateDotFile(); return 0; }