HDU 6145 Arithmetic of Bomb II or Aizu 2789 Compressed Formula 表达式展开求值 矩阵快速幂

链接

http://acm.hdu.edu.cn/showproblem.php?pid=6145
http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=2789

题意

给n个表达式，每个表达式重复k次，问这个表达式的值是多少

思路

一开始自己打算分类讨论的，结果不知道哪里写丑了……写了200+还是wa，后来去问snowy_smile大神，他告诉我这题和百度之星2017复赛02一样，于是自己去best coder看了看杜教的代码，结合icpc camp的题解知道了做法（每次就你废话最多……）

考虑维护当前表达式前面多项式的和s，当前多项式的乘积pd，以及当前多项式除去最后一个因子的乘积pd2，为了方便再加一个值为1的常量$\epsilon$，面对后面加进来的字符，有如下的矩阵运算（要详细解释也很麻烦，在纸上推演下很容易就懂了）：

1. 式子后面添加一个数字d

   相当于

   ${\begin{bmatrix} s \\ pd \\ pd2 \\ \epsilon \end{bmatrix}} ^ T$ = ${\begin{bmatrix} s^{'} \\ pd^{'} \\ pd2^{'} \\ \epsilon^{'} \end{bmatrix}}^T$ * $\begin{bmatrix} 1 & 0 & 0 & 0 \\ 0& 10 & 0 & 0 \\ 0 & d & 1 & 0 \\ 0 & 0 & 0 & 1 \end{bmatrix}$

   即s，pd2和$\epsilon$保持不变，当前单项式的最后一个系数进位并加上新的数字

2. 式子后面添加一个加号或减号

   相当于

   ${\begin{bmatrix} s \\ pd \\ pd2 \\ \epsilon \end{bmatrix}} ^ T$ = ${\begin{bmatrix} s^{'} \\ pd^{'} \\ pd2^{'} \\ \epsilon^{'} \end{bmatrix}}^T$ * $\begin{bmatrix} 1 & 0 & 0 & 0 \\ 1& 0 & 0 & 0 \\ 0 & 0 & 0 & 0 \\ 0 & 0 & \pm1 & 1 \end{bmatrix}$

   即$\epsilon$保持不变，s加上当前单项式pd置为0，pd2变为±对应的±1

3. 式子后面添加一个乘号

   相当于

   ${\begin{bmatrix} s \\ pd \\ pd2 \\ \epsilon \end{bmatrix}} ^ T$ = ${\begin{bmatrix} s^{'} \\ pd^{'} \\ pd2^{'} \\ \epsilon^{'} \end{bmatrix}}^T$ * $\begin{bmatrix} 1 & 0 & 0 & 0 \\ 0& 0 & 1 & 0 \\ 0 & 0& 0 & 0 \\ 0 & 0 & 0 & 1 \end{bmatrix}$

   即s，$\epsilon$保持不变，pd置为0（由于*后面跟着的一定是数字，所以pd在当前矩阵乘上情况1的矩阵后又会恢复正常），pd2变为pd

根据输入的字符不断使用矩阵乘法，最后快速幂即可。

求答案的时候用[0，0，1，1]这个矩阵去求s和pd，两者相加就是答案

代码

aizu 2789的代码，hdu 6145也是相同

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long LL;
const int N = 4;
const int MOD = 1e9 + 7;
typedef LL Mat[N][N];

int n, r;
char s[20];
Mat tmp, ret, ans;

void mul(Mat &a, Mat &b) {
  for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) {
    tmp[i][j] = 0;
    for (int k = 0; k < N; ++k) tmp[i][j] += a[i][k] * b[k][j];
  }
  for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) a[i][j] = tmp[i][j] % MOD;
}

void Pow(Mat &a, int b) {
  for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) ret[i][j] = (i==j);
  for (; b; b >>= 1) {
    if (b&1) mul(ret, a);
    mul(a, a);
  }
  for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) a[i][j] = ret[i][j];
}

void solve() {
  Mat pre, nxt;
  for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) pre[i][j] = (i==j);
  for (int k = 0; s[k]; ++k) {
    for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) nxt[i][j] = 0;
    nxt[0][0] = nxt[3][3] = 1;
    if (s[k] == '+' || s[k] == '-') {
      nxt[1][0] = 1; nxt[3][2] = s[k] == '+' ? 1 : MOD - 1;
    }
    else if ('0' <= s[k] && s[k] <= '9') {
      nxt[1][1] = 10; nxt[2][2] = 1; nxt[2][1] = s[k] - '0';
    }
    else {
      nxt[1][2] = 1;
    }
    mul(pre, nxt);
  }
  Pow(pre, r);
  mul(ans, pre);
}

int main() {
  scanf("%d", &n);
  for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) ans[i][j] = (i==j);
  for (int i = 0; i < n; ++i) {
    scanf("%d%s", &r, s);
    solve();
  }
  LL s = (ans[2][0] + ans[3][0] + MOD) % MOD, pd = (ans[2][1] + ans[3][1] + MOD) % MOD;
  printf("%lld\n", (s + pd) % MOD);
}