【HDU 2433】Travel (BFS+最短路径树)

传送门

    Travel
    题意：一张n点m条边的图，边权均为1。对于每一条边，求删除它之后所有点对之间的最短路径和。

I think

    由于边权均为1,则可用BFS对每个点i求最短路并用Sum[i]记所有点到i的最短距离和。枚举删除边j时,若该边存在于点i的最短路径树中,对i进行屏蔽边j及其反向边的BFS再次求最短路。

Code

#include<queue>#include<cstdio>#include<cstring>#define s second using namespace std;typedef pair<int,int>pii;const int sm = 10005;const int sn = 105;const int Inf = 0x3f3f3f3f;int tot,N,M,ret,ans; bool fl=0; int fm[sm],to[sm],hd[sn],nxt[sm],tag[sm];int d[sn],pre[sn][sn],Sum[sn];int T[sn],vis[sn];void Add(int u,int v) {    fm[++tot]=u,to[tot]=v,nxt[tot]=hd[u],hd[u]=tot;    fm[++tot]=v,to[tot]=u,nxt[tot]=hd[v],hd[v]=tot;}int Bfs(int x,int ind){    if(ind) memcpy(T,pre[x],sizeof(T));    int t,cnt=1,sum=0;     for(int i=1;i<=N;++i) vis[i]=0;    d[x]=0;vis[x]=1;    queue<int>Q; Q.push(x);    while(!Q.empty()) {        t=Q.front();Q.pop();        for(int i=hd[t];i;i=nxt[i])            if(!vis[to[i]]&&!tag[i]) {                pre[x][to[i]]=t;                vis[to[i]]=1; ++cnt;                d[to[i]]=d[t]+1;                Q.push(to[i]);            }        if(cnt==N) break;    }    if(cnt!=N) { fl=1; return 0; }    for(int i=1;i<=N;++i) sum+=d[i];    if(ind) memcpy(pre[x],T,sizeof(T));    return sum;}void Init(){    tot=ret=0,fl=0;    for(int i=1;i<=N;++i) hd[i]=0;}int main() {    while(scanf("%d%d",&N,&M)!=EOF) {        Init(); int t,q,u,v;        for(int i=1;i<=M;++i)             scanf("%d%d",&u,&v),Add(u,v);        for(int i=1;i<=N;++i) {            Sum[i]=Bfs(i,0);            ret+=Sum[i];            if(fl)break;         }        if(fl) {            for(int i=1;i<=M;++i) puts("INF");            continue;        }        for(int i=1;i<=M;++i) {            t=(i-1)<<1|1; fl=0; ans=ret;             for(int j=1;j<=N;++j)                if(pre[j][fm[t]]==to[t]||pre[j][to[t]]==fm[t]) {                    tag[t]=tag[t+1]=1;                    q=Bfs(j,1);                    tag[t]=tag[t+1]=0;                    if(fl) break;                    ans=ans-Sum[j]+q;                }            if(fl) puts("INF");            else printf("%d\n",ans);        }    }    return 0;}