【Bzoj1875】HH去散步

题目描述

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

输入输出格式

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

输出格式：
一行，表示答案。

输入输出样例

输入样例#1：
4 5 3 0 0
0 1
0 2
0 3
2 1
3 2
输出样例#1：
4
说明

对于30%的数据，N ≤ 4，M ≤ 10，t ≤ 10。

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

---

　　对于一般的路径题，我们直接把邻接矩阵k次方然后f[i][j]就是i到j路径长为k的方案数了。但是对于这道题，要求不能走回头的路径，所以这种方法显然是不可以成功的。umm…这时候，我们考虑边点互化，大致就是说如果s的终点为t的起点，那么f[s][t]=1，所以这样我们把矩阵k-1次方就可以求出任意两条边之间距离为k-1的方案数了。(如果两条边u,v之间经过k-1个点，也就是说从u的起点到v的终点之间就是长度为k的路径了)。对于这道题的答案，也就是把矩阵k-1次方以后。用一个1*cnt(边数)的矩阵，将与起点连接的边标为1，然后去乘以转移矩阵。举个栗子，答案矩阵中1，3，7被标为1，乘完以后答案矩阵ans[1][1]=f[1][1]+f[3][1]+f[7][1]、ans[1][k]=f[1][k]+f[3][k]+f[7][k]。也就是起点通过t条边所达到k边的方案。然后在这个矩阵中扫一遍，如果k与终点相连，累加k的答案。注意这里把cnt初值赋为1，所以第一条双向边存起来就是2，3。第二条4，5。以此类推，之后用i^1就能直接找到i的反向边。

#include <cstdio>
#include <algorithm>
#include <cstring>

#define Rep(i,s,t) for(int i=s;i<=t;i++)

using namespace std;

const int maxx = 120 + 5;
const int p = 45989;

int n,m,t,B,E,cnt=1,x,y,sum;
int head[maxx],next[maxx],to[maxx];

void Add(int x,int y){
    next[++cnt]=head[x];to[cnt]=y;head[x]=cnt;
}

namespace matrix_mul{
    struct mat{
        int x[maxx][maxx];
        mat(){memset(x,0,sizeof(x));}
    };

    mat mul(mat a,mat b){
        mat tmp;Rep(i,1,m*2+1) Rep(j,1,m*2+1) Rep(k,1,m*2+1)
        tmp.x[i][j] = (tmp.x[i][j]+a.x[i][k]*b.x[k][j])%p;
        return tmp;
    }

    mat pwr(mat a,int n){
        mat ans = a,tmp = a;n--;
        while(n){if(n&1) ans=mul(ans,tmp);tmp=mul(tmp,tmp);n>>=1;}
        return ans;
    }
}

int main(){
    using namespace matrix_mul;
    mat Ans,tmp;
    scanf("%d%d%d%d%d",&n,&m,&t,&B,&E);B++;E++;
    Rep(i,1,m) scanf("%d%d",&x,&y),x++,y++,Add(x,y),Add(y,x);
    for(int i=head[B];i;i=next[i]) Ans.x[1][i]=1;
    for(int i=2;i<=cnt;i++)
        for(int j=head[to[i]];j;j=next[j])
            if(i != (j^1))
                tmp.x[i][j]++;    
    tmp = pwr(tmp,t-1);
    Ans = mul(Ans,tmp);
    for(int i=2;i<=cnt;i++)
        if(to[i]==E)
            sum=(sum+Ans.x[1][i])%p;
    printf("%d",sum);
    return 0;
}