做了些小改动，最后一步可以不用算，直接出结果