CodeForces 570E DP Pig and Palindromes

题意:给出一个n行m列的字符矩阵,从左上角走到右下角,每次只能往右或者往下走,求一共有多少种走法能得到回文串。

分析:

可以从两头开始考虑,每次只走一样字符的格子,这样得到的两个字符串拼起来之后就是一个回文串。

设d(step, x1, y2, x2, y2)表示从左上角(1, 1)z往右下走step个格子走到(x1, y1),同时从右下角(n, m)往左上走step个格子走到(x2, y2),而且两边得到一样字符串的方法数。

一开始判断一下左上角和右下角的字符是不是一样,若一样,d(1, 1, 1, n, m) = 1,若不一样则直接输出0.

状态转移:

如果s[x1][y2] = s[x2][y2],而且x1 ≤ y1 && y1 ≤ y2

d(step, x1, y1, x2, y2) = d(step - 1, x1 - 1, y1, x2 + 1, y2) + d(step - 1, x1, y1 - 1, x2 + 1, y2) + d(step - 1, x1 - 1, y1, x2, y2 + 1) + d(step - 1, x1, y1- 1, x2, y2+ 1)

这是一个五维的状态空间太大了,空间优化:

首先如果已知step的话,x1和y1只要知道其中一个,就可以计算出另一个,x2和y2同理,所以我们可以把y1和y2去掉,这样把五维优化成三维。

另外一个优化就是,我们在递推的时候是按照step从小到大递推的,所以可以用滚动数组来优化,这样空间复杂度为O(n^2)。

另外要考虑一下奇偶的问题:

如果整个字符串长度为奇数的话,两个字符串会相遇在同一个格子

如果整个字符串长度为偶数的话,两个字符串会走到相邻的格子,要么是同一行左右相邻,要么是同一列上下相邻

 1 #include <iostream>
 2 #include <cstdio>
 3 #include <cstring>
 4 #include <algorithm>
 5 using namespace std;
 6 
 7 typedef long long LL;
 8 
 9 const int maxn = 500 + 10;
10 const int MOD = 1000000007;
11 
12 int d[2][maxn][maxn];
13 
14 char G[maxn][maxn];
15 
16 void add(int& a, int b) { a += b; if(a >= MOD) a -= MOD; }
17 
18 int main()
19 {
20     //freopen("in.txt", "r", stdin);
21 
22     int n, m;
23     scanf("%d%d", &n, &m);
24     for(int i = 1; i <= n; i++) scanf("%s", G[i] + 1);
25 
26     if(G[1][1] != G[n][m]) { puts("0"); return 0; }
27 
28     d[1][1][n] = 1;
29     int cur = 1;
30     for(int s = 2; s <= (n + m) / 2; s++)
31     {
32         cur ^= 1;
33         memset(d[cur], 0, sizeof(d[cur]));
34         for(int x1 = 1; x1 <= s; x1++)
35             for(int x2 = n; x2 >= n - s + 1; x2--)
36             {
37                 int y1 = s + 1 - x1;
38                 int y2 = n + m + 1 - s - x2;
39                 if(x1 > x2 || y1 > y2) continue;
40                 if(G[x1][y1] == G[x2][y2])
41                 {
42                     add(d[cur][x1][x2], d[cur^1][x1][x2]);
43                     add(d[cur][x1][x2], d[cur^1][x1-1][x2]);
44                     add(d[cur][x1][x2], d[cur^1][x1][x2+1]);
45                     add(d[cur][x1][x2], d[cur^1][x1-1][x2+1]);
46                 }
47             }
48     }
49 
50     int ans = 0;
51     if((n + m) & 1)
52     {
53         for(int i = 1; i <= n; i++) add(ans, d[cur][i][i]), add(ans, d[cur][i][i+1]);
54     }
55     else
56     {
57         for(int i = 1; i <= n; i++) add(ans, d[cur][i][i]);
58     }
59 
60     printf("%d
", ans);
61 
62     return 0;
63 }
代码君
原文地址:https://www.cnblogs.com/AOQNRMGYXLMV/p/4753881.html