转自:https://www.cnblogs.com/–ZHIYUAN/p/6238148.html
题目链接:https://vjudge.net/problem/HDU-2433
题意:有n个点,m条边,每条边的长度是1,问如果将第i(1<=i<=m)条边删去剩下的边能否将每两个点都联通,如果能输出总的最短路否则输出INF。可能会有重边。
思路:暴力是肯定超时的,这时候可以用一个巧妙的办法避免大量的时间浪费。用一个二维数组mp[a][b]保存a到b的边的条数,先跑一边bfs(其实跟dijkstra一样),记录下单源最短路,然后再尝试去掉边。这时候会有几种情况:1.去掉这条边后还有重边,因此不影响 2.最短路中没有用到这条边,也不影响 3.最短路中用到了这条边,只能在重新跑一遍dfs。
#include <bits/stdc++.h>
using namespace std;
int n,m;
int mp[105][105];//保存两点之间的边的数目
int use[105][105][105];//use[x][y][z]表示在以x为起点跑bfs时如果用到了y到z的道路 那就标记一下
int vis[105];//标记顶点是否被访问过
int dis[105];//每个点到起点的距离
int sum[105];//每个点跑bfs时的总距离
int a[3005],b[3005];//存储边的起点与终点
vector <int> v[105];
void init()
{
memset(mp,0,sizeof(mp));
memset(sum,0,sizeof(sum));
memset(use,0,sizeof(use));
for(int i=0; i<=100; i++)
{
v[i].clear();
}
}
int bfs(int x,int f)
{
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
queue <int> q;
q.push(x);
vis[x]=1;
while(!q.empty())
{
int y=q.front();
q.pop();
for(int i=0; i<v[y].size(); i++)
{
int z=v[y][i];
if(vis[z])
continue;
if(mp[y][z]<=0)//如果已经删掉了这条边使得yz不连通
continue;
dis[z]=dis[y]+1;
vis[z]=1;
q.push(z);
if(!f)//f为1时是一开始跑bfs来记录哪些道路会用到
{
use[x][y][z]=1;
use[x][z][y]=1;
}
}
}
int s=0;
for(int i=1; i<=n; i++)
{
if(i==x)
continue;
if(dis[i]==0)//有的点不连通
return -1;
s+=dis[i];
}
return s;
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
init();
for(int i=0; i<m; i++)
{
scanf("%d%d",&a[i],&b[i]);
mp[a[i]][b[i]]++;
mp[b[i]][a[i]]++;
v[a[i]].push_back(b[i]);
v[b[i]].push_back(a[i]);
}
int ans=0;
for(int i=1; i<=n; i++)
{
sum[i]=bfs(i,0);
if(sum[i]==-1)//原图不连通
{
ans=-1;
break;
}
ans+=sum[i];
}
for(int i=0; i<m; i++)
{
if(ans==-1)
{
printf("INF\n");
continue;
}
mp[a[i]][b[i]]--;//删除边
mp[b[i]][a[i]]--;
if(mp[a[i]][b[i]]>0)//这条边有重边替代 不影响
{
printf("%d\n",ans);
}
else
{
int anss=ans;
for(int j=1; j<=n; j++)
{
if(use[j][a[i]][b[i]]==0)//没用到这条边
{
continue;
}
int tem=bfs(j,1);//用到了这条边,只能跑一遍bfs
if(tem==-1)
{
anss=-1;
break;
}
anss-=sum[j];
anss+=tem;
}
if(anss==-1)
printf("INF\n");
else
printf("%d\n",anss);
}
mp[a[i]][b[i]]++;//恢复原来的图
mp[b[i]][a[i]]++;
}
}
return 0;
}