beam search和greedy search Python代码实现