[Atcoder] AtCoder Beginner Contest 236 G - Good Vertices

题意:

给一个有向图,每条边按照顺序一条一条往里面加。问对于每个点$i$,最早在第几条边加入的时候,存在一条$1$到$i$,长度为$L$的路径。不存在输出$-1$。

$n \le 100, m \le n^2$

$L \le 10^9$

 

解答:

$L\le 10^9$比较容易想到矩阵运算。

一个比较naive的想法是:对于每条边加入后,求邻接矩阵的$L$次幂,这样就知道哪些点可以从1号点经过$L$条边到达,暴力算就行,时间复杂度为$O(n^5logL)$。

然而看着正解,发现虽然是矩阵快速幂,但是和我上面这个想法差距有点大。

我们把原问题转化为:一开始加入所有边,但是第$i$条边的边权是$i$,那么对于每个点的答案就是求从$1$号点,到$i$号点,经过$L$条边的路径中最大边权最小是多少。

我们设$dp[i][u]$表示经过$i$条边到达$u$号节点,经过路径中最大边权最小是多少。

显然可以得到以下这个转移$dp[i][v] = min_{(u,v)\in g}{max{f[i - 1][u], g[u][v]}}$

 

我们可以用矩乘优化这个操作,我们设min为+,max为×。

那么

(图来自官网)

我们只要设$dp[0][1] = 0, dp[0][i] = inf$。这样答案就是后矩乘后的第一列。

 

代码很好写,但是为什么把加法和乘法定义成min,max,从而适用矩阵,官方题解中给出了解释。

一个运算可以使用矩阵优化,必须要他的乘法满足交换幺半群,加法满足幺半群。

并且乘法对加法有左右分配率。

幺半群就是存在e使得(A + e) = A的,满足结合律的代数系统。

交换幺半群是满足交换律的幺半群。

而分配率这个很容易搞。

想要记住这个结论,可以画3个二阶矩阵乘起来的展开式,分别对12,23加括号得到的式子,看看必须满足什么性质。

 

代码:

//
// Created by onglu on 2022/3/22.
//

#include <bits/stdc++.h>

#define all(a) a.begin(),a.end()
#define rall(a) a.rbegin(),a.rend()
#define endl '\n'
#define lson (rt << 1)
#define rson (rt << 1 | 1)
#define Mid ((l + r) / 2)
//#define int long long
using namespace std;
const int N = 2e6 + 1009;
//const int N = 2e5 + 1009;
//const int N = 5009;
//const int N = 309
int n, m;
int l;
struct mat {
    int n;
    int g[109][109];
};
mat operator*(const mat &a, const mat &b) {
    mat c;
    int n = a.n;
    c.n = a.n;
    for(int i = 1; i <= n; i++) {
        memset(c.g[i], 0x3f, sizeof(int) * (c.n + 5));
    }
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            for(int k = 1; k <= n; k++) {
                c.g[i][j] = min(c.g[i][j], max(a.g[i][k], b.g[k][j]));
            }
        }
    }
    return c;
}
mat Pow(mat &a, int p) {
    mat ans;
    int f = 0;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            ans.g[i][j] = 0x3f3f3f3f;
        }
        ans.g[i][i] = 0;
    }
    for( ; p; p = p >> 1, a = a * a) {
        if(p & 1) {
            ans = a * ans;
        }
    }
    return ans;
}
void work() {
    mat a;
    cin >> n >> m >> l;
    a.n = n;
    memset(a.g, 0x3f, sizeof(a.g));

    for(int i = 1; i <= m; i++) {
        int x, y;
        cin >> x >> y;
        a.g[y][x] = min(a.g[y][x], i);
    }
    a = Pow(a, l);
    for(int i = 1; i <= n; i++) {
        if(a.g[i][1] == 0x3f3f3f3f) {
            cout << -1 << " " ;
        } else {
            cout << a.g[i][1] << " ";
        }
    }
    cout << endl;
}

signed main() {
#ifdef LOCAL
    freopen("C:\\Users\\onglu\\CLionProjects\\acm\\data.in", "r", stdin);
    freopen("C:\\Users\\onglu\\CLionProjects\\acm\\data.out", "w", stdout);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    int Case = 1;
//  cin >> Case;
    while (Case--) work();
    return 0;
}
View Code

 

posted @ 2022-03-26 01:23  _onglu  阅读(61)  评论(0编辑  收藏  举报