ACdream 1415 最短路+桥

点击打开链接

题意：给个图，问你从1到n的最短路的路径上，有多少桥

思路：先是要满足条件最短路，然后判断每条边是不是最短路里的边，怎么判断也很简单，先从1开始求最短路和从n开始求最短路，对于边U到V来说，若1到U的最短路加上n到V的最短路在加上这条边的权值若等于1到n的最短路，那么这条边就是我们要的，就是这个条件if(dis1[U[i]]+COST[i]+dis2[V[i]]==maxdis||dis1[V[i]]+COST[i]+dis2[U[i]]==maxdis)maxdis是1到n的最短路长度，然后题目说了可能有重边，好处理，之前做题时队友跟我讲题意说没有重边，WA了好久，然后将边的id和个数找到输出就行了，只是模版的应用而已，不难的~~~

#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f3f3f3f3fll;
const int maxn=20010;
struct edge{
    int to,cost;
    edge(){}
    edge(int a,int b){to=a;cost=b;}
};
typedef pair<int,long long>P;
vector<edge>G[maxn];
ll dis1[maxn],dis2[maxn];
void dijkstra(int st,ll dis[]){
    priority_queue<P,vector<P>,greater<P> >que;
    fill(dis,dis+maxn,inf);
    dis[st]=0;
    que.push(P(0,st));
    while(!que.empty()){
        P p=que.top();que.pop();
        int v=p.second;
        if(dis[v]<p.first) continue;
        for(unsigned int i=0;i<G[v].size();i++){
            edge e=G[v][i];
            if(dis[e.to]>dis[v]+(ll)e.cost){
                dis[e.to]=dis[v]+(ll)e.cost;
                que.push(P(dis[e.to],e.to));
            }
        }
    }
}
int U[100010],V[100010],COST[100010];
struct edge1{
    int to,id,num;
    edge1(int a,int b,int c){to=a;id=b;num=c;}
};
vector<edge1>GG[maxn];
int L[maxn],E[maxn],ans[maxn],vis[maxn],stack1[maxn];
int k,kk,cnt,n,m;
void dfs(int x,int fa){
    vis[x]=1;L[x]=k;E[x]=k++;stack1[kk++]=x;
    int flag=0;
    for(unsigned int i=0;i<GG[x].size();i++){
        edge1 e=GG[x][i];
        if(e.to!=fa){
            if(!vis[e.to]){
                dfs(e.to,x);
                L[x]=min(L[x],L[e.to]);
                if(L[e.to]>E[x]) GG[x][i].num=1;
            }else L[x]=min(L[x],E[e.to]);
        }else{
            if(flag) L[x]=min(L[x],E[e.to]);
            flag++;
        }
    }
    if(L[x]==E[x]){
        while(stack1[kk]!=x&&kk>0){
            L[stack1[kk-1]]=L[x];
            kk--;
            vis[stack1[kk]]=0;
        }
    }
}
void tarjan(){
    kk=0;k=1;dfs(1,1);
    cnt=0;
    for(int i=1;i<=n;i++){
        for(unsigned int j=0;j<GG[i].size();j++){
            edge1 e=GG[i][j];
            if(L[i]!=L[e.to]&&e.num==1){
                ans[cnt++]=e.id;
            }
        }
    }
}
int main(){
    while(scanf("%d%d",&n,&m)!=-1){
        for(int i=0;i<maxn;i++) G[i].clear();
        for(int i=0;i<maxn;i++) GG[i].clear();
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&U[i],&V[i],&COST[i]);
            G[U[i]].push_back(edge(V[i],COST[i]));
            G[V[i]].push_back(edge(U[i],COST[i]));
        }
        dijkstra(1,dis1);
        dijkstra(n,dis2);
        ll maxdis=dis1[n];
        for(int i=0;i<m;i++){
            if(dis1[U[i]]+COST[i]+dis2[V[i]]==maxdis||dis1[V[i]]+COST[i]+dis2[U[i]]==maxdis){
                GG[U[i]].push_back(edge1(V[i],i+1,0));
                GG[V[i]].push_back(edge1(U[i],i+1,0));
            }
        }
        tarjan();
        printf("%d\n",cnt);
        sort(ans,ans+cnt);
        for(int i=0;i<cnt-1;i++) printf("%d ",ans[i]);
        printf("%d\n",ans[cnt-1]);
    }
    return 0;
}