题目链接
http://acm.hdu.edu.cn/showproblem.php?pid=2807
题意
给定n个m*m的矩阵,对于A矩阵,C矩阵,如果存在B矩阵满足A*B = C,那么图中A->B有条权值为1的边。求最短路
思路
蛮简单的一题居然给我写T了。建图直接按照题意描述建,跑Floyd获取最短路。
#include<cstdio>
#include<queue>
#include<iostream>
#include<vector>
#include<map>
#include<cstring>
#include<string>
#include<set>
#include<stack>
#include<algorithm>
#define cle(a) memset(a,0,sizeof(a))
#define inf(a) memset(a,0x3f,sizeof(a))
#define ll long long
#define Rep(i,a,n) for(int i=a;i<=n;i++)
using namespace std;
#define INF2 9223372036854775807ll
const int INF = ( 2e9 ) + 2;
const ll maxn = 90;
int mp[maxn][maxn][maxn];
int c[maxn][maxn],res[maxn][maxn];
int n,m;
void mul(int b[maxn][maxn],int c[maxn][maxn])
{
int t;
memset(res,0,sizeof(res));
for(int i=1;i<=m;i++)
for(int j=1;j<=m;j++)
{
for(int k=1;k<=m;k++)
res[i][j]+=b[i][k]*c[k][j];
}
}
bool same(int a[maxn][maxn])
{
for(int i=1;i<=m;i++)
for(int j=1;j<=m;j++)
if(res[i][j]!=a[i][j])return false;
return true;
}
int main()
{
while(~scanf("%d%d",&n,&m)&&(n+m))
{
memset(c,0,sizeof(c));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
for(int k=1;k<=m;k++)
scanf("%d",&mp[i][j][k]);
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
if(i==j)continue;
mul(mp[i],mp[j]);
for(int k=1;k<=n;k++)
{
if(k!=i&&k!=j&&same(mp[k]))
c[i][k]=1;
}
}
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
{
if(c[i][k]==0)continue;
for(int j=1;j<=n;j++)
if(c[k][j]!=0)
{
if(c[i][j]==0)
c[i][j]=c[i][k]+c[k][j];
if(c[i][j]>c[i][k]+c[k][j])
c[i][j]=c[i][k]+c[k][j];
}
}
int q;
scanf("%d",&q);
for(int T=0;T<q;T++)
{
int u,v;
scanf("%d%d",&u,&v);
if(c[u][v])
printf("%d\n",c[u][v]);
else
printf("Sorry\n");
}
}
}