Codeforces Dytechlab Cup 2022 D

Codeforces Dytechlab Cup 2022 D

题目传送门

除了题目很绕以外，第一眼应该联想到的都是最短路罢（。但是这道题其实跟贪心的性质更吻合。

设最短路上的一条最短边两端连接的是 $i$ 和 $j$ ，如果我们试着把这条边的两个端点改为 $1$ 和 $n$ 的话，可以发现，我们我们仅仅需要 $dis(1,i)+dis(j,n)$ 的次数执行操作（这里的 $dis$ 为最短边数），之后直接执行一次程序包传输就可以实现了，总的代价为 $w(i,j)\dot{(}dis(1,i)+dis(j,n)+1)$ 。由于我们是在最短路上执行操作，实际上后面的 $dis(1,i)+dis(j,n)+1$ 就是最短路径的边数，显然这样的操作的结果一定是小于等于直接走最短路的代价的（边相等的时候就等于了）。因此，我们有了第一种方案，就是直接将边连接到 $1$ 和 $n$ ，不管它是不是在最短路径上，至于为什么不管还有下面的一个操作。

第二种操作，就是将边先连接到一个点上，再借由这个点进行连接操作，这也是第三个样例的解决方案。如果这条边不在最短路径上，那么我们直接连接的时候显然会带来更多的额外操作，此时的 $dis(1,i)+dis(j,n)+1$ 显然是大于最短路径的。但是还有一种方式连接边，也就是将边的一边连接到一个最短路径的一个顶点上，另外一边连接到一个与最短路直接相连的一个点，然后通过像螃蟹一样对两边进行扩展延伸，最终连接到目的顶点。为什么这样会更少？如果要将一条边直接两端扩展的话，那么代价显然是两个点的距离和，但如果将这条边借由一个端点逼近最短路的话，那么显然代价就会少掉一部分，这样就可以实现延伸了。因此，我们只需要找到一个中转点 $k$ ，执行我们所说的操作，第一步先将边缩成一个环，在一个点上，第二部就可以随便选一个这个点的邻接点连接了，这样的操作数为 $\min (dis(i,k),dis(j,k))+1$ ，1 为自环的操作，之后以 $k$ 为基准，将边一步一步扩展，这样的操作数为 $dis(1,k)+dis(k,n)$ 。因此，这样的操作总的代价为 $w(i,j)\dot{(}\min (dis(i,k),dis(j,k))+1+dis(1,k)+dis(k,n)+1)$ 。 $k$ 要咋整？直接枚举就可以了，非最小代价的情况一定会被最小代价的情况覆盖掉，因此直接枚举就可以了。

至于为什么可以不管是不是在最短路径上，一方面这道题时间没卡那么严，另一方面直接两种方案枚举一遍，非最小的情况也会被最小的情况掩盖掉。

void Solution() {
    int n, m;
    cin >> n >> m;
    vector f(n + 1, vector<ll>(n + 1, inf<ll>));
    vector dis(n + 1, vector<int>(n + 1, inf<int>));
    for (int i = 1; i <= n; i ++) {
        f[i][i] = dis[i][i] = 0;
    }
    for (int i = 1; i <= m; i ++) {
        int a, b;
        ll w;
        cin >> a >> b >> w;
        f[a][b] = f[b][a] = min(f[a][b], w);
        dis[a][b] = dis[b][a] = 1;
    }
    for (int k = 1; k <= n; k ++)
        for (int i = 1; i <= n; i ++) 
            for (int j = 1; j <= n; j ++) 
                dis[i][j] = min(dis[i][j], dis[i][k] + dis[k][j]);
    ll ans = inf<ll>;
    for (int i = 1; i <= n; i ++) {
        for (int j = 1; j <= n; j ++) {
            if (i != j && f[i][j] != inf<ll>) {
                ans = min(ans, f[i][j] * (1 + dis[1][i] + dis[j][n]));
                for (int k = 1; k <= n; k ++) {
                    ans = min(ans, f[i][j] * (1 + min(dis[i][k], dis[k][j]) + 1 + dis[1][k] + dis[k][n]));
                }
            }
        }
    }
    cout << ans << '\n';
}