PTA 深度优先搜索模板

编程语言：cpp
模板：

// 单源
#include <iostream>
#include <vector>
using namespace std;
int C,N,S,M;// C 最大容量, N 节点数量, S 出问题节点, M 边;
// 用edge存储边的长度
int bikes[501], edge[501][501], dis[501];
// 存储整个图
vector<int> v[501];
// 存储中间路径等变量
vector<int> path;
vector<int> final_path;
int final_take,final_send, final_dis=9999999;
void dfs(int cur, int cur_dis, int send, int take){
	// 当当前路不是最优路时一定要及时止损
    if(cur_dis>dis[cur]) return;
    // path_back和pop_back刚好完成了当前节点的进和出
    // 中间用来处理到达目的地节点和没到达目的地节点应该做的动作
    path.push_back(cur);
    // 到达目的地记录下最有路径
    if(cur==S){     
        if(cur_dis<final_dis||(cur_dis==final_dis&&send<final_send)||(cur_dis==final_dis&&send==final_send&&take<final_take)){
            final_path=path;
            final_dis=cur_dis;
            final_take=take;
            final_send=send;
        }
    // 没有到达目的地时就要继续遍历
    }else{ 
    	// 路径变短了就记录一下
        if(cur_dis<dis[cur]) dis[cur]=cur_dis;
        // 对当前节点的邻居节点进行遍历
        for(int i:v[cur]){
            if(take+bikes[i]<C/2){
                dfs(i, cur_dis+edge[cur][i], send+C/2-take-bikes[i],0);
            }else{
                dfs(i, cur_dis+edge[cur][i], send, take+bikes[i]-C/2);
            }
        }
    }
    path.pop_back();
}
int main(){
    int i,j;
    // 输入数据的处理
    cin>>C>>N>>S>>M;
    for(i=1;i<=N;++i){
        cin>>bikes[i];
    }
    for(i=0; i<=N; ++i){
        dis[i]=9999999;
    }
    for(i=0;i<M;++i){
        int a,b,time;
        cin>>a>>b>>time;
        // 记录整个图和边的权重
        v[a].push_back(b);
        v[b].push_back(a);
        edge[a][b]=time;
        edge[b][a]=time;
    }
    // 从出发点深度遍历
    dfs(0,0,0,0);
    // 根据题目格式输出
    cout<<final_send<<" "<<final_path[0];
    for(i=1;i<final_path.size();i++){
        cout<<"->"<<final_path[i];
    }
    cout<<" "<<final_take;
    return 0;
}

PTA 甲级1018