PAT--666题目的题解

题目：

小明有一张m*n的好习惯记录卡，记录每一天的好习惯目标达成度（数字0-9表示）。某天目标完成达成，就在当天的格子里写上数字6，目标没有完全达成就写上一个小于6的数字（0-5），目标超额完成就写上一个大于6的数字（7-9）。记录卡上如果能找到一条长度为3的路径并且路径上的三个数字都大于等于6（这里的路径是指从某个格子出发，可以向左、右、上、下格子移动，并且不能重复经过一个格子），则小明就能得到一个“666”奖励。

请你帮小明统计下他总共能得到多少“666”奖励。

输入格式:

输入第一行给出两个正整数m,n(1=<m,n<=100),随后是m行，每行包含n个0-9之间的数字。

输出格式:

先输出m行，每行包括n个整数，代表从当前格子出发得到的“666”奖励个数，中间用空格分割，最后一个数字后面不带空格。然后再在下一行输出得到的“666”奖励总数。

输入样例:

3 3
6 6 7
3 8 3
7 9 5

输出样例:

2 1 2
0 3 0
1 1 0
10

个人理解：该题目第一次读的话应该会想到用搜索算法来写，我想到的是用dfs深搜来写，根据题意的话，我们应该开两个二维数组，一个用来储存地图，一个用来储存可行的路径数，很明显，如果起点值小于6，那么我们就没必要在该点进行搜索。那么在大于5的起点上我们又该如何进行搜索呢，四个方向，选中其中一个方向后，在判断该点的值是否大于5，不大于就换个方向，大于的话就开始继续走，直到完成3个步数，那么我们开始代码吧

#include<bits/stdc++.h>
using namespace std;
int n,m;
int ma[200][200];
int mp[200][200];
int dfs(int i,int j,int undo,int direction)
{
	//i,j分别表示位置，undo是已经完成的标准步数，direction是方向  1右，2下，3左，4上
	int num=0,ans=0;//分别表示标准步数的路径数和总路径数
	//向右方向不能向左
	if(ma[i][j+1]>5&&direction!=3){
		if(undo==2)num++;
		else{
			ans+=dfs(i,j+1,undo+1,1);
		}
	}
	//向下走
	if(ma[i+1][j]>5&&direction!=4){
		if(undo==2)num++;
		else ans+=dfs(i+1,j,undo+1,2);
	}
	if(ma[i][j-1]>5&&direction!=1){
		if(undo==2)num++;
		else ans+=dfs(i,j-1,undo+1,3);
	}
	if(ma[i-1][j]>5&&direction!=2){
		if(undo==2)num++;
		else ans+=dfs(i-1,j,undo+1,4);
	}
	if(undo==2)return num;
	else return  ans;
}
int main()
{
	cin>>n>>m;
	int sum=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			cin>>ma[i][j];
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			if(ma[i][j]>5){
				mp[i][j]=dfs(i,j,1,0);
			}
			else mp[i][j]=0;
			sum+=mp[i][j];
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<m;j++){
			cout<<mp[i][j]<<" ";
		}
		cout<<mp[i][m]<<endl;
	}
	cout<<sum;
	return 0;
}