CF_2069_C. Beautiful Sequence

题目链接：Problem - C - Codeforces

---

题目大意：

给定一个只包含{1，2，3}且长度为 n 的 数组a，

问：有多少个序列以'1'开始，'3'结尾长度至少为 $3$的子串,  对答案取 mod=998244353

---

DP：

用DP求子串数方法，

f[i]表示：以第 i 个字符结尾的最大子串数

  f[i] = f[i-1] + f[i-1] + 1

  f[i] =  ↑第 i - 1个字符结尾的最大子串数  + 在第 i - 1个最大子串数的基础上加上第 i 个字符 + 只选第 i 个字符

---

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<deque>
#include<stack>
#include<set>
#include<map>
#include<unordered_set>
#include<unordered_map>
#include<bitset>
#include<tuple>
#define inf 72340172838076673
#define int long long
#define endl '\n'
#define F first
#define S second
#define  mst(a,x) memset(a,x,sizeof (a))
using namespace std;
typedef pair<int, int> pii;

const int N = 200086, mod = 998244353;

int n, m;
int a[N];
int f[N];

void solve() {

    cin >> n;
    int c1 = 0, res = 0;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        f[i] = f[i - 1] % mod;
        if (a[i] == 1) c1++;// 记录1的数量
        else if (a[i] == 2) f[i] += f[i - 1] + c1;//不选这个数的子串+选这个数的子串+只选这个数和前面的1
        else res = (res + f[i]) % mod;//记录答案
    }
    
    cout << res << endl; 
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);
    
    int T = 1;
    cin >> T;
    while (T--) solve();
    
    return 0;
}