机器人的运动范围剑指offer66题

最新推荐文章于 2021-03-29 22:35:29 发布

weixin_30877227

最新推荐文章于 2021-03-29 22:35:29 发布

阅读量107

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/wdan2016/p/6858301.html

本文介绍了一个算法问题：在给定阈值、行数和列数的情况下，计算矩阵中可移动的格子数量。通过递归遍历矩阵并利用回溯的方法，实现了对符合条件的格子计数。

include "stdafx.h"

#include<vector>
#include<algorithm>
#include<string>
#include<iostream>
#include<stack>
using namespace std;

class Solution {
public:
    int getSum(int rows, int cols)
    {
        int sum = 0;
        while (rows!=0)
        {
            sum += rows % 10;
            rows = rows / 10;
        }
        while (cols != 0)
        {
            sum += cols % 10;
            cols = cols / 10;
        }
        return sum;
    }
    int count = 0;
    int movingCount(int threshold, int rows, int cols)
    {
        vector<vector<bool>> visited(rows,vector<bool>(cols,false));
        getCount(threshold, 0, 0, visited, rows, cols);
        return count;
    }
    void getCount(int threshold, int rows, int cols, vector<vector<bool>> &visited,int m,int n)
    {
        if (getSum(rows, cols) <= threshold)
        {
            count++;
        //  cout << count << endl;
            visited[rows][cols] = true;
            if (cols - 1 >= 0&&visited[rows][cols-1]==false)
            {
                getCount(threshold, rows, cols - 1, visited,m,n);
            }
            if (cols + 1 < n && visited[rows][cols + 1]==false)
            {
                getCount(threshold, rows, cols + 1, visited,m,n);
            }
            if (rows - 1 >= 0 && visited[rows-1][cols]==false)
            {
                getCount(threshold, rows-1, cols , visited,m,n);
            }
            if (rows + 1 < m && visited[rows + 1][cols]==false)
            {
                getCount(threshold, rows + 1, cols, visited,m,n);
            }
        //  visited[rows][cols] = false;
            //count--;
            
        }
    

    }

};
int main()
{
    Solution s;
    cout << s.movingCount(5, 10, 10) << endl;

    return 0;
}

转载于:https://www.cnblogs.com/wdan2016/p/6858301.html