Codeforces Round #316 (Div. 2) E. Pig and Palindromes (DP)

最新推荐文章于 2022-08-21 01:05:54 发布

Aerolite坠落

最新推荐文章于 2022-08-21 01:05:54 发布

阅读量926

点赞数 1

分类专栏： DP 文章标签： ACM 算法与数据结构编程 DP

本文链接：https://blog.csdn.net/scf0920/article/details/47725865

版权

DP 专栏收录该内容

59 篇文章 0 订阅

订阅专栏

题目地址：http://codeforces.com/contest/570/problem/E
比赛的时候急着睡觉这题没看。。。
最显然的一个DP方法是dp[i][x1][y1][x2][y2]表示以s[x1][y1]和s[x2][y2]为第i步分别与s[1][1]和s[n][m]形成的回文串的个数，但是数组大小太大。。这时候可以发现知道步数后，y1和y2可以根据x1,x2来求出来，所以可以优化成dp[i][x1][x2]，然后再由于每一步都只从上一步来传递，所以可以用滚动数组滚一下就好了。
代码如下：

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
#include <time.h>
using namespace std;
#define LL long long
#define pi acos(-1.0)
#pragma comment(linker, "/STACK:1024000000")
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-9;
const int MAXN=500+10;
LL dp[2][MAXN][MAXN];
char s[MAXN][MAXN];
int main()
{
        int n, m, i, j, k, x1, x2, y1, y2, now, pre, tot;
        LL ans;
        while(scanf("%d%d",&n,&m)!=EOF){
                for(i=1;i<=n;i++){
                        scanf("%s",s[i]+1);
                }
                if(s[1][1]!=s[n][m]){
                        puts("0");
                        continue ;
                }
                dp[0][1][n]=1;
                tot=n+m-2>>1;
                for(i=1;i<=tot;i++){
                        pre=i+1&1; now=i&1;
                        memset(dp[now],0,sizeof(dp[now]));
                        for(j=1;j<=i+1;j++){
                                for(k=n;k>=j&&k>=n-i;k--){
                                        x1=j; y1=i-j+2;
                                        x2=k; y2=m-(i-(n-k));
                                        if(s[x1][y1]!=s[x2][y2]) continue ;
                                        dp[now][j][k]+=dp[pre][j-1][k];
                                        dp[now][j][k]+=dp[pre][j-1][k+1];
                                        dp[now][j][k]+=dp[pre][j][k];
                                        dp[now][j][k]+=dp[pre][j][k+1];
                                        dp[now][j][k]%=mod;
                                }
                        }
                }
                ans=0;
                if(n+m&1){
                        for(i=1;i<=n;i++){
                                ans=(ans+dp[now][i][i]+dp[now][i][i+1])%mod;
                        }
                }
                else{
                        for(i=1;i<=n;i++){
                                ans=(ans+dp[now][i][i])%mod;
                        }
                }
                printf("%I64d\n",ans);
        }
        return 0;
}