HDU 6416 Rikka with Seam（dp）

Description

给出一个$n\times m$的$01$矩阵，要求从每一行删除一个元素，且相邻行所删除元素的列差值不超过$K$，问删除后得到的不同矩阵的数量

Input

第一行一整数$T$表示用例组数，每组用例首先输入三个整数$n,m,K$表示矩阵行列数和对列的限制，之后输入一个$n\times m$的$01$矩阵

$(1\le T\le 10^3,2\le n,m\le 2\cdot 10^3,1\le K\le m)$

Output

输出所得到不同矩阵的数量，结果模$998244353$

Sample Input

3
2 2 1
00
10
5 5 1
00100
10101
00100
01000
11101
5 5 2
00100
10101
00100
01000
11101

Sample Output

2
70
199

Solution

令$f[i][j]$表示切到$(i,j)$处对应的不同轨迹数量，则显然有转移

$$f[i][j]=\sum\limits_{l=max(1,j-K)}^{min(m,j+K)}f[i-1][l]$$
但是注意到对于一行连续相同的一段数字，删除任意一个得到的是相同的结果，故直接计算轨迹数量显然会重，考虑去重，以 $g[i][j]$表示切到 $(i,j)$处所得到的轨迹中与切到 $(i,j-1)$处得到的轨迹等价的数量，显然若 $s[i][j-1]$和 $s[i][j]$不相同时，到达两者的轨迹不会等价，考虑 $s[i][j]=s[i][j-1]$，其相交贡献区间为 $[max(1,j-K),min(m,j-1+K)]$，那么有
$$g[i][j]=\sum\limits_{l=max(1,j-K)}^{min(m,j-1+K)}f[i-1][l]-\sum\limits_{l=max(1,j-K)+1}^{min(m,j-1+K)}g[i-1][l]$$
该朴素 $dp$时间复杂度为 $O(nm^2)$，注意到都是区间加，故求一遍前缀和即可，时间复杂度 $O(nm)$

Code

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
#define maxn 2005
#define mod 998244353
int mul(int x,int y)
{
    ll z=1ll*x*y;
    return z-z/mod*mod;
}
int add(int x,int y)
{
    x+=y;
    if(x>=mod)x-=mod;
    return x;
}
int T,n,m,K,f[maxn][maxn],g[maxn][maxn];
char s[maxn][maxn];
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&K);
        for(int i=1;i<=n;i++)scanf("%s",s[i]+1);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                f[i][j]=g[i][j]=0;
        for(int i=1;i<=m;i++)
        {
            f[1][i]=1;
            if(s[1][i]==s[1][i-1])g[1][i]=1;
        }
        for(int i=2;i<=n;i++)
        {
            for(int j=2;j<=m;j++)
            {
                f[i-1][j]=add(f[i-1][j],f[i-1][j-1]);
                g[i-1][j]=add(g[i-1][j],g[i-1][j-1]);
            }
            for(int j=1;j<=m;j++)
            { 
                int L=max(1,j-K),R=min(m,j+K);
                f[i][j]=add(f[i-1][R],mod-f[i-1][L-1]);
                f[i][j]=add(f[i][j],mod-add(g[i-1][R],mod-g[i-1][L]));
                if(s[i][j]!=s[i][j-1])g[i][j]=0;
                else
                {
                    R=min(m,j-1+K);
                    g[i][j]=add(f[i-1][R],mod-f[i-1][L-1]);
                    g[i][j]=add(g[i][j],mod-add(g[i-1][R],mod-g[i-1][L]));
                }
            }
        }
        int ans=0;
        for(int i=1;i<=m;i++)ans=add(ans,add(f[n][i],mod-g[n][i]));
        printf("%d\n",ans);
    }
    return 0;
}