【以前的空间】bzoj1009 [HNOI2008]GT考试

  动态规划+kmp+矩阵快速幂

  关于这题可以写出一个dp方程(f[i,j]表示准考证前i位中后j位为不吉利的数字的前j位的情况的个数）

  f[i,j]=Σf[i-1,k]，其中j表示不吉利数字前k个数字加上某个数字后变成为不吉利数字的前j位（比如不吉利数字122123，然后现在k=5，那么如果填个3，j=6（123123）；填个2，j=3（122）；填个1，j=1（1）；填个0,j=0。

  然后我们就可以发现……好像可以用kmp算法来优化每次k+某个数字可以转移到的j的位置……因为j包括了前k个数字，那么j一定是在k的失配函数+1位或者不存在。

  但是这样的话时间复杂度依然因为n过大而报表……。

  于是神奇的矩阵来了！

  没有发现似乎式子是递推式么？！

  也就是F(n)是可以通过F(n-1)得到的！F(n)表示f[n,0],f[n,1],f[n,2]……f[n,m-1]的集合，或者说……一个1*(m-1)的矩阵。

  矩阵的定义白书有句话说的特别好：“把一个向量v变成另一个向量v'，并且v'的每个分量都是v各个分量的线性组合”。什么是线性组合……就是一个线性方程组！对于这题来说，线性方程组就是：

  F(n)=A*F(n-1) （这里A叫友矩阵，其实就是转移关系）

  如果写出矩阵，式子就变成

  [f(n,0)  ]     [a[0,0]  ,a[1,0]  ,a[2,0]  ,……，a[m-1,0]  ]   [f(n-1,0)  ]

  [f(n,1)  ]     [a[0,1]  ,a[1,1]  ,a[2,1]  ,……，a[m-1,1]  ]   [f(n-1,1)  ]

  [f(n,2)  ]   = [a[0,2]  ,a[1,2]  ,a[2,2]  ,……，a[m-1,2]  ] * [f(n-1,2)  ]

  [ ……   ]     [  ……                                     ]   [ ……     ]

  [f(n,m-1)]     [a[m-1,1],a[m-1,2],a[m-1,3],……，a[m-1,m-1]]   [f(n-1,m-1)]

  式子中的a[i,j]表示不吉利数字前i位加上某个数字后可以转的j位的数量。

  前面说的线性组合是什么意思？矩阵其实是一个线性方程组，形式就是：

  a[0,0]*f[n-1,0]+a[1,0]*f(n-1,1)+a[2,0]*f(n-1,2)+……+a[m-1,0]*f[n-1,m-1]=f[n,0];

  a[0,1]*f[n-1,0]+a[1,1]*f(n-1,1)+a[2,1]*f(n-1,2)+……+a[m-1,1]*f[n-1,m-1]=f[n,1];

  a[0,2]*f[n-1,0]+a[1,2]*f(n-1,1)+a[2,2]*f(n-1,2)+……+a[m-1,2]*f[n-1,m-1]=f[n,2];

  ………………………………………………… 

  a[0,m-1]*f[n-1,0]+a[1,m-1]*f（n-1,1)+a[2,m-1]*f(n-1,2)+……+a[m-1,m-1]*f[n-1,m-1]=f[n,m-1]

  看出来了么？

  回到题目上，如果递推算F(n),显然不行，但是我们有一个递推式F(n)=A*F(n-1),它可以化为F(n)=A^n*F(0)。

  那么现在要做的就是如何优化A^n，想到快速幂？是的，矩阵也有快速幂（具体到网上找吧，这就不是理解方面的了）

  然后答案就是F(n)=A^n*F(0)。但是这个F（0）没什么意义啊，f[0,0]=1，其他都等于0。

  这样像上面那样写出线性方程组的形式你就发现这个F(n)=a[0,0]+a[0,1]+a[0,2]+……+a[0,m-1],那最后就没有必要算F(0)*A^n,直接算上a[0,i]的和就行了！

  然后联系一下A的意义……是不是有点感觉？A^i就是表示a[j,k]原来是前j位递推i次后能变成前k位的方案！蒟蒻感觉好神奇！

  然后就好像初步学会了矩阵快速幂优化……

2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
type
  arr=array[0..25,0..25]of longint;
 
var
  a,c,tmp:arr;
  b,p:array[0..30]of longint;
  i,j,k,l,n,m,mm,sum:longint;
  ch:char;
 
 
procedure mul(var x,y:arr);
var
  i,j,k:longint;
begin
  for i:=0 to m-1 do
    for j:=0 to m-1 do begin
      tmp[i,j]:=0;
      for k:=0 to m-1 do
        tmp[i,j]:=(tmp[i,j]+x[i,k]*y[k,j])mod mm;
    end;
  for i:=0 to m-1 do
    for j:=0 to m-1 do
      x[i,j]:=tmp[i,j];
end;
 
 
begin
  readln(n,m,mm);
  for i:=1 to m do begin
    read(ch);
    b[i]:=ord(ch)-ord('0');
  end;
  j:=0;
  p[1]:=0;
  for i:=2 to m do begin
    while (j>0) and (b[j+1]<>b[i]) do j:=p[j];
    if b[j+1]=b[i] then inc(j);
    p[i]:=j;
  end;
  for i:=0 to m-1 do
    for j:=0 to 9 do begin
      k:=i;
      while (k>0) and (b[k+1]<>j) do k:=p[k];
      if b[k+1]=j then inc(k);
      if k<>m then c[i,k]:=(c[i,k]+1)mod mm;
    end;
  fillchar(a,sizeof(a),0);
  for i:=0 to m-1 do
    a[i,i]:=1;
  while n>0 do begin
    if n and 1=1 then mul(a,c);
    mul(c,c);
    n:=n>>1;
  end;
  sum:=0;
  for i:=0 to m-1 do
    sum:=(sum+a[0,i]) mod mm;
  writeln(sum);
  readln;
  readln;
end.