【GDOI三校联考】字串转换

Description:

给出两个字符串S和T，它们长度相等，都是由小写字母‘a’、‘b’、‘c’组成。每次操作，你可以对字符串S进行如下操作之一：

1、 选择S的一个字符‘a’,把该字符变成字符‘b’，费用是cost[0]。

2、 选择S的一个字符‘b’,把该字符变成字符‘c’，费用是cost[1]。

3、 选择S的一个字符‘c’,把该字符变成字符‘a’，费用是cost[2]。

你的目标是把字符串S变成T，但是总费用不能超过给定的maxCost，有多少种不同的操作方式？下面定义“不同的操作方式”：假设有两种操作方式A和B，只要满足如下两种情况之一，就认为是“不同的操作方式”：

1、A和B的操作次数不同。

2、存在一个整数K，在第K轮操作时，操作方式A是选择第i个字符进行变化，而操作方式B是选择第j个字符进行变化，且i不等于j。

题解：

woc真的是数论学傻了，连矩阵乘法裸题都不会了。

woc又发现连矩阵乘法的求和套路都忘了……

Code：

#include<cstdio>
#include<cstring>
#define ll long long
#define fo(i, x, y) for(int i = x; i <= y; i ++)
#define fd(i, x, y) for(int i = x; i >= y; i --)
using namespace std;

const int mo = 1e9 + 7;

char s[20], t[20];
int n, a[20], c[3], mx;
int num[20][20], tn;
ll b[200][200], e[200][200], d[200][200];

int main() {
    scanf("%s", s + 1); scanf("%s", t + 1);
    scanf("%d %d %d %d", &c[0], &c[1], &c[2], &mx);
    n = strlen(s + 1);
    fo(i, 1, n) {
        if(s[i] == t[i]) continue;
        int y = t[i] - 'a' - 1;
        if(s[i] > t[i]) y += 3;
        fo(j, s[i] - 'a', y) mx -= c[j % 3], a[i] ++;
    }
    int g = mx / (c[1] + c[2] + c[0]);
    fo(i, 0, n) fo(j, 0, n - i) num[i][j] = ++ tn;
    fo(i, 0, n) fo(j, 0, n - i) {
        if(i > 0) b[num[i][j]][num[i - 1][j]] = i;
        if(j > 0) b[num[i][j]][num[i + 1][j - 1]] = j;
        if(i + j < n) b[num[i][j]][num[i][j + 1]] = n - (i + j);
    }
    g *= 3; fo(i, 1, n) g += a[i];
    tn ++; b[num[0][0]][tn] = 1; b[tn][tn] = 1;
    fo(i, 1, tn) d[i][i] = 1;
    g ++;
    while(g) {
        if(g & 1) {
            fo(i, 1, tn) fo(j, 1, tn) e[i][j] = d[i][j], d[i][j] = 0;
            fo(k, 1, tn) fo(i, 1, tn) fo(j, 1, tn)
                d[i][j] = (d[i][j] + e[i][k] * b[k][j]) % mo;
        }
        fo(i, 1, tn) fo(j, 1, tn) e[i][j] = b[i][j], b[i][j] = 0;
        fo(k, 1, tn) fo(i, 1, tn) fo(j, 1, tn) b[i][j] = (b[i][j] + e[i][k] * e[k][j]) % mo;
        g /= 2;
    }
    int p1 = 0, p2 = 0;
    fo(i, 1, n) {
        if(a[i] == 1) p1 ++;
        if(a[i] == 2) p2 ++;
    }
    printf("%lld", d[num[p1][p2]][tn]);
}