C - Digital Path 计蒜客 - 42397 05-29

Posted 2021-06-06 Jozky86

C - Digital Path 计蒜客 - 42397

题意：

题意就是给出一个n ∗ m的数字矩阵每个矩阵元素之间只能上下左右走，而且下一个位置必须比当前位置的数字大1，入口和出口必须数边缘元素，求可以有多少条路径。

题解：

第一反应dfs，dfs+记忆化搜索+剪枝
vis[x][y]=1表示这个点能作为dfs的起点

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int maxn=1e6+9;
int vis[2000][2000];
int n,m;
int a[2000][2000];
int dp[2000][2000];
int dx[]={0,1,0,-1};
int dy[]={1,0,-1,0};
int sum=0;
bool f=0;
inline int dfs(int x,int y,int tot)
{
	if(dp[x][y])return dp[x][y];
	dp[x][y]=0;
	ll ans=0;
	int pp=0;
	for(int i=0;i<4;i++)
	{
		int newx=x+dx[i];
		int newy=y+dy[i];
		if(newx>n||newx<1||newy>m||newy<1)continue;
		if(a[newx][newy]==a[x][y]+1)
		{
			pp=1;
			ans=ans+dfs(newx,newy,tot+1);
			ans=ans%mod;
		}	
	}
	if(tot>=4&&pp==0)//如果长度大于4且已走到头 
	{
		return 1; 
	}
	dp[x][y]+=ans;
	return dp[x][y]%mod;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			scanf("%d",&a[i][j]);
		}
	}
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			for(int k=0;k<4;k++)
			{
				int newx=i+dx[k];
				int newy=j+dy[k];
				if(newx>n||newx<1||newy>m||newy<1)continue;
				if(a[newx][newy]==a[i][j]-1)
				{
					vis[i][j]=1;
				}
			}
		}
	}
	ll ans=0;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			if(vis[i][j]==0){
				dp[i][j]=dfs(i,j,1);
				ans=ans+dp[i][j];
				ans=ans%mod;
			}
				
		}
	}
	cout<<ans%mod;
}