setdemo2.cpp

C++&datastructure书籍源码,以前外教提供现在与大家共享

#include <iostream>
#include <fstream>   // for ifstream and ofstream
#include <string>
using namespace std;

#include "worditer.h"
#include "stringset.h"
#include "strutils.h"
#include "prompt.h"

void Print(StringSetIterator& ssi, ostream& output)
{
    for(ssi.Init(); ssi.HasMore(); ssi.Next())
    {    output << ssi.Current() << endl;
    }
}

int main()
{
    string filename = PromptString("enter file name: ");
    WordStreamIterator wstream;
    wstream.Open(filename);
    string word;
    
    StringSet wordset;
    for(wstream.Init(); wstream.HasMore(); wstream.Next())
    {   word = wstream.Current();
        ToLower(word);
        StripPunc(word);
        wordset.insert(word);
    }
    StringSetIterator ssi(wordset);  
    Print(ssi,cout);
    cout << "# different words = " << wordset.size() <<  endl;
   
    filename = PromptString("file for output: ");
    ofstream output(filename.c_str());
    Print(ssi,output);
    
    return 0;
}