POJ- 1511 Invitation Cards

 

题目看懂了，然而没注意看数组，肿么辣么大~~天真的用了flody，超超超，，，

 

就是有n个车站，m条路线，每条路线有一个价格。

要求从起始站到每一站，然后再从每一站回去，求最少的钱数。

 

看起来就是最短路径的问题，就是求单源点到其他各点的最但距离（票价），然后建个反向图，求的返回的钱数。也就是1到个点的价格加上各点返回1的价格。

 

最短路径很多方法，Dijkstra算法，Bell-ford算法，floyd-washall，以及本题目用的spfa算法。

由于数据比较大，Dijkstra算法 （o(n^2)）   floyd-washall(o(n^3))肯定会超时。

 

其实spfa比Dijkstra算法还要简单，代码更少，就是维护一个队列，队列里面放着可能经过的点，知道队列为空。

~~~然而目前我还不是太会用

 

注意要用邻接表保存图，邻接矩阵会超内存的。

 

天真代码：超超超~~~

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

#define N 1550000
#define INF 0x3f3f3f3f

int P,maps[2500][2500];

void Init()
{
    for(int i=1;i<=P;i++)
        for(int j=1;j<=P;j++)
        maps[i][j]=(i==j)?0:INF;
}

void flody()
{
    for(int k=1;k<=P;k++)
        for(int i=1;i<=P;i++)
            for(int j=1;j<=P;j++)
            maps[i][j]=min(maps[i][j],maps[i][k]+maps[k][j]);
}

int main()
{
    int T,Q,i,a,b,price;
    scanf("%d", &T);

    while(T--)
    {
        scanf("%d %d", &P,&Q);
        Init();

        for(i=1;i<=Q;i++)
        {
            scanf("%d %d %d", &a,&b,&price);
            maps[a][b]=price;
        }

        flody();

        int sum=0;
        for(i=1;i<=P;i++)
            sum+=maps[1][i]+maps[i][1];

        printf("%d\n", sum);
    }
    return 0;
}

 

AC代码：

 

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

#define N 1550000
#define INF 0x3f3f3f3f

long long dist[N];
int vis[N],head[N];
int k;

struct node
{
    int u,v,nb;
    long long price;
}edge[N];

struct point
{
   int a,b;
   long long price;
}d[N];

void Add(int a,int b,long long price,int k)
{
    edge[k].u=a;
    edge[k].v=b;
    edge[k].price=price;
    edge[k].nb=head[a];
    head[a]=k;
}

long long spfa(int n)
{
    queue<int>Q;
    Q.push(1);
    vis[1]=1;

    for(int i=1;i<=n;i++)
        dist[i]=INF;
    dist[1]=0;

    while(Q.size())
    {
        int x=Q.front();
        Q.pop();
        vis[x]=0;

        for(int j=head[x];j;j=edge[j].nb)
        {
            int u=edge[j].u;
            int v=edge[j].v;
            long long  price=edge[j].price;

            if(dist[u]+price<dist[v])
            {
                dist[v]=dist[u]+price;
                if(vis[v]==0)
                {
                    Q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
    long long  sum=0;

    for(int i=1;i<=n;i++)
        sum+=dist[i];

    return sum;
}

int main()
{
    int T,q,i,P;
    scanf("%d", &T);

    while(T--)
    {
        scanf("%d %d", &P,&q);
        memset(head,0,sizeof(head));

        for(i=1;i<=q;i++)
        {
            scanf("%d %d %lld", &d[i].a,&d[i].b,&d[i].price);
            Add(d[i].a,d[i].b,d[i].price,i);
        }

     long long sum;

        sum=spfa(P);

        memset(head,0,sizeof(head));
        for(i=1;i<=q;i++)
            Add(d[i].b,d[i].a,d[i].price,i);

        sum+=spfa(P);

        printf("%lld\n", sum);
    }
    return 0;
}