1. 程式人生 > >Cow Relays(倍增floyd)

Cow Relays(倍增floyd)

Cow Relays

題目大意

s t 經過 n

條邊的最短路。
點數小於等於100,n≤1000000














解:

本來很容易想到拆點,但是這裡的 n 太大了,拆點做不了。但是注意到點數很少,考慮floyd。
其實進一步看floyd可以當做一個矩陣,每次取min運算。稍微想一下是有結合律的。不如直接上矩陣快速冪。
重要的一點就是我們轉移的時候不和原來的答案取min,強制做那麼多步即可。矩乘做 k 次就代表走了 k + 1 步。
複雜度就是 O ( T 3 l o g n )

code:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int cnt;
struct lxy{
    long long a[105][105];
    lxy operator * (const lxy &QAQ){
        lxy c;
        memset(c.a,0x3f,sizeof(c.a));
        for(int i=1;i<=cnt;i++)
          for(int j=1;j<=cnt;j++)
            for(int k=1;k<=cnt;k++)
              c.a[i][j]=min(c.a[i][j],a[i][k]+QAQ.a[k][j]);
        return c;
    }
}f;

int mmp[1005];
int n,m,s,t,a1,a2,a3;
void coldcold()
{
    lxy T=f;
    while(n!=0){
        if((n&1)==1) T=T*f;
        f=f*f;
        n=n>>1;
    }
    printf("%lld",T.a[mmp[s]][mmp[t]]);
}

int main()
{
    memset(f.a,0x3f,sizeof(f.a));
    scanf("%d%d%d%d",&n,&m,&s,&t);
    n--;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&a3,&a1,&a2);
        if(mmp[a1]==0) mmp[a1]=++cnt;
        if(mmp[a2]==0) mmp[a2]=++cnt;
        if(a3<f.a[mmp[a1]][mmp[a2]]){
          f.a[mmp[a1]][mmp[a2]]=a3,f.a[mmp[a2]][mmp[a1]]=a3;
        }
    }
    coldcold();
}