Flipping Game

PTA | 程序设计类实验辅助教学平台 (pintia.cn)https://pintia.cn/problem-sets/91827364500/exam/problems/91827370509?type=7&page=0题意 

        给定两个长度为n的字符串，代表初始灯泡状态和目标灯泡状态，经过k个回合，按下m个不同的开关，问有多少种方案能使灯泡从初始状态到目标状态

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <deque>
#include <cstdlib>
#include <stack>
#include <list>
#include <set>
#include <map>
#include <unordered_map>

using namespace std;

const int mod = 998244353;

int t;
int n, k, m, ans;
char start[105], target[105];
long long C[105][105], dp[105][105];

int main()
{
    C[0][0] = C[1][0] = C[1][1] = 1;//初始化组合数数组
    for(int i = 2;i < 105;i ++ )
    {
        C[i][0] = C[i][i] = 1;

        for(int j = 1;j < i;j ++ )
        {
            C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % mod;
        }

    }

    scanf("%d", &t);

    while(t -- )
    {
        ans = 0;

        scanf("%d %d %d", &n, &k, &m);
        scanf("%s %s", start, target);

        for(int i = 0;i < k;i ++ )
        {
            for(int j = 0;j <= 105;j ++ )
            {
                dp[i][j] = 0;
            }
        }

        int num = 0;//记录初始灯状态与目标灯状态相差个数

        for(int i = 0;i < n;i ++ )
        {
            if(start[i] != target[i])
            {
                num ++;
            }
        }

        //dp[i][j]代表在进行第i次操作后，仍有j个灯泡状态与目标状态不符
        dp[0][num] = 1;

        for(int i = 0;i < k;i ++ )
        {
            for(int j = 0;j <= n;j ++ )
            {
                if(dp[i][j])
                {
                    for(int k = 0;k <= j;k ++ )
                    {
                        if(n - j >= m - k)
                        {
                            int to = j - k + m - k;
                            //C[j][k]表示在j个不符灯泡中挑选k个灯泡的情况数 C[n - j][m - k]表示在符合的灯泡中挑选m - k的情况数
                            dp[i + 1][to] = (dp[i + 1][to] + C[j][k] * C[n - j][m - k] % mod * dp[i][j] % mod) % mod;
                        }
                    }
                }
            }
        }
        printf("%lld\n", dp[k][0]);
    }

    return 0;
}