Por Costel and the Algorithm 【Gym - 100923B】【最短路记录边】

题目链接

题目大意

有一个人写了一个Bellman-Ford算法来编写最短路，但是这个算法的效率实在是太慢了，他想让你给图的边排序，使得Bellman-Ford算法里的while循环最多跑两次。

解题思路

其实就是先跑最短路上的边，更新就按顺序就只需要一次，所以跑一遍最短路，记录一下边就好，（邻接表从1开始）。

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;

const int N=1e5+5;
struct node
{
    int v,nex;
    long long w;
};
node e[N*4];
int n,m,tot,f[N],book[N],in[N*2],pre[N*2],ans;
long long dis[N];
long long inf=0x3f3f3f3f3f3f3f3f;
struct Node
{
    int v;
    long long c;
    Node() {}
    Node(int V,long long C)
    {
        v=V;
        c=C;
    }
    bool operator <(const Node &r)const
    {
        return c>r.c;
    }
};

void add(int u,int v,long long w)
{
    e[tot].v=v;
    e[tot].w=w;
    e[tot].nex=f[u];
    f[u]=tot++;
}
void dij()
{
    for(int i=1; i<=n; i++)
        dis[i]=inf;
    memset(book,0,sizeof(book));
    priority_queue<Node>que;
    while(!que.empty())
        que.pop();
    dis[1]=0;
    que.push(Node(1,0));
    Node tmp;
    while(!que.empty())
    {
        tmp=que.top();
        que.pop();
        int u=tmp.v;
        if(book[u])
            continue;
        for(int i=f[u]; i!=-1; i=e[i].nex)
        {
            int v=e[i].v;
            if(!book[v]&&dis[v]>dis[u]+e[i].w)
            {
                dis[v]=dis[u]+e[i].w;
                que.push(Node(v,dis[v]));
                in[i]=1;
                in[pre[v]]=0;
                pre[v]=i;
            }
        }
    }
}
void dfs(int u)
{
    for(int i=f[u]; i!=-1; i=e[i].nex)
    {
        int v=e[i].v;
        if(in[i])
        {
            ans++;
            printf("%d%c",i,ans==m?'\n':' ');
            dfs(v);
        }
    }
}
int main()
{
    freopen("algoritm.in","r",stdin);
    freopen("algoritm.out","w",stdout);

    int T;
    scanf("%d",&T);
    while(T--)
    {
        int x,y;
        long long z;
        tot=1;
        ans=0;
        memset(f,-1,sizeof(f));
        memset(in,0,sizeof(in));
        memset(pre,0,sizeof(pre));
        scanf("%d %d",&n,&m);
        for(int i=1; i<=m; i++)
        {
            scanf("%d %d %lld",&x,&y,&z);
            add(x,y,z);
        }
        dij();
        dfs(1);
        for(int i=1; i<=m; i++)
        {
            if(in[i]==0)
            {
                ans++;
                printf("%d%c",i,ans==m?'\n':' ');
            }
        }
    }
    return 0;
}