BZOJ4773: 負環(倍增Floyd)

自為風月馬前卒發表於2018-10-27

題意

題目連結

Sol

倍增Floyd，妙妙喵

一個很顯然的思路(然而我想不到是用(f[k][i][j])表示從(i)號點出發，走(k)步到(j)的最小值

但是這樣複雜度是(O(n^4))的

考慮倍增優化，設(f[k][i][j])表示從(i)號點出發，走(2^k)步到(j)的最小值

每次轉移相當於把兩個矩陣乘起來，複雜度(O(n^3logn))

注意答案不一定有單調性，可以對每個點連一條向自己邊權為(0)的邊，這樣就滿足單調性了

感覺最近抄寫程式碼很有手感啊qwq

#include<bits/stdc++.h>
#define chmin(a, b) (a = a < b ? a : b)
using namespace std;
const int MAXN = 301;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < `0` || c > `9`) {if(c == `-`) f = -1; c = getchar();}
    while(c >= `0` && c <= `9`) x = x * 10 + c - `0`, c = getchar();
    return x * f;
}
int N, M, base;
struct Ma {
    int m[MAXN][MAXN];
    Ma() {
        memset(m, 0x3f, sizeof(m));
    }
    Ma operator * (const Ma &rhs) const {
        Ma ans;
        for(int k = 1; k <= N; k++)
            for(int i = 1; i <= N; i++)
                for(int j = 1; j <= N; j++) 
                    chmin(ans.m[i][j], m[i][k] + rhs.m[k][j]);
        return ans;
    }
}f[31], now, nxt;
int main() {
    N = read(); M = read();
    for(int i = 1; i <= M; i++) {
        int x = read(), y = read(), w = read();
        f[0].m[x][y] = w;
    }
    for(int i = 1; i <= N; i++) f[0].m[i][i] = now.m[i][i] = 0;
    for(int i = 1; (1ll << i) <= N; i++) f[i] = f[i - 1] * f[i - 1], base = i;
    int ans = 0;
    for(int i = base; i >= 0; i--) {
        bool flag = 0;
        nxt = f[i] * now;
        for(int j = 1; j <= N; j++) if(nxt.m[j][j] < 0) {flag = 1; break;}
        if(!flag) ans += 1 << i, now = nxt;
    }
    printf("%d", ans + 1 > N ? 0 : ans + 1);
    return 0;
}

相關文章