61-递归和动态规划-汉诺塔II(记录当前序列是最有序列中的第几步)

链接：https://www.nowcoder.com/practice/b2d552cd60b7415fad2612a32e799812?tpId=49&tqId=29342&tPage=4&rp=4&ru=/ta/2016test&qru=/ta/2016test/question-ranking
来源：牛客网

题目描述

有一个int数组arr其中只含有1、2和3，分别代表所有圆盘目前的状态，1代表左柱，2代表中柱，3代表右柱，arr[i]的值代表第i+1个圆盘的位置。比如，arr=[3,3,2,1]，代表第1个圆盘在右柱上、第2个圆盘在右柱上、第3个圆盘在中柱上、第4个圆盘在左柱上。如果arr代表的状态是最优移动轨迹过程中出现的状态，返回arr这种状态是最优移动轨迹中的第几个状态。如果arr代表的状态不是最优移动轨迹过程中出现的状态，则返回-1。

给定一个int数组arr及数组的大小n，含义如题所述，请返回一个int，代表所求的结果。

测试样例：

[3,3]

返回：3

递归

# -*- coding:utf-8 -*-

class Hanoi:
    def chkStep(self, arr, n):
        # write code here
        result = [1] * n
        if result == arr:
        	return 0
        res = []

        def hanoi(n, x, y, z):
        	if n == 1:
        		result[n - 1] = z
        		res.append(result[:])
        		return
        	hanoi(n - 1, x, z, y)
        	result[n - 1] = z
        	res.append(result[:])
        	hanoi(n - 1, y, x, z)

        hanoi(n, 1, 2, 3)
        for x, dis in enumerate(res):
        	if dis == arr:
        		return x + 1

        return -1

if __name__ == "__main__":
	a = Hanoi()
	# 注意下面的几组测试数据的输出
	print a.chkStep([3, 3], 2)
	print a.chkStep([3, 3, 3, 3, 3, 3, 3], 7)
	print a.chkStep([2, 1, 1, 1, 3, 2, 2, 3], 8)
	print a.chkStep([1], 1)