BZOJ 1297 迷路(矩阵快速幂)

很容易想到记忆化搜索的算法。 令dp[n][T]为到达n点时时间为T的路径条数。则dp[n][T]=sigma(dp[i][T-G[i][n]]); 但是空间复杂度为O(n*T),时间复杂度O(n*n*T).

虽然本题的n<=10,但T最大可到1e9。行不通。

如果题目中的边的权值非0即1的话，显然1-n的长度为T的路径中数为 该图的邻接矩阵的T次幂。

实际上题目中的边权值<10. 可以用拆点的方法转化为边权值非0即1的情况。

即 将图中的每个点拆成至多9个点，首先将每个点的第i个点和第i+1个点连一条权值为1的边。另外，如果原图中Eij=m，则将新图的第i个点拆成的第m点和j点的第一个点连一条权值

为1的边。这样就完全转化为我们可以解决的问题形式了。矩阵快速幂可以在O(n'^3*logT)的时间内完成。

# include <cstdio>
# include <cstring>
# include <cstdlib>
# include <iostream>
# include <vector>
# include <queue>
# include <stack>
# include <map>
# include <set>
# include <cmath>
# include <algorithm>
using namespace std;
# define lowbit(x) ((x)&(-x))
# define pi 3.1415926535
# define eps 1e-
# define MOD
# define INF
# define mem(a,b) memset(a,b,sizeof(a))
# define FOR(i,a,n) for(int i=a; i<=n; ++i)
# define FO(i,a,n) for(int i=a; i<n; ++i)
# define bug puts("H");
# define lch p<<,l,mid
# define rch p<<|,mid+,r
# define mp make_pair
# define pb push_back
typedef pair<int,int> PII;
typedef vector<int> VI;
# pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long LL;
int Scan() {
    int res=, flag=;
    char ch;
    if((ch=getchar())=='-') flag=;
    else if(ch>=''&&ch<='') res=ch-'';
    while((ch=getchar())>=''&&ch<='')  res=res*+(ch-'');
    return flag?-res:res;
}
void Out(int a) {
    if(a<) {putchar('-'); a=-a;}
    if(a>=) Out(a/);
    putchar(a%+'');
}
const int N=;
//Code begin...

struct Matrix{int matrix[][];}a, sa, unit;
int n, T;
char G[][];

Matrix Mul(Matrix a,Matrix b) //矩阵乘法(%m)
{
    Matrix c;
    for (int i=; i<*n; ++i) for (int j=; j<*n; ++j) {
          c.matrix[i][j]=;
          for (int l=; l<*n; ++l) c.matrix[i][j]+=a.matrix[i][l]*b.matrix[l][j];
          c.matrix[i][j]%=;
    }
    return c;
}
Matrix Cal(int exp)  //矩阵快速幂
{
    Matrix p=a, q=unit;
    if (exp==) return p;
    while (exp!=) {
        if (exp&) exp--, q=Mul(p,q);
        else exp>>=, p=Mul(p,p);
    }
    return Mul(p,q);
}
int main ()
{
    scanf("%d%d",&n,&T);
    FO(i,,n) scanf("%s",G[i]);
    FO(i,,n) FOR(j,,) a.matrix[i*+j][i*+j+]=;
    FO(i,,n) FO(j,,n) {
        if (G[i][j]=='') continue;
        a.matrix[i*+G[i][j]-''][j*]=;
    }
    unit=a; sa=Cal(T-);
    printf("%d\n",sa.matrix[][*(n-)]);
    return ;
}