bzoj 1875 [SDOI2009]HH去散步（矩乘）

 

Description

HH有个一成不变的习惯，喜欢饭后百步走。所谓百步走，就是散步，就是在一定的时间 内，走过一定的距离。 但是同时HH又是个喜欢变化的人，所以他不会立刻沿着刚刚走来的路走回。 又因为HH是个喜欢变化的人，所以他每天走过的路径都不完全一样，他想知道他究竟有多 少种散步的方法。 现在给你学校的地图（假设每条路的长度都是一样的都是1），问长度为t，从给定地 点A走到给定地点B共有多少条符合条件的路径

Input

第一行：五个整数N，M，t，A，B。其中N表示学校里的路口的个数，M表示学校里的 路的条数，t表示HH想要散步的距离，A表示散步的出发点，而B则表示散步的终点。 接下来M行，每行一组Ai，Bi，表示从路口Ai到路口Bi有一条路。数据保证Ai = Bi，但 不保证任意两个路口之间至多只有一条路相连接。 路口编号从0到N − 1。 同一行内所有数据均由一个空格隔开，行首行尾没有多余空格。没有多余空行。 答案模45989。

Output

一行，表示答案。

Sample Input

4 5 3 0 0
0 1
0 2
0 3
2 1
3 2

Sample Output

4

HINT

对于30%的数据，N ≤ 4，M ≤ 10，t ≤ 10。
对于100%的数据，N ≤ 20，M ≤ 60，t ≤ 230，0 ≤ A,B

 

【思路】

      

       矩阵乘法，边点互换

       题目中有不能走回头路的要求但同时存在重边。边点互换一下就可以满足这个要求啦。按照边的连接构造A矩阵并求出A^(t-1)，新建一个虚拟节点，使之向a的所有出边连通，B*A即可求出虚拟节点到各边长度为t的路径数，然后统计所有与b相连的边。

 

【代码】

#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

const int maxn = 25*5;
const int MOD = 45989;
struct Edge { int v,next;
}es[maxn<<1];

struct Matrix {
    int r,c,N[maxn][maxn];
    void init(int r,int c) {
        this->r=r,this->c=c;
        memset(N,0,sizeof(N));
    }
    Matrix operator * (const Matrix B) const{
        Matrix C; C.init(r,B.c);
        for(int i=0;i<r;i++)
            for(int j=0;j<B.c;j++)
                for(int k=0;k<c;k++)
                    C.N[i][j]=(C.N[i][j]+N[i][k]*B.N[k][j])%MOD;
        return C;
    }
    Matrix pow(int p) {
        Matrix tmp=*this,ans;
        ans.init(r,r);
        for(int i=0;i<r;i++) ans.N[i][i]=1;
        while(p) {
            if(p&1) ans=ans*tmp;
            tmp=tmp*tmp; p>>=1;
        }
        return ans;
    }
}A,B;

int n,m,t,a,b,u,v;
int front[maxn],en=-1;
void adde(int u,int v) {
    en++; es[en].next=front[u]; es[en].v=v; front[u]=en;
}

int main() {
    freopen("in.in","r",stdin);
    freopen("out.out","w",stdout);
    memset(front,-1,sizeof(front));
    scanf("%d%d%d%d%d",&n,&m,&t,&a,&b);
    for(int i=0;i<m;i++) {
        scanf("%d%d",&u,&v);
        adde(u,v),adde(v,u);
    }
    en+=3;
    A.init(en,en),B.init(en,en);
    for(int i=front[a];i>=0;i=es[i].next) B.N[en-2][i]=1;
    for(int i=0;i<n;i++) {
        for(int u=front[i];u>=0;u=es[u].next) {
            int j=es[u].v;
            for(int v=front[j];v>=0;v=es[v].next) 
                if(u!=(v^1))  A.N[u][v]++;
        }
    }
    A=A.pow(t-1);
    A=B*A;
    int ans=0;
    for(int i=front[b];i>=0;i=es[i].next)
        ans=(ans+A.N[en-2][i^1])%MOD;
    printf("%d",ans);
    return 0;
}