题目描述
给出一个 $n$ 个点的有向图,每条边的权值都在 $[1,9]$ 之间。给出 $t$ ,求从 $1$ 到 $n$ ,经过路径边权和恰好为 $t$ 的方案数模2009。
输入
第一行包含两个整数,N T。 接下来有 N 行,每行一个长度为 N 的字符串。 第i行第j列为‘0‘表示从节点i到节点j没有边。 为‘1‘到‘9‘表示从节点i到节点j需要耗费的时间。
输出
包含一个整数,可能的路径数,这个数可能很大,只需输出这个数除以2009的余数。
样例输入
5 30
12045
07105
47805
12024
12345
样例输出
852
题解
矩阵乘法
傻题,显然如果没有边权的话就是裸的矩乘。如果有边权的话,拆边的话点数为 $9m$ 的,难以接受。
考虑拆点,将1个点拆成9个连成链,每次将出点对应边权的点连到入点上。这样点数就是 $9n$ 了。
时间复杂度 $O((9n)^3)$
#include <cstdio> #include <cstring> char str[15]; int m; struct data { int v[100][100]; data() {memset(v , 0 , sizeof(v));} int *operator[](int a) {return v[a];} data operator*(data &a) { data ans; int i , j , k; for(i = 0 ; i < m ; i ++ ) for(j = 0 ; j < m ; j ++ ) for(k = 0 ; k < m ; k ++ ) ans[i][j] = (ans[i][j] + v[i][k] * a[k][j]) % 2009; return ans; } }A; data pow(data x , int y) { data ans; int i; for(i = 0 ; i < m ; i ++ ) ans[i][i] = 1; while(y) { if(y & 1) ans = ans * x; x = x * x , y >>= 1; } return ans; } int main() { int n , k , i , j; scanf("%d%d" , &n , &k) , m = n * 9; for(i = 0 ; i < n ; i ++ ) { scanf("%s" , str); for(j = 0 ; j < 8 ; j ++ ) A[i * 9 + j][i * 9 + j + 1] = 1; for(j = 0 ; j < n ; j ++ ) if(str[j] != ‘0‘) A[i * 9 + str[j] - ‘1‘][j * 9] = 1; } printf("%d\n" , pow(A , k)[0][n * 9 - 9]); return 0; }