BZOJ-1875-HH去散步-SDOI2009-矩阵乘法

描述

无向图边长均为1, 求从给定地点A走到给定地点B共有多少条长度为 t 的路径(不能连续走重复的边). 模45989.

分析

• f[i][k] : 当走到边 i 的终点的时候路径长度为 j 的方案数量
• f[i][k] : 可以由 f[j][k-1] 转移来的条件是边 j 的终点是i的起点
• 那么就可以构造矩阵来解了...
• 不用矩阵是不是也能解？

代码

#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
        
        
using namespace std;
typedef long long lli;

const int maxn = 120 + 10;
const int mod = 45989;

struct Edge {
	int u, v;
};

vector
        
        
         
          edges;

struct Matrix {
	int n, m;
	int a[maxn][maxn];
	
	Matrix(int n=0, int m=0) : n(n), m(m) { memset(a, 0, sizeof(a)); }
	
	Matrix operator ^ (int p) {
		Matrix t = Matrix(n, n), A = *this;
		for(int i = 0; i < n; i++) t.a[i][i] = 1;
		for(; p; p>>=1, A=A*A) if(p & 1) t = t*A;
		return t;
	}
	
	Matrix operator * (Matrix A) {
		Matrix ret = Matrix(n, A.m);
		for(int i = 0; i < n; i++)
			for(int j = 0; j < A.m; j++) {
				ret.a[i][j] = 0;
				for(int k = 0; k < m; k++)
					ret.a[i][j] = ((lli)a[i][k]*A.a[k][j] + ret.a[i][j]) % mod;
			}
		return ret;
	}
	
	void print() {
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < m; j++)
				printf("%d ", a[i][j]);
			printf("\n");
		}
	}
};

void AddEdge(int u, int v) {
	edges.push_back((Edge){u, v});
	edges.push_back((Edge){v, u});
}

int main()
{
	int n, m, t, a, b;
	scanf("%d %d %d %d %d", &n, &m, &t, &a, &b);
	a--; b--; t--;
	for(int i = 0; i < m; i++) {
		int u, v;
		scanf("%d %d", &u, &v);
		u--; v--;
		AddEdge(u, v);
	}
	m += m;
	Matrix F = Matrix(1, m), A = Matrix(m, m);
	
	for(int i = 0; i < m; i++) {
		Edge& e = edges[i];
		if(e.u == a) F.a[0][i] = 1;
		
		for(int j = 0; j < m; j++) {
			Edge& e2 = edges[j];
			if(e.v == e2.u && i != (j^1)) A.a[i][j] = 1;
		}
	}
	F = F * (A^t);
	
	int ans = 0;
	for(int i = 0; i < m; i++) {
		Edge& e = edges[i];
		if(e.v == b) ans = (ans + F.a[0][i]) % mod;
	}
	printf("%d\n", ans);
	return 0;
}