Acdream 1415 最短路+双连通

思路:先找到1到n的所有最短路径上的边,然后在这些边中间找边双连通桥即为答案。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#include<stack>
#define maxn 1<<29
using namespace std;
struct edge
{
    int from,to;
    long long len;
};
struct ed
{
    int u,v;
};
struct node
{
    int ve;
    long long dis;
};
vector<edge>edges;
stack<int>st;
vector<int>g[22222];
set<int>s[22222];
set<int>ss[22222];
int pre[22222];
bool bb[222222];
int num;
bool vis[22222];
long long dis[22222];
vector<int>ans;
bool inst[22222];
int dfn[22222],low[22222];
int n,m;
void add(int from,int to,long long len)
{
    edges.push_back((edge){from,to,len});
    g[from].push_back(edges.size()-1);
}
struct cmp
{
    bool operator()(node &x,node &y)
    {
        return x.dis>y.dis;
    }
};
void tarjan(int u)
{
    dfn[u]=low[u]=++num;
    st.push(u);
    inst[u]=1;
    //int size=s[u].size();
    set<int>::iterator it;
    for(it=ss[u].begin();it!=ss[u].end();it++)
    //for(int i=0;i<size;i++)
    {
        if(bb[*it])continue;
        bb[*it]=bb[(*it)^1]=1;
        edge e=edges[*it];
        if(!dfn[e.to])
        {
            tarjan(e.to);
            low[u]=min(low[u],low[e.to]);
            if(low[e.to]>dfn[u])
            {
                ans.push_back((*it)/2+1);
            }
        }
        else if(inst[e.to])
        {
            low[u]=min(low[u],dfn[e.to]);
        }
    }
    int v;
    if(dfn[u]==low[u])
    do
    {
        v=st.top();
        st.pop();
        inst[v]=0;
    }while(u!=v);
}
void solve()
{
    priority_queue<node,vector<node>,cmp>q;
    memset(pre,0,sizeof(pre));
    for(int i=1;i<=n;i++)dis[i]=-1;
    //for(int i=1;i<=n;i++)cout<<dis[i]<<endl;
    dis[1]=0;
    node temp;
    temp.ve=1;
    temp.dis=0;
    q.push(temp);
    while(!q.empty())
    {
        //cout<<1<<endl;
        node a=q.top();
        q.pop();
        if(vis[a.ve])continue;
        else
        {
            int u=a.ve;
            vis[u]=1;
            int size=g[u].size();
            for(int i=0;i<size;i++)
            {
                edge e=edges[g[u][i]];
                int v=e.to;
                int dd=e.len;
                if(dis[u]+dd==dis[v]&&dis[v]!=-1)
                {
                    s[v].insert(g[u][i]^1);
                }
                else if(dis[u]+dd<dis[v]||dis[v]==-1)
                {
                    s[v].clear();
                    s[v].insert(g[u][i]^1);
                    dis[v]=dis[u]+dd;
                    pre[v]=g[u][i];
                    temp.ve=v;
                    temp.dis=dis[v];
                    q.push(temp);
                }
                //cout<<dis[v]<<endl;
            }
        }
    }
    queue<int>qq;
    memset(vis,0,sizeof(vis));
    qq.push(n);
    vis[n]=1;
    while(!qq.empty())
    {
        int u=qq.front();
        qq.pop();
        set<int>::iterator it;
        for(it=s[u].begin();it!=s[u].end();it++)
        {
            edge e=edges[*it];
            ss[u].insert(*it);
            ss[e.to].insert((*it)^1);
            if(!vis[e.to])
            {
                qq.push(e.to);
                vis[e.to]=1;
            }
        }
    }
//    for(int i=1;i<=n;i++)
//    {
//        //int size=s[i].size();
//        set<int>::iterator it;
//        for(it=s[i].begin();it!=s[i].end();it++)
//        //for(int j=0;j<size;j++)
//        {
//            //cout<<(*it)<<endl;
//            edge e=edges[*it];
//            s[e.to].insert((*it)^1);
//        }
//    }
    memset(dfn,0,sizeof(dfn));
    for(int i=1;i<=n;i++)
    {
        if(!dfn[i])tarjan(i);
    }
    printf("%d\n",ans.size());
    for(int i=0;i<ans.size();i++)
    {
        if(i==0)printf("%d",ans[i]);
        else printf(" %d",ans[i]);
    }
    printf("\n");
}
int main()
{
    int u,v;
    long long l;
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%lld",&u,&v,&l);
        add(u,v,l);
        add(v,u,l);
    }
    num=0;
    solve();
    return 0;
}


你可能感兴趣的:(Baoge)