面试题13 机器人的运动范围

地上有一个m行n列的方格,从座标 [0,0] 到座标 [m-1,n-1] 。一个机器人从座标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行座标和列座标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?

示例 1:

输入:m = 2, n = 3, k = 1
输出:3
示例 2:

输入:m = 3, n = 1, k = 0
输出:1

用最笨的遍历的方法做了半个小时也没做出来,先把我的弱智垃圾还有错的代码贴在这,明天更新

class Solution:
    def movingCount(self, m: int, n: int, k: int) -> int:
        # 方法1 全部遍历
        num = 0
        col_flag = 0
        for i in range(m):
            i_bitSum = int(i/10)+(i%10)
            print(i_bitSum)
            if k >= i_bitSum: # 行的位和不大于k
                for j in range(n):
                    j_bitSum = int(j/10)+(j%10)
                    print(j_bitSum)
                    if k >= (i_bitSum + j_bitSum):
                        num = num + 1 
                    else:   # 出现某一行的某一列中断,则后面的列也必然不连通,直接break,执行下一行
                        if i == 0:
                            col_flag = j
                        elif j >= col_flag:  # 第一行的列决定了最远的位置
                            break
            else:   # 若只是行的位和都比k大,则直接下一次循环即可
                break   # 出现一行一个都没有,则 这个图不连通,即后面的都是错误的。直接break返回
        return num

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章