[PTA]L2-001 紧急救援 （25 分)

L2-001 紧急救援 （25 分)

Description

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

Input

输入第一行给出4个正整数N、M、S、D，其中N（2）是城市的个数，顺便假设城市的编号为0 ~ (；M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。

第二行给出N个正整数，其中第i个数是第i个城市的救援队的数目，数字间以空格分隔。随后的M行中，每行给出一条快速道路的信息，分别是：城市1、城市2、快速道路的长度，中间用空格分开，数字均为整数且不超过500。输入保证救援可行且最优解唯一。

output

第一行输出最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔，输出结尾不能有多余空格。

Examples

Input

4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

Output

2 60
0 1 3

 

正确解法：

跟最短路计数一样。用的是dijkstra。

一个计数，如果新路径，就相等。相同的最短路就相加。

一个最多数目。如果新路经就相加。相同的就更新最大值。并且更新前缀点。

求路径就是记录一个前缀点。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<map>
#include<set>
#include<vector>
#include<queue>
#include<algorithm>
//#include<cmath>
using namespace std;
typedef long long ll;
const int inf=0x7fffffff;
const int N=1001;
const int M=9999999;
const ll mod=1000000000+7;
int n,m,s,d,len=0;
int Link[N],dis[N],con[N],num[N],val[N],pre[N],bok[N];
int ans[N],cnt=0;
struct node
{
    int y,v,next;
}e[M];
void insert(int xx,int yy,int vv)
{
    e[++len].next=Link[xx];
    Link[xx]=len;
    e[len].y=yy;
    e[len].v=vv;
}
void init()
{
    scanf("%d %d %d %d",&n,&m,&s,&d);
    for(int i=0;i<n;i++)
    {    scanf("%d",&val[i]);
        dis[i]=inf;
        pre[i]=-1;
        num[i]=val[i];
    }
    for(int i=1;i<=m;i++)
    {
        int xx,yy,vv;
        scanf("%d %d %d",&xx,&yy,&vv);
        insert(xx,yy,vv);
        insert(yy,xx,vv);
    }
    dis[s]=0;
    bok[s]=1;
    con[s]=1;
    for(int i=Link[s];i;i=e[i].next)
    {    dis[e[i].y]=e[i].v;
        con[e[i].y]=1;
    }
    for(int i=1;i<n;i++)
    {
        int minn=inf,u=s;
        for(int j=0;j<n;j++)
            if(bok[j]==0&&dis[j]<minn)
            {
                minn=dis[j];
                u=j;
            }
        bok[u]=1;
       // cout<<u<<endl;
        for(int j=Link[u];j;j=e[j].next)
        {
            if(dis[e[j].y]>dis[u]+e[j].v)
            {
                dis[e[j].y]=dis[u]+e[j].v;
                con[e[j].y]=con[u];
                num[e[j].y]=num[u]+val[e[j].y];
                pre[e[j].y]=u;
            }
            else if(dis[e[j].y]==dis[u]+e[j].v)
            {
                con[e[j].y]+=con[u];
                if(num[e[j].y]<num[u]+val[e[j].y])
                {
                    num[e[j].y]=num[u]+val[e[j].y];
                    pre[e[j].y]=u;
                }
            }
        }
        //cout<<num[3]<<endl;

    }

}
int main()
{
    init();
    cout<<con[d]<<" "<<num[d]+num[s]<<endl;
    for(int i=d;i!=-1;i=pre[i])
        ans[++cnt]=i;
    printf("%d",s);
    for(int i=cnt;i>0;i--)
        printf(" %d",ans[i]);
    printf("\n");

    return 0;
}

 

等会有spfa写一下QAQ

错了一个数据的spfa（）

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<map>
#include<set>
#include<vector>
#include<queue>
#include<algorithm>
//#include<cmath>
using namespace std;
typedef long long ll;
const int inf=0x7fffffff;
const int N=1001;
const int M=9999999;
const ll mod=1000000000+7;
int n,m,s,d,len=0;
int Link[N],dis[N],con[N],num[N],val[N],pre[N],bok[N];
int que[M+100];
int ans[N],cnt=0;
struct node
{
    int y,v,next;
}e[M];
void insert(int xx,int yy,int vv)
{
    e[++len].next=Link[xx];
    Link[xx]=len;
    e[len].y=yy;
    e[len].v=vv;
}
void init()
{
    scanf("%d %d %d %d",&n,&m,&s,&d);
    for(int i=0;i<n;i++)
    {    scanf("%d",&val[i]);
        dis[i]=inf;
        pre[i]=-1;
        num[i]=val[i];
    }
    for(int i=1;i<=m;i++)
    {
        int xx,yy,vv;
        scanf("%d %d %d",&xx,&yy,&vv);
        insert(xx,yy,vv);
        insert(yy,xx,vv);
    }
    dis[s]=0;
    bok[s]=1;
    con[s]=1;
    for(int i=Link[s];i;i=e[i].next)
    {
        //dis[e[i].y]=e[i].v;
        con[e[i].y]=1;
    }
    int head=1,tail=2;
    que[1]=s;
    while(head<tail)
    {
        int tt=que[head];
        //cout<<tt<<" "<<num[3]<<endl;
        bok[tt]=0;
        for(int i=Link[tt];i;i=e[i].next)
        {
            if(dis[e[i].y]>dis[tt]+e[i].v)
            {
                dis[e[i].y]=dis[tt]+e[i].v;
                con[e[i].y]=con[tt];
                num[e[i].y]=num[tt]+val[e[i].y];
                pre[e[i].y]=tt;
                if(bok[e[i].y]==0)
                {
                    que[tail++]=e[i].y;
                    bok[e[i].y]=1;
                }
            }
            else if(dis[e[i].y]==dis[tt]+e[i].v)
            {
                con[e[i].y]+=con[tt];
                if(num[e[i].y]<num[tt]+val[e[i].y])
                {
                    num[e[i].y]=num[tt]+val[e[i].y];
                    pre[e[i].y]=tt;
                }
            }
        }
        head++;
    }
}
int main()
{
    init();
    cout<<con[d]<<" "<<num[d]<<endl;
    for(int i=d;i!=-1;i=pre[i])
        ans[++cnt]=i;
    //printf("%d",s);
    for(int i=cnt;i>1;i--)
        printf("%d ",ans[i]);
    printf("%d\n",ans[1]);

    return 0;
}