题目描述 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;
}