紧急救援 (25 分)

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

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

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

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

输入样例:
4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2
输出样例:
2 60
0 1 3

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3fffffff
int n,m,s,d;
int v[510],e[510][510],l[510][510];
int sums,path[510],w[510][510],p=0,counts;
void init()
{
    cin>>n>>m>>s>>d;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(i==j) l[i][j]=e[i][j]=0;
            else l[i][j]=e[i][j]=inf;
        }
    }
    for(int i=0;i<n;i++)
    {
        cin>>v[i];
        w[i][i]=v[i];
    }
    int r,c,wl;
    for(int i=1;i<=m;i++)
    {
        cin>>r>>c>>wl;
        l[r][c]=l[c][r]=e[c][r]=e[r][c]=wl;
        w[r][c]=v[r]+v[c];
    }
}
void spa()
{
    for(int k=0;k<n;k++)
    {
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(e[i][k]<inf&&e[k][j]<inf)
                {
                    if(e[i][j]>e[i][k]+e[k][j])
                    {
                        e[i][j]=min(e[i][j],e[i][k]+e[k][j]);
                        w[i][j]=w[i][k]+w[k][j]-v[k];
                    }
                    else if(e[i][j]==e[i][k]+e[k][j])
                    {
                        w[i][j]=max(w[i][j],w[i][k]+w[k][j]-v[k]);
                    }
                }
            }
        }
    }
}
void func()
{
    int s0=s;
    int ss;
    int k=0;
    for(int i=0;i<n;i++)
    {
        if(l[s0][i]<inf&&e[s0][d]==e[s0][i]+e[i][d]&&s0!=i)
        {
            counts++;
            if(w[s0][d]==w[s0][i]+w[i][d]-v[i])
            {
                p++;
                path[p]=i;
                ss=i;
            }
        }
        
    }
             for(int i=0;i<n;i++)
    {
        if(l[ss][i]<inf&&e[ss][d]==e[ss][i]+e[i][d]&&ss!=i)
        {
            if(w[ss][d]==w[ss][i]+w[i][d]-v[i])
            {
                p++;
                ss=i;
                path[p]=i;
                i=0;
                if(ss==d) break;
            }
        }
        
    }

}
int main()
{
    init();
    spa();
    path[0]=s;
    func();
    cout<<counts<<" "<<w[s][d]<<endl;
    for(int i=0;i<p;i++)
    {
        cout<<path[i]<<" ";
    }
    cout<<d<<endl;
    return 0;
}

想直接利用Floyd,失败了，记录一下，明天看看SPFA和dijstra看看能不能搞⛏。。。。。，出来了再更新一下

#include<bits/stdc++.h>
using namespace std;
#define maxn 510
#define maxm 250010
const int inf = 999999999;

typedef struct node
{
    int u;
    int ws;
    int ww;
    bool operator<(const node &a) const
    {
        if(a.ws>ws) return true;
        else if(ws==a.ws)
        {
            return a.ww>ww;
        }
        else 
            return false;
    }
}node;

typedef struct
{
    int frist;
    int w;
}vnode;

typedef struct
{
    int index;
    int w;
    int next;
}enode;

vnode v[maxn];enode e[maxm];
int n,m,s,d,di,ans[maxn],path[maxn],p=0,answ[maxn];
bool vis[maxn];int counts = 0;
bool visa[maxn];
bool vvs[maxn];
int f[maxn];    bool vsv[maxn];

void create()
{
    cin>>n>>m>>s>>d;
    s++;d++;
    for(int i=1;i<=n;i++)
    {
        cin>>v[i].w;
        ans[i] = inf ;
    }
    int x,y,z;
    int j;
    for(int i=1;i<=m;i++)
    {
        cin>>x>>y>>z;
        x++;y++;
        e[i].w=z;
        e[i].index=y;
        e[i].next=v[x].frist;
        v[x].frist=i;
        j=i+m;
        swap(x,y);
        e[j].w=z;
        e[j].index=y;
        e[j].next=v[x].frist;
        v[x].frist=j;
    }
}

void dijkstra()
{
    priority_queue<node> Q;
    Q.push(node{s,0,v[s].w});
    ans[s]=0;
    answ[s]=v[s].w;
    while(!Q.empty())
    {
        node fr = Q.top();
        Q.pop();
        int ul = fr.u;
        if(vis[ul])
        {
            //if(ul == d) counts++;
            continue;
        } 
        vis[ul]=true;
        for(int i=v[ul].frist;i!=0;i=e[i].next)
        {
            //if(e[i].index==d) counts++;
            if(ans[e[i].index]>ans[ul]+e[i].w)
            {
                ans[e[i].index]=ans[ul]+e[i].w;
                answ[e[i].index]=answ[ul]+v[e[i].index].w;
                if(!vis[e[i].index])
                Q.push(node{e[i].index,ans[e[i].index],answ[e[i].index]});
            }
            else if(ans[e[i].index]==ans[ul]+e[i].w)
            {
                if(answ[e[i].index]<answ[ul]+v[e[i].index].w)
                {
                    answ[e[i].index]=answ[ul]+v[e[i].index].w;
                    if(!vis[e[i].index])
                    Q.push(node{e[i].index,ans[e[i].index],answ[e[i].index]});
                }
            }
        }
    }
}

void inputsg()
{
    f[s]=1;
    queue<int> qq;
    qq.push(s);
    vvs[s]=true;
    while(!qq.empty())
    {
        int fs=qq.front();
        qq.pop();
        visa[fs]=true;
        for(int i=v[fs].frist;i!=0;i=e[i].next)
        {
            if(visa[e[i].index]||e[i].index==d) {continue;}
            f[e[i].index]+=f[fs];
            if(!vvs[e[i].index])
            {
                vvs[e[i].index]=true;
                qq.push(e[i].index);
            }
        }
    }
    for(int i=v[d].frist;i!=0;i=e[i].next)
        {
            f[d]+=f[e[i].index];
        }
        path[p] = d;
    int d0=d;
    vsv[d0]=true;
    for(int i=v[d0].frist;i!=0;)
    {
        bool fl=false;
            //cout<<e[i].index<<' '<<i;
            //if(vsv[e[i].index]) continue;
            if(ans[d0]==(ans[e[i].index]+e[i].w)&&(answ[d0]==(answ[e[i].index]+v[d0].w)))
            {
                //cout<<e[i].index<<' '<<s;
                path[++p] = e[i].index;
                if(e[i].index==s) break;
                d0 = e[i].index;
                i=v[d0].frist;
                fl=true;
            }
            if(!fl) i=e[i].next;
        
    }
    printf("%d ",f[d]);
    printf("%d\n",answ[d]);
    for(int i=p;i>=1;i--)
    {
        printf("%d ",path[i]-1);
    }
    printf("%d",path[0]-1);
    //printf("\n%d",ans[d]);
}

int main()
{
    create();
    dijkstra();
    inputsg();
    return 0;
}

吐了，为啥总是过不了例2和例四。。。。。。