CODEVS 1482 路线统计

CODEVS 1482路线统计

【问题描述】

给出一个n个点有向图，求从s点到f点恰好经过时间t的路径总数。不能在某个点停留，可以重复的走各点。

 

【输入数据】

第一行包含一个整数n, 所有点是从0到n-1编号.

接下来n行,每行包含n个字符. 第i行第j个字符表示i到j需要的时间，字符只可能是’1’到’5’,或者是’.’表示i不能到达j， 保证主对角线都是’.’。

接下来一行3个整数s, f, t。

 

【输出数据】

输出总方案数mod502630的值。

 

【样例输入】route.in

3

.12

2.1

12.

0 2 5

 

【样例输出】route.out

8

 

【数据范围及提示】

对于20%的数据, 输入的字符不是’1’就是’.’;

对于100%的数据, 1 <= n <= 10; 1<= s,f <= n; 1 <= t <= 10^9

 

【解题报告】

第一眼，dfs，但看t的范围，显然超时。

再看，点很少（矩阵的n次幂耗时少），时间很多（走法复杂）

矩阵乘法的标志啊！！！

   

我们知道无边权图从s经k步到f怎么求：即01矩阵：

建立矩阵A  当且仅当存在一条边i->j ，A(i,j)=1。

令C=A*A，那么C(i,j)=ΣA(i,k)*A(k,j)，实际上就等于从点i到点j恰好经过2条边的路径数（枚举k为中转点）

类似地，C*A的第i行第j列就表示从i到j经过3条边的路径数。

同理，如果要求经过k步的路径数，我们只需要二分求出A^k即可。

 

但图有边权，怎么办？

两个字：拆点！

将每个点之间的关系用矩阵存储，i能1步到j标记为1，不能到标记为0，注意题中边权为1-5，则可拆点，将每个点拆成边权个点，如图：

但这样还不够，我们要建（n*n*5）^2 = 500*500的矩阵，矩阵乘法t达到500 ^ 3 这显然太多了。

 

于是：我们遇到一个边i,j，权为c，把它拆成i –> i+n*1 -> i+n*2 ->… -> i+n*(c-1)-> j

如图:

于是就只有(n*5)^2=50*50的矩阵了。

 

拆完点，矩阵就变成了01矩阵

 

则这个矩中的A[i][j]就保存了1步能从i到j的方案数，要求t步，则直接将矩阵自乘t次即得答案。

 

下面是华丽丽の代码实践：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int mod=502630;
int f,t,s,n;
char c;
int ans;
int cnt;
struct node
{
    int f[60][60];
}E,x;
void clean()
{
	for(int i=0;i<=n;i++)
	{
		E.f[i][i]=1;
	}
}
node cheng(node a,node b)    //矩阵乘法
{
	node ne;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			ne.f[i][j]=0;
			for(int k=0;k<n;k++)
			{
				ne.f[i][j]=(ne.f[i][j]+((long long)a.f[i][k]*b.f[k][j])%mod)%mod;
			}
		}
	}
	return ne;
}
int answer()
{
	node ne=x;
	node ass=E;
	int b=t;
	while(b)      //矩阵求幂
	{
		if(b&1)
		{
			ass=cheng(ass,ne);
		}
		b/=2;
		ne=cheng(ne,ne);
	}
	return ass.f[s][f];
}
int main()
{
	freopen("route.in","r",stdin);
	freopen("route.out","w",stdout);
	scanf("%d\n",&n);
	cnt=n;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			scanf("%c",&c);
			if(c>='0'&&c<='9')
			{	
				cnt=c-'0';
				for(int k=1;k<cnt;k++)  //拆点
				{
					x.f[n*(k-1)+i][n*k+i]=1;
				}
				x.f[n*(cnt-1)+i][j]=1;
			}
		}
		scanf("\n");
	}
	n*=5;                       //最后n要乘5
	scanf("%d%d%d",&s,&f,&t);
	clean();                    //定义单位矩阵
	printf("%d",answer());
	return 0;
}