一个n*m的网格,每个格子有一个字符,你需要从左上角走到右下角(只能往下或往右走),让经过的字符连起来是回文串,问有多少种方案。
dp。dp(i,j,k),表示从起点走到i行j列的格子,且对称的格子在第k行(列可以计算出来)的方案数,最后的答案是斜线对称轴的方案数的和。但是这样做会超内存,需要滚动数组压第一维变为i%2。
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
using namespace std;
#define ll long long
const int mod = 1000000007;
char mp[510][510];
int dp[2][510][510];
int n,m;
inline bool verrify(int a,int b,int c,int d){
if(a<1||a>n||b<1||b>m||c<a||c>n||d<b||d>m)return 0;
return 1;
}
int main(){
cin>>n>>m;
int mid = (n+m+2)>>1;
for(int i=1;i<=n;i++){
scanf("%s",mp[i]+1);
}
if(mp[1][1]==mp[n][m]){
dp[1][1][n]=1;
}
int ans=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=m&&(i+j)<=mid;j++){
if(i==1&&j==1)continue;
for(int ii=i;ii<=n;ii++){
int jj=2+n+m-i-j-ii;
if(jj<j||jj>m)continue;
dp[i%2][j][ii]=0;
if(mp[i][j]==mp[ii][jj]){
if(verrify(i-1,j,ii+1,jj))dp[i%2][j][ii]+=dp[(i-1)%2][j][ii+1];
dp[i%2][j][ii]%=mod;
if(verrify(i-1,j,ii,jj+1))dp[i%2][j][ii]+=dp[(i-1)%2][j][ii];
dp[i%2][j][ii]%=mod;
if(verrify(i,j-1,ii+1,jj))dp[i%2][j][ii]+=dp[i%2][j-1][ii+1];
dp[i%2][j][ii]%=mod;
if(verrify(i,j-1,ii,jj+1))dp[i%2][j][ii]+=dp[i%2][j-1][ii];
dp[i%2][j][ii]%=mod;
}
if(i+j==mid){
ans+=dp[i%2][j][ii];
ans%=mod;
}
}
}
}
cout<<ans<<endl;
return 0;
}