【题解】codeforces25C Roads in Berland

There are n cities numbered from 1 to n in Berland. Some of them are connected by two-way roads. Each road has its own length — an integer number from 1 to 1000. It is known that from each city it is possible to get to any other city by existing roads. Also for each pair of cities it is known the shortest distance between them. Berland Government plans to build k new roads. For each of the planned road it is known its length, and what cities it will connect. To control the correctness of the construction of new roads, after the opening of another road Berland government wants to check the sum of the shortest distances between all pairs of cities. Help them — for a given matrix of shortest distances on the old roads and plans of all new roads, find out how the sum of the shortest distances between all pairs of cities changes after construction of each road.

Input
The first line contains integer n (2 ≤ n ≤ 300) — amount of cities in Berland. Then there follow n lines with n integer numbers each — the matrix of shortest distances. j-th integer in the i-th row — di, j, the shortest distance between cities i and j. It is guaranteed that di, i = 0, di, j = dj, i, and a given matrix is a matrix of shortest distances for some set of two-way roads with integer lengths from 1 to 1000, such that from each city it is possible to get to any other city using these roads.

Next line contains integer k (1 ≤ k ≤ 300) — amount of planned roads. Following k lines contain the description of the planned roads. Each road is described by three space-separated integers ai, bi, ci (1 ≤ ai, bi ≤ n, ai ≠ bi, 1 ≤ ci ≤ 1000) — ai and bi — pair of cities, which the road connects, ci — the length of the road. It can be several roads between a pair of cities, but no road connects the city with itself.

Output
Output k space-separated integers qi (1 ≤ i ≤ k). qi should be equal to the sum of shortest distances between all pairs of cities after the construction of roads with indexes from 1 to i. Roads are numbered from 1 in the input order. Each pair of cities should be taken into account in the sum exactly once, i. e. we count unordered pairs.

Examples
input
2
0 5
5 0
1
1 2 3
output
3
input
3
0 4 5
4 0 9
5 9 0
2
2 3 8
1 2 1
output
17 12

【题解】
本题要求的是任意两点之间的最短路,且n ≤ 300,想到用floyd
不过不能O(n^3k)死暴力,会Tle的

所以我们要加个小优化,因为每次只更新一条边,所以每次floyd也只更新与更新的边有关的边。即,若a到b的边为输入的,那么以a和b为中转站去更新其他点,时间复杂度O(n^2k)

Code:

#include <bits/stdc++.h>
#define res register int
#define ll long long
#define maxn 1010
using namespace std;
ll dis[maxn][maxn];
int n,m;

inline int read(){
    int s=0,w=1;
    char c=getchar();
    while (c<'0' || c>'9'){if (c=='-')w=-1;c=getchar();}
    while (c>='0' && c<='9') s=s*10+c-'0',c=getchar();
    return s*w;
}
int main(){
    n=read();
    for (res i=1;i<=n;++i)
        for (res j=1;j<=n;++j) dis[i][j]=read();
    m=read();
    while (m--){
        int x=read(),y=read(),z=read();
        if (z<dis[x][y]){
            dis[x][y]=dis[y][x]=z;
            for (res i=1;i<=n;++i)
                for (res j=1;j<=n;++j)
                    if (i!=j){
                        dis[i][j]=min(dis[i][j],min(dis[i][x]+dis[x][j],dis[i][y]+dis[y][j]));
                        dis[j][i]=dis[i][j];
                    }
        }
        ll ans=0;
        for (res i=1;i<=n;++i)
            for (res j=1;j<i;++j) ans+=dis[i][j];
        printf("%lld ",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/ModestCoder_/article/details/81434616
今日推荐