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;
}