https://codeforces.com/contest/1433/problem/G
思路:开始看到了路线免费,以为是出了分层最短路板子,看了看发现只让一条路免费。
开始直接考虑暴力,枚举每条边,然后每个点进行dijkstra,最后取最小。复杂度O(n^2klogm);
考虑一下对每个点其实可以先预处理,提前处理好每个点的对应的最短路。O(n^2logm)
考虑免费的边的贡献。
对每一个ki来说,边(a,b)有三种情况。
1.开始不在其最短路路径上,免费后也不在其最短路路径上。
2.开始不在其最短路路径上,免费后在其最短路路径上。
3.开始在其最短路路径上,免费后在其最短路路径上。
对于第一种情况,ki的最短路长度不变,仍然是原来的dis[ki.first][ki.second] (ki.first和ki.second分别代表其起点和终点)
对于第二三种情况,ki的最短路长度可能变化,dis[ki.first][a]+dis[ki.second][b];dis[ki.first][b]+dis[ki.second][a];
那么枚举边,再枚举每个k,求出总和的最小.
总时间复杂度O(mk+n^2logm)
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e3+100;
typedef long long LL;
typedef pair<LL,LL>P;///first距离,second编号
LL dis[maxn];
LL n,m,k,start;
struct edge{
LL to,cost;
};
vector<edge>g[maxn];
LL f[maxn][maxn];
bool vis[maxn];
void dijkstra()
{
LL s=start;
memset(dis,0x3f,sizeof(dis));
memset(vis,0,sizeof(vis));
dis[s]=0;
priority_queue< P, vector<P>, greater<P> >que;
que.push({0,s});
while(!que.empty())
{
P p=que.top();que.pop();
LL v=p.second;
if(vis[v]) continue;
vis[v]=1;
for(LL i=0;i<g[v].size();i++)
{
edge e=g[v][i];
if(dis[e.to]>dis[v]+e.cost)
{
dis[e.to]=dis[v]+e.cost;
que.push({dis[e.to],e.to});
}
}
}
for(LL j=1;j<=n;j++){
f[s][j]=dis[j];
}
}
int main(void)
{
cin.tie(0);std::ios::sync_with_stdio(false);
cin>>n>>m>>k;
vector<pair<LL,LL> >Edge;
for(LL i=1;i<=m;i++)
{
LL u,v,w;cin>>u>>v>>w;
g[u].push_back({v,w});
g[v].push_back({u,w});
Edge.push_back({u,v});
}
vector<pair<LL,LL>> v;
for(LL i=1;i<=k;i++)
{
LL a,b;cin>>a>>b;
v.push_back({a,b});
}
for(LL i=1;i<=n;i++)
{
start=i;
dijkstra();
}
LL sum=1e18;
for(auto i:Edge)///遍历每条边
{
LL a=i.first;LL b=i.second;
LL ans=0;
for(auto j:v)///遍历所有人
{
ans+=min( f[j.first][j.second],min(f[j.first][a]+f[j.second][b],f[j.first][b]+f[j.second][a]) );
}
sum=min(sum,ans);
}
cout<<sum<<endl;
return 0;
}