移动式除虫 SPFA

移动式除虫 SPFA

描述：
XXX 是除虫大队的队员。 今天， 他从家出发到除虫大队总部大楼参会。他从不放弃能够 利用的时间。所以，他决定在路上顺便进行除虫工作。他所在的 Bugs 镇上的人十分崇拜苹 果，所以他们以苹果树作为“地标” 。镇上共有 N 片苹果林，第 i 片中有虫wi吨。
有 M 条无向街道连接这些苹果林，第 i 条街道的长度为Li ，并且每片苹果林都可以通过道路到达其 他所有苹果林。xxx 的家在 1 号苹果林，而除虫大队总部大楼在 N 号苹果林。xx 自 然不能在路上花费太多的时间， 所以他只会走最短的路。 假设他到达一片苹果林就能瞬间除 掉其中的所有虫。也就是说，xx想选择一条在最短的前提下能除掉最多的虫的路径。 现在你需要求出 xxx的最短路长度以及在走最短路的前提下最多能除多少虫。

输入：
第一行两个整数 M N, 。 第二行 N 个整数，为wi
接下来 M 行，每行三个整数
Ai,Bi,Ci描述一条连接ai,bi,且长度为Li的道路。

输出：
输出包含两行
第一行一个整数表示最短路长度。
第二行一个整数表示在走最短路前提下最多除虫数。

样例输入：
5 5
1 2 3 4 5
1 2 1
1 3 1
2 4 1
3 4 1
4 5 1

样例输出
3
13

解题思路：
本题近似SPFA模板，松弛边的时候，d[y]>=d[x]+len的情况都要进行考虑，当d[y]=d[x]+len的时候松弛可以消灭的虫子数，即val[y]=max(val[y],val[x]+w[y]);当d[y]>d[x]+len的时候松弛距离，同时更改消灭的虫子数。

代码如下

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;
#define N 100010
struct Node{//边
    int x,y,len;//x起点，y终点，len长度（权值）
    Node(int x,int y,int len):x(x),y(y),len(len){}
};
int n,m;
vector<Node> s[N];
int inqueue[N];//判断是否在队列中
int d[N],w[N],val[N];//d距离，w每个地点虫子数，val总虫子数
void SPFA(int start){
    queue<int> q;
    d[start]=0;
    val[start]=w[start];
    q.push(start);
    while(!q.empty()){
        int o=q.front();q.pop();
        inqueue[o]=0;
        int so=s[o].size();
        for(int i=0;i<so;i++){
            Node e=s[o][i];
            int x=e.x,y=e.y,len=e.len;
            if(d[y]>=d[x]+len){//松弛
                if(d[y]==d[x]+len)//判断距离松弛虫子数量
                    val[y]=max(val[y],val[x]+w[y]);
                else{d[y]=d[x]+len;val[y]=val[x]+w[y];}
                if(!inqueue[y]){inqueue[y]=true;q.push(y);}
            }
        }
    }
}
int main(){
    memset(inqueue,0,sizeof(inqueue));
    memset(d,127,sizeof(d));
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>w[i];
    for(int i=1;i<=m;i++){
        int x,y,len;cin>>x>>y>>len;
        s[x].push_back(Node(x,y,len));//双向建边
        s[y].push_back(Node(y,x,len));
    }
    SPFA(1);
    cout<<d[n]<<endl<<val[n]<<endl;
    return 0;
}