HDU1535(反向建边)

标签：

题意：求从出发点到每个车站的最短距离+从每个车站回到出发点的最短距离是多少

思路：如果从每个点到出发点求一次最短路10^5个点10^5次求SPFA也会超时，那么如果我们反向建边的话可以只用一次SPFA，因为是单向边嘛，把箭头反过来，再求一次起点到各点的最短距离不就行了？

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int inf = 0x3f3f3f3f;
const int M = 1e6+10;
struct Edge
{
    int u,v,w;
} e[M];
int v[M],w[M],next[M],head[M],d[M],edge;//head初始化为-1，模拟邻接表

void init(int V)
{
   // fill(v,v+V,0);
   // fill(w,w+V,inf);
    //fill(next,next+V,-1);
    fill(d,d+V,inf);
    fill(head,head+V,-1);
    edge = 0;
}
void addedge(int a,int b,int x)//主要如果是无向图的话,每条边要执行两次这个，a,b反过来就行
{
    v[edge]=b;
    w[edge]=x;
    next[edge]=head[a];
    head[a]=edge++;
}

void SPFA(int s)
{
    queue<int> q;
    d[s]=0;
    q.push(s);
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        for(int i=head[u]; i!=-1; i=next[i])
        {
            if(d[v[i]]>d[u]+w[i])
            {
                d[v[i]]=d[u]+w[i];
                q.push(v[i]);
            }
        }
    }
}

int main()
{
    #ifdef xxz
    freopen("in.txt","r",stdin);
    #endif // xxz
    int n,sum,p,q;
    scanf("%d",&n);
    while(n--)
    {

        sum = 0;
        scanf("%d%d",&p,&q);
        init(p+1);
        for(int i = 0; i < q; i++)
        {
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
            addedge(e[i].u,e[i].v,e[i].w);
        }

        SPFA(1);
        for(int i = 2; i <= p; i++)
        {
            sum += d[i];
        }

        init(p+1);
        for(int i = 0; i < q; i++)
        {
            addedge(e[i].v,e[i].u,e[i].w);
        }
        SPFA(1);
        for(int i = 2; i <= p; i++)
        {
            sum += d[i];
        }
        printf("%d\n",sum);


    }
    return 0;
}

HDU1535(反向建边)

标签：

原文地址：http://blog.csdn.net/u013445530/article/details/44204569