hdu-2433 bfs（dijkstra）最短路

转自：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;
}