Viterbi Algorithm Geeksforgeeks Python Questions