User:Fako Berkers/Markovchain

text = "He saw the cat before he saw the potato."
dic = {}
prevword = ''

# Read the text into memory.
for word in text.split():

	if prevword != '': # doesn't occur the first time
		dic[prevword].append(word) # write current word under previous word in dic

	if word not in dic: 
		dic[word] = [] # create word in dic
	
	prevword = word # remember previous word to write next word under current word

else: # Looped through entirely
	dic[word].append("EOF;") # Add end marker to last word

print dic

# Pop the words in memory in their original order.
def readnorm(word):
	
	if word != "EOF;":
		print word,
		readnorm(dic[word].pop(0))

# Start random generator
import random
random.seed()

# Print words in a random order (but do not pop())
def readrand(word):
	
	if word != "EOF;":
		print word,
		ind = random.randrange(len(dic[word]))
		readrand(dic[word][ind])	

# Don't use recursion
def readrandloop():
	
	word = text.split()[0]
	while (word != "EOF;"):
		print word,
		ind = random.randrange(len(dic[word])) # Get index for new random word
		word = dic[word][ind]	
			

# Print output.
print "Random version: ", 
readrandloop()
print
print "Normal version: ", 
readnorm("He")