AtCoder Beginner Contest 220 D 线性DP 初始化注意项 动态规划

题目

给你一个长度为N的序列 (只有 0 到 9 的数) ， 你要进行操作F或者G使序列只有一个数
F
删除最左边两个数 a b 插入 ( a + b) % 10
G
删除最左边两个数 a b 插入 ( a* b) % 10

题解思路

坑点
如果非得从自己初始化第一次合并
必须考虑全面
不加 += 的时候 2 2 下面的代码就会错
所以初始化最好定义默认值

    f[2][(a[2]+a[1])%10] = 1 ;
    f[2][(a[2]*a[1])%10] += 1 ;

之间枚举肯定是不行的 ， 让我们想到dp。
定义集合 f[ i ] [ j ] 操作了i 次 最左边的数为 j 的 情况的总次数 。
f[ 0 ] [ a[0] ] = 1
操作0次 最左边肯定是 a[ 0 ] 初始化
转移
我们不知道前一个数到底是什么 所以进行枚举 前一个数 为 j 的情况 来进行转移 。

        for (int j = 0 ; j <= 9 ; j++ )
        {
            f[i][ (a[i] + j)%10] = ( f[i-1][j] + f[i][ (a[i] + j)%10] ) % mod ;
            f[i][ (a[i] * j)%10] = ( f[i-1][j] + f[i][ (a[i] * j)%10] ) % mod ;
        }

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;

//const  int  INF =  0x3f3f3f3f;
const  int  mod =  998244353 ;

int a[100100];

int f[100100][20];

int main ()

{
    ios::sync_with_stdio(false);

    int n;

    cin>>n;

    for (int i = 0 ;  i < n ; i++ )
        cin>>a[i];

    f[0][a[0]] = 1 ;
    for (int i = 1 ; i < n ; i++ )
    {
        for (int j = 0 ; j <= 9 ; j++ )
        {
            f[i][ (a[i] + j)%10] = ( f[i-1][j] + f[i][ (a[i] + j)%10] ) % mod ;
            f[i][ (a[i] * j)%10] = ( f[i-1][j] + f[i][ (a[i] * j)%10] ) % mod ;
        }
    }
    for (int i = 0 ; i <= 9 ; i++ )
    {
        cout<<f[n-1][i]<<"\n";
    }
    return 0 ;
}