HDU 3658 How many words

HDU_3658

可以用f[i][j]表示第i个字符为j时满足要求（有至少一对相邻的字符的ASC码值之差是32）的合法（任意一对相邻的字符的ASC码值之差都不超过32）字符串个数，g[i][j]表示第i个字符为j时不满足要求但合法的字符串个数。找到递推关系之后构造矩阵并用二分矩阵的方法求解即可。

#include<stdio.h>
#include<string.h>
#define MAXD 110
#define D 1000000007
int N;
struct Matrix
{
int a[MAXD][MAXD];
Matrix()
{
memset(a, 0, sizeof(a));
}
};
Matrix multiply(Matrix &x, Matrix &y)
{
int i, j, k;
Matrix z;
for(k = 0; k < 104; k ++)
for(i = 0; i < 104; i ++)
if(x.a[i][k])
{
for(j = 0; j < 104; j ++)
if(y.a[k][j])
z.a[i][j] = (z.a[i][j] + (long long)x.a[i][k] * y.a[k][j]) % D;
}
return z;
}
void powmod(Matrix &unit, Matrix &mat, int n)
{
while(n)
{
if(n & 1)
unit = multiply(mat, unit);
n >>= 1;
mat = multiply(mat, mat);
}
}
void solve()
{
int i, j;
Matrix mat, unit;
long long ans = 0;
for(i = 52; i < 104; i ++)
unit.a[i][0] = 1;
for(i = 0; i < 26; i ++)
{
for(j = 0; j <= 26 + i; j ++)
mat.a[i][j] = 1;
mat.a[i][i + 78] = 1;
}
for(i = 0; i < 26; i ++)
{
for(j = i; j < 52; j ++)
mat.a[i + 26][j] = 1;
mat.a[i + 26][i + 52] = 1;
}
for(i = 0; i < 26; i ++)
for(j = 52; j < 78 + i; j ++)
mat.a[i + 52][j] = 1;
for(i = 0; i < 26; i ++)
for(j = 52 + i + 1; j < 104; j ++)
mat.a[i + 78][j] = 1;
powmod(unit, mat, N - 1);
for(i = 0; i < 52; i ++)
ans += unit.a[i][0];
printf("%I64d\n", ans % D);
}
int main()
{
int t;
scanf("%d", &t);
while(t --)
{
scanf("%d", &N);
solve();
}
return 0;
}