codevs1482【路线统计】

1482 路线统计

 时间限制: 1 s

 空间限制: 256000 KB

 题目等级 : 钻石 Diamond

题解

 查看运行结果

题目描述 Description

N个节点的有向图, 求从start到finish刚好经过时间time的总方案数 mod 502630.

输入描述 Input Description

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

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

       接下来一行3个整数start, finish, time.

输出描述 Output Description

输出总方案数.

样例输入 Sample Input

       3

       .12

       2.1

       12.

       0 2 5

样例输出 Sample Output

8

数据范围及提示 Data Size & Hint

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

       对于100%的数据, 1 <= n <= 10; 1 <= start,finish <= n; 1 <= time <= 10^9.

分类标签 Tags 点此展开 

很容易想到dp.

定义dp[i][j][k]表示从i到j用时间为k的方案数，dp[i][j][k]=∑dp[i][p][k-cost[p][j]]*dp[p][j][cost[p][j]]，转移是O(n)的，枚举3个变量，加起来大致是n^3*time的时间复杂度，很明显是不能接受的。

如果等式右边的第3维是1的话，左边k=2，这样不考虑第三维，这种转移是不是很眼熟，没错，就是矩阵的转移。

当所有花费为1的时候，我们直接用矩阵快速幂一下就好了，快速幂k次用的时间就为k.

题意中花费不一定为1啊？- -

但是最大也才5，我们想象把一个点"。"拆成"。->。->。->。->。"       每条边权值为1，如果权值为3，就把第3个点和j相连为1，这样就构成了时间全为1的01矩阵，然后快速幂跑一次就好了。

开始写挂了，发现数据死活不对，把数据开成long long发现就对了，奇葩- -

附上大神都懒得发的码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<cmath>
#define LL long long
using namespace std;
const LL M=502630;
const LL maxn=65;
/*
定义dp[i][j][k]为从i到j经过k秒的方案数
dp[i][j][k]=∑dp[i][p][i~p]*dp[p][j][p~j]
如果等式右边的第三维经过的都是1s，经过两秒就是等式左边。
考虑拆点构造一个矩形，将其自乘n次(i,j)就是答案 
*/
LL n;
struct Mat
{
	LL mat[maxn][maxn];
	Mat()
	{
		memset(mat,0,sizeof(mat));
	}
};
Mat operator*(Mat a,Mat b)
{
	Mat c;
	for(LL k=1;k<=60;k++)
	{
		for(LL i=1;i<=60;i++)
		{
			for(LL j=1;j<=60;j++)
			{
				c.mat[i][j]+=a.mat[i][k]%M*(b.mat[k][j]%M);
				c.mat[i][j]%=M;
			}
		}
	}
	return c;
}
Mat operator^(Mat a,LL k)
{
	Mat c;
	for(LL i=1;i<=60;i++)c.mat[i][i]=1;
	for(;k;k>>=1)
	{
		if(k&1)c=c*a;
		a=a*a;
	}
	return c;
}
char ss[65];
LL tot;
int main()
{
	Mat f,g;
	scanf("%lld",&n);
	for(LL i=1;i<=n;i++)
	{
		for(LL j=1;j<5;j++)
		{
			f.mat[i+(j-1)*n][i+j*n]=1;
		}
	}
	for(LL i=1;i<=n;i++)
	{
		scanf("%s",ss+1);
		for(LL j=1;j<=n;j++)
		{
			if(ss[j]=='.')continue;
			else
			{
				f.mat[i+(ss[j]-'0'-1)*n][j]=1;
			}
		}
	}
	LL st,ft,k;
	scanf("%lld%lld%lld",&st,&ft,&k);
	st++;
	ft++;
	g=f^k;
	printf("%lld\n",g.mat[st][ft]);
	return 0;
}