7-35 城市间紧急救援

参考：PTA 7-35 城市间紧急救援（Dijkstra+DFS） - 南风sa - 博客园 (cnblogs.com)

先 dijkstra找 最短路； 统计每个节点的前驱点（们）；

再（dfs） 在路径中找出符合第二条件的 最优路线。

#include<iostream>
#include<vector>
#include<queue> 
#include<cstring>
using namespace std;
const int N=555;
int n,m,s,t;
struct bian{
	int qian,nxt,d;
	bool operator<(const bian &n1)const{
		return d>n1.d; 
	}
};
int dis[N],vist[N];
int b[N];
vector<int>q[N]; //前面的结点 
vector<bian>e[N];
priority_queue<bian>pq;
int ansb,ansct,tpb;
vector<int>anstp;
vector<int>ans;
void dijkstra(){
	memset(dis,0x3f3f3f3f,sizeof(dis));
	vist[s]=1;
	for(auto it:e[s]){
		pq.push({s,it.nxt,it.d});
	}
	while(pq.size()){
		bian x=pq.top(); pq.pop();
		if(vist[x.nxt]){
			if(dis[x.nxt]==x.d) {
		//	  printf("******%d\n",x.nxt);
			  q[x.nxt].push_back(x.qian);	
			}
		}
		else{
		  vist[x.nxt]=1; q[x.nxt].push_back(x.qian);
		  dis[x.nxt]=x.d; 
		  for(auto it:e[x.nxt]){
		  	if(vist[it.nxt]) continue;
		  	pq.push({x.nxt,it.nxt,dis[x.nxt]+it.d});
		  }	
		}
	}
}
void f(int v){
    if(v==s){
        ansct++;
        if(ansb<tpb){
            ansb=tpb;
            ans=anstp;
        }
    }
    for(int i=0;i<q[v].size();i++){
        int u=q[v][i];
        anstp.push_back(u);
        tpb+=b[u];
        f(u);
        tpb-=b[u];
        anstp.pop_back();
    }
}
int main(){
	scanf("%d%d%d%d",&n,&m,&s,&t);
	for(int i=0;i<n;i++)
	   scanf("%d",&b[i]);
	for(int i=0;i<m;i++){
		int x,y,d; scanf("%d%d%d",&x,&y,&d);
		e[x].push_back({x,y,d});
		e[y].push_back({y,x,d});
	}
    dijkstra();
  //  printf("$$$$%d",q[3].size());
    f(t);
    ansb+=b[t];
    printf("%d %d\n",ansct,ansb);
    for(int i=ans.size()-1;~i;i--)
         printf("%d ",ans[i]);
    printf("%d",t);
}