机器人的运动范围

原文地址:https://www.jianshu.com/p/4df50a647831

时间限制:1秒 空间限制:32768K

题目描述

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

我的代码

class Solution {
public:
    int movingCount(int threshold, int rows, int cols)
    {
        if(threshold<0 || rows<1 ||cols<1)
            return 0;
        int* flag=new int[rows*cols]();
        return movingCountHelper(threshold,rows,cols,flag,0,0);
    }
    int movingCountHelper(int threshold, int rows, int cols, int* flag, int i,int j){
        if(i<0 || i>=rows || j<0 || j>=cols 
           || flag[i*cols+j]==1 || mySum(i)+mySum(j)>threshold)
            return 0;
        flag[i*cols+j]=1;
        return 1+movingCountHelper(threshold,rows,cols,flag,i+1,j)
            +movingCountHelper(threshold,rows,cols,flag,i-1,j)
            +movingCountHelper(threshold,rows,cols,flag,i,j+1)
            +movingCountHelper(threshold,rows,cols,flag,i,j-1);
    }
    int mySum(int x){
        int res=0;
        while(x){
            res+=x%10;
            x/=10;
        }
        return res;
    }
};

运行时间:5ms
占用内存:504k

上一篇:懒加载


下一篇:IsPostBack