E - Easy genes

E

题目大意

求长度为 $l$ 的不包含子串 101 和 111 的 01 串有多少个。

solution

考虑对于 01 串 101 和 111 建立关于字符串匹配的自动机（AC自动机），如图：

当匹配到 $4$ 号节点的时候说明匹配到了一个 101 或者 111，我们最终要求的就是匹配不到 $4$ 号节点的串的数目。

记录 $f[i][j]$ 为当前填到第 $i$ 个数，匹配到自动机上的状态 $j$ 的方案数，这样我们就可以根据自动机的转移来建立 dp 方程如下：

$$f[i][0]=f[i-1][0]+f[i-1][2]$$

$$f[i][1]=f[i-1][0]$$

$$f[i][2]=f[i-1][1]+f[i-1][3]$$

$$f[i][3]=f[i-1][1]$$

这样我们就得到了一个时间复杂度为 $O(m)$ 的做法。
这个 dp 方程可以通过矩阵快速幂优化到 $O(\log n)$。

#include <cstdio>
#include <algorithm>
#include <vector>
#define mp(x, y) std::make_pair(x, y)
#define pb push_back
#define FOR(x, l, r) for(int x = l; x <= r; x++)
#define DO(x, r, l) for(int x = r; x >= l; x--)
#define vi std::vector
#define ALL(x) x.begin(), x.end()
#define AL(x) x.begin() + 1, x.end()
#define fir first
#define sec second

const int MAXN = 1e8 + 1;
const int MAXM = 30;
const int MOD = 2005;
const int oo = 0x3f3f3f3f;

typedef long long ll;
// typedef __int128 lll;
typedef unsigned long long ull;
typedef std::pair<ll, ll> pll;
typedef std::pair<int, int> pii;

struct Matrix {
	vi<vi<int> > v;
	Matrix(int x = 0) : v(4, vi<int>(4)) {
		FOR(i, 0, 3) v[i][i] = x;
	}
	
	Matrix operator * (Matrix t) {
		Matrix res;
		FOR(i, 0, 3)
			FOR(j, 0, 3)
				FOR(k, 0, 3)
					res.v[i][j] = (res.v[i][j] + v[i][k] * t.v[k][j]) % MOD;
		return res;
	}
};

Matrix fp(Matrix a, int b) {
	Matrix res = 1;
	for(; b; b >>= 1, a = a * a)
		if(b & 1) res = res * a;
	return res;
}

int main() {
	int n;
	while(~scanf("%d", &n)) {
		Matrix m;
		m.v[0][0] = m.v[0][1] = m.v[1][2] = m.v[1][3] = m.v[2][0] = m.v[3][2] = 1;
		m = fp(m, n);
		printf("%d\n", (m.v[0][0] + m.v[0][1] + m.v[0][2] + m.v[0][3]) % MOD);
	}
	return 0;
}