剑指Offer面试题13: 机器人的运动范围

// 面试题13：机器人的运动范围
	// 题目：地上有一个m行n列的方格。一个机器人从坐标(0, 0)的格子开始移动，它
	// 每一次可以向左、右、上、下移动一格，但不能进入行坐标和列坐标的数位之和
	// 大于k的格子。例如，当k为18时，机器人能够进入方格(35, 37)，因为3+5+3+7=18。
	// 但它不能进入方格(35, 38)，因为3+5+3+8=19。请问该机器人能够到达多少个格子？

代码实现如下:

#include<iostream>
#include<string.h>
using namespace std;

int getWeight(int num1,int num2);
int movingCount(int k,int rows,int cols);
int movingCountCore(int k,int rows,int cols,int row,int col,bool *visted,int &count);

void test(char *testName,int k,int rows,int cols,int expected);
void test1();
void test2();
void test3();
void test4();
void test5();
void test6();
void test7();
void test8();
void test9();

int main(){
	
	test1();
	test2();
	test3();
    test4();
    test5();
    test6();
    test7();
    test8();
    test9();
	
	return 0;
}

void test(char *testName,int k,int rows,int cols,int expected){
	if(testName != nullptr)
		cout<<testName<<" begin:"<<endl;
	
	cout<<"k = "<<k<<" ,rows = "<<rows<<" ,cols = "<<cols<<endl;
	int count = movingCount(k,rows,cols);
	cout<<"expected = "<<expected<<" ,count = "<<count<<endl;
	if(expected == count)
		cout<<expected<<" is passed!"<<endl<<endl;
	else
		cout<<"not passed!"<<endl<<endl;
}

void test1(){
	test("Test1",5,10,10,21);
}

void test2()
{
    test("Test2", 15, 20, 20, 359);
}


void test3()
{
    test("Test3", 10, 1, 100, 29);
}

// 方格只有一行，机器人能到达所有方格
void test4()
{
    test("Test4", 10, 1, 10, 10);
}

// 方格只有一列，机器人只能到达部分方格
void test5()
{
    test("Test5", 15, 100, 1, 79);
}

// 方格只有一列，机器人能到达所有方格
void test6()
{
    test("Test6", 15, 10, 1, 10);
}

// 方格只有一行一列
void test7()
{
    test("Test7", 15, 1, 1, 1);
}

// 方格只有一行一列
void test8()
{
    test("Test8", 0, 1, 1, 1);
}

// 机器人不能进入任意一个方格
void test9()
{
    test("Test9", -10, 10, 10, 0);
}

int movingCount(int k,int rows,int cols){
	if(k < 0 || rows < 0 || cols < 0)
		return 0;
	
	bool *visted = new bool[rows * cols];
	//全部置为0,即false ,不要误用为sizeof(visted) = 4,应该是rows * cols * 1
	memset(visted,0,rows * cols);
	
	int count = 0;
	count = movingCountCore(k,rows,cols,0,0,visted,count);
	
	delete []visted;
	
	return count;
}

int movingCountCore(int k,int rows,int cols,int row,int col,bool *visted,int &count){
	if(k < 0 || rows < 0 || cols < 0)
		return 0;
	if(row < 0 || row >= rows || col < 0 || col >= cols)
		return 0;
	
	if(getWeight(row,col) > k)
		return 0;
		
	if(visted[row * cols + col] == true)
		return 0;
	
	if( (getWeight(row,col) <= k) && (visted[row * cols + col] == false) ){
		//cout<<"row = "<<row<<" ,col = "<<col<<" ,visted = "<<visted[row * cols + col]<<endl;
		visted[row * cols + col] = true;
		count = 1 + movingCountCore(k,rows,cols,row,col-1,visted,count) + 
		movingCountCore(k,rows,cols,row,col+1,visted,count) +
		movingCountCore(k,rows,cols,row-1,col,visted,count) +
		movingCountCore(k,rows,cols,row+1,col,visted,count);	
	}
	
	return count;
}

int getWeight(int num1,int num2){
	int sum = 0;
	while(num1 > 0){
		sum += num1 % 10;
		num1 /= 10;
	}
	
	while(num2 > 0){
		sum += num2 % 10;
		num2 /= 10;
	}
	return sum;
}

运行结果如下: