#include "stdafx.h"
#include "algorithm.h"
#include <cstdlib>
#include<ctime>
using std::iostream ;
using std::cout;
using std::rand;
using std::srand;

void algorithm::build(Prefix &prefix,istream &in )
{
    string buf;
	while(in>>buf)
       add(prefix,buf);
}
void algorithm::add(Prefix &prefix,const string &s)
{
    if(prefix.size ()==NPREF)
	{
	  table[prefix].push_back(s);
	   prefix.pop_front ();
	  
	}
	prefix.push_back (s);
}
void algorithm::generate(int nwords,ostream &out)
{
	Prefix prefix;
	for(int i=0;i<NPREF;i++)
	{
	   add(prefix,NONWORD);
	}
	for(int i=0;i<nwords;i++)
	{
	  vector<string > &suf= table[prefix];
	 // srand((unsigned int) time(0));
      const string &w=suf[rand() % suf.size()];
	  if(w==NONWORD)
		  break;
	  out<<w<<"  ";
	  prefix.pop_front ();
	  prefix.push_back(w); 
	}
  
}