[POJ3613]Cow Relays && Floyd倍增

最新推荐文章于 2020-08-27 21:20:14 发布

原创最新推荐文章于 2020-08-27 21:20:14 发布 · 517 阅读

0 ·

CC 4.0 BY-SA版权

矩阵专栏收录该内容

10 篇文章

订阅专栏

本文详细介绍了如何通过阅读论文《矩阵乘法-俞华程》来理解和实现矩阵乘法和Floyd倍增算法，并提供了一个实际的C++代码示例。文章着重于解决在实现过程中遇到的常见问题，如离散化处理和单位矩阵的正确建立，以确保算法的正确性和效率。

怎么说呢如果有看论文（矩阵乘法-俞华程）这道题还是相当异常非常的简单（除了本人自己玩自己忘记离散化直接函数爆栈调N久甚至求助吴老师之外）

唯一需要注意的就是单位矩阵的建立是对角线为0 其他全部为正无穷

Floyd倍增的原因论文上有说其实我也没怎么看

废话就说到这里上代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
#define MAXN 100
#define MAXM 100
#define INF 0x3f3f3f3f
typedef int LL;
struct Matrix{  
    int n, m;  
    LL A[MAXN+10][MAXN+10];  
}E, Z, A;
int Map[MAXN+10][MAXN+10];
int n, m, ST, ED, K, Fish;
int ID[MAXN*10+10], idcnt;
int N;
void Get_E()
{
	memset(E.A, 0x3f, sizeof(E.A));
	for(int i = 0; i < MAXN; i++) 
        E.A[i][i] = 0;
}
void New_mul(Matrix AA, Matrix BB, Matrix &Ans)
{
	Matrix D;
    memset(Ans.A, 0x3f, sizeof(D.A));
    Ans.n = D.n = AA.n;
    Ans.m = D.m = BB.m;
	for(int k = 0; k < MAXN; k++)  
        for(int i = 0; i < MAXN; i++)  
            for(int j = 0; j < MAXN; j++)  
                if(Ans.A[i][j] > AA.A[i][k] + BB.A[k][j])  
                    Ans.A[i][j] = AA.A[i][k] + BB.A[k][j];
}
void pow_mod(Matrix A, int k, Matrix &Ans)
{
    Matrix ans = E, t = A, tmp;
    Ans.n = ans.n = A.n;
    Ans.m = ans.m = A.m;
    while(k)
    {
        if(k & 1)
		{
			New_mul(ans, t, tmp);
			ans = tmp;
		}
        New_mul(t, t, tmp);
		t = tmp;
        k >>= 1;
    }
    memcpy(Ans.A, ans.A, sizeof(ans.A));
}
Matrix Ans;
int main()
{
	ios::sync_with_stdio(false);
	Get_E();
	memset(Map, 0x3f, sizeof(Map));
	cin >> n >> m >> ST >> ED;
	ST--; ED--;
	memset(ID, -1, sizeof(ID));
	for(int i = 0; i < m; i++)
	{
		int u, v, d;
		cin >> d >> u >> v;
		u--;
		v--;
		if(~ID[u] == 0) ID[u] = idcnt++;
		if(~ID[v] == 0) ID[v] = idcnt++;
		Map[ID[u]][ID[v]] = Map[ID[v]][ID[u]] = d;
	}
	E.n = E.m = Z.n = Z.m = A.n = A.m = MAXN;
	memcpy(A.A, Map, sizeof(Map));
	pow_mod(A, n, Ans);
	cout << Ans.A[ID[ST]][ID[ED]] << '\n';
}