Luogu 2151 [SDOI2009]HH去散步

BZOJ 1875

矩阵乘法加速递推。

如果不要求不能走同一条边，那么直接构造出矩阵快速幂即可，但是不走相同的道路，怎么办？

发现边数$m$也很小，我们直接把$2 * m$开成一个矩阵，相当于记录上一条边走过了编号为$j$的边的方案总数，这样子我们在构造转移矩阵的时候就可以不用计算往回走带来的贡献了。

时间复杂度$O(m^3log(MaxInt))$。

Code：

#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
 
const int N = ;
const int M = ;
const int P = ;
 
int n, m, tot = , from[M], to[M];
 
template <typename T>
inline void read(T &X) {
    X = ; char ch = ; T op = ;
    for(; ch > '' || ch < ''; ch = getchar())
        if(ch == '-') op = -;
    for(; ch >= '' && ch <= ''; ch = getchar())
        X = (X << ) + (X << ) + ch - ;
    X *= op;
}
 
template <typename T>
inline void inc(T &x, T y) {
    x += y;
    if(x >= P) x -= P;
}
 
struct Matrix {
    int len, wid, s[M][M];
 
    inline void init() {
        len = wid = ;
        memset(s, 0LL, sizeof(s));
    } 
 
    friend Matrix operator * (const Matrix &x, const Matrix &y) {
        Matrix res; res.init();
        res.len = x.len, res.wid = y.wid;
        for(int k = ; k < x.wid; k++)
            for(int i = ; i < x.len; i++)
                for(int j = ; j < y.wid; j++)
                    inc(res.s[i][j], (int) (1LL * x.s[i][k] * y.s[k][j] % P));
        return res;
    }
 
} f, tra;
 
inline Matrix fpow(Matrix x, int y) {
    Matrix res; res.init();
    res.len = x.len, res.wid = x.wid;
    for(int i = ; i < res.len; i++) res.s[i][i] = ;
 
    for(; y > ; y >>= ) {
        if(y & ) res = res * x;
        x = x * x;
    }
 
    return res;
}
 
inline int opp(int now) {
    return (now & ) ? now +  : now - ;
}
 
int main() {
    int tim, st, ed;
    read(n), read(m), read(tim), read(st), read(ed);
    for(int x, y, i = ; i <= m; i++) {
        read(x), read(y);
        ++tot, from[tot] = x, to[tot] = y;
        ++tot, from[tot] = y, to[tot] = x;
    }
 
    f.init();f.len = , f.wid =  * m;
    for(int i = ; i <= tot; i++)
        if(from[i] == st) ++f.s[][i - ];
    tra.init(); tra.len = tra.wid =  * m;
    for(int i = ; i <= tot; i++)
        for(int j = ; j <= tot; j++) {
            if(j == i || j == opp(i)) continue;
            if(to[i] == from[j])
                ++tra.s[i - ][j - ];
        }
 
    tra = fpow(tra, tim - );
    f = f * tra;
 
    int ans = ;
    for(int i = ; i <= tot; i++)
        if(to[i] == ed) inc(ans, f.s[][i - ]);
 
    printf("%d\n", ans);
    return ;
}