POJ1511---Invitation Cards (最短路:邻接表+2次spfa)

【题目来源】https://vjudge.net/problem/POJ-1511
【题意】
一大批志愿者去一大批的公交站点发传单,这里的一大批数量为n, (^ ,^),完了之后呢,还有回到出发点,题中描述出发点为编号1的站点,然后,问题就变成了这样的一个模型:
求从1到其余个点需要的最短路径之和,加上,其余各点到1号点的最短路径之和。
【思路】
两次spfa。但是由于站点的数量较多,可以用邻接表存图,给出邻接表博客:啊哈雷
【代码】

#include<cmath>
#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int d[1000010];
bool vis[1000010];
int first[1000000+10];
int nexx[1000000+10];
int n,m;
long long sum;
struct pp
{
    int u,v,x;
} edge[1000000+10];
struct cmp
{
    bool operator() (int &a,int &b)
    {
        return a>b;
    }
};
void init()
{
    for(int i=1; i<=n; i++)//初始化
        first[i]=nexx[i]=-1;
    for(int i=1; i<=m; i++)
    {
        scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].x);
        nexx[i]=first[edge[i].u];//重点
        first[edge[i].u]=i;
    }
}
void spfa(bool flag)
{
    for(int i=1; i<=n; i++)
    {
        d[i]=i==1?0:INF;
        vis[i]=0;
    }
    priority_queue<int,vector<int>,cmp>que;
    que.push(1);
    vis[1]=1;
    while(!que.empty())
    {
        int res=que.top();
        que.pop();
        vis[res]=0;
        for(int i=first[res]; i!=-1; i=nexx[i])//邻接表的使用
        {
            int now=(flag?edge[i].v:edge[i].u);//为了判断,用左端点还是右端点,毕竟交换之后有所变化
            if(d[now]>d[res]+edge[i].x)
            {
                d[now]=d[res]+edge[i].x;
                if(!vis[now])
                {
                    que.push(now);
                    vis[now]=1;
                }
            }
        }

    }
    for(int i=1; i<=n; i++)
       sum+=d[i];
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        init();
        sum=0;
        spfa(1);
        for(int i=1; i<=n; i++)
        {
            first[i]=nexx[i]=-1;
        }
        for(int i=1; i<=m; i++)
        {
            nexx[i]=first[edge[i].v];
            first[edge[i].v]=i;
        }
        spfa(0);
        printf("%lld\n",sum);
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值