PAT (Advanced Level)1018 Public Bike Management (30 分)

题目链接https://pintia.cn/problem-sets/994805342720868352/problems/994805489282433024
这道题真的拖了好久好久啊，今天终于写完了。感觉dij求最短路越来越熟练了，但是dfs还是有点勉强。看了柳神的代码，终于完全理解了。然后凭着自己的理解敲了一遍，一发过！开心！希望每天进步都能那么一点点！冲冲冲！

若有错误，还望大神指正！

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e3+10;
const int INF=0x3f3f3f3f;
int k,n,t,m;//每个点的最大容量、站点数量、问题站点、路径数量
int c[maxn];//每个站点的自行车数量
int vis[maxn];//标记站点是否已被访问
int dis[maxn];//从起点（PMBC）到某点的最短距离
int e[maxn][maxn];//存图
int minneed=INF,minback=INF;//需从PBMC带走的最小自行车数量、最终带回PBMC的最小自行车数量
vector<int>pre[maxn];//记录最短路中，某点的前驱站点.如pre[3]={1,2};pre[1]=0;pre[2]=0;表示从起点到站点3有两条最短路，分别为0->1->3和0->2->3;
vector<int>path,temppath;
void dfs(int v){
	temppath.push_back(v);//获取最短路
	if(v==0){//当搜索到v==0即起点(PMBC)时，temppath中已“得到”了一条最短路
		int need=0,back=0;
		for(int i=temppath.size()-1;i>=0;i--){//遍历最短路上的站点
			int id=temppath[i];
			if(c[id]>0){//若站点自行车数量超限
				back+=c[id];//需带回多余部分
			}
			else{//站点自行车数量不足
				if(back>(0-c[id])){//准备带回的自行车足以弥补
					back+=c[id];//从要带回的自行车中扣除
				}
				else{//准备带回的自行车不足以弥补
					need+=(0-c[id]-back);//扣除要带回的自行车后，还需从PBMC再拿
					back=0;//此时没有可以带回的自行车了
				}
			}
		}
		if(need<minneed){//找最小值
			minneed=need;
			minback=back;
			path=temppath;
		}
		else if(need==minneed&&back<minback){//若need相同，则选back较小的
			minback=back;
			path=temppath;
		}
	}
	for(int i=0;i<pre[v].size();i++){
		dfs(pre[v][i]);
	}
	temppath.pop_back(); //清除最后一个元素，寻找下一条最短路
}
//dij单源最短路模板
void dij(){
	memset(vis,0,sizeof(vis));
	memset(dis,INF,sizeof(dis));
	for(int i=0;i<=n;i++){
		dis[i]=e[0][i];
	}
	for(int i=1;i<=n;i++){
		int mmin=INF,u=-1;
		for(int j=0;j<=n;j++){
			if(vis[j]==0&&dis[j]<mmin){
				mmin=dis[j];
				u=j;
			}
		}
		if(u==-1) break;
		vis[u]=1;
		for(int j=0;j<=n;j++){
			if(vis[j]==0&&dis[u]+e[u][j]<dis[j]){
			//更新最短路,以及站点的前驱结点
				dis[j]=dis[u]+e[u][j];
				pre[j].clear(); //经u到j的是最短路，之前记录的东西就全部清空（因为不是最短的）
				pre[j].push_back(u); 
			}
			else if(vis[j]==0&&dis[u]+e[u][j]==dis[j]){
			//多条最短路的情况
				pre[j].push_back(u);
			}
		}
	}
}
int main()
{
	cin>>k>>n>>t>>m;
	for(int i=1;i<=n;i++){
		cin>>c[i];
		c[i]=c[i]-k/2; //若c[i]>0,说明此站点的自行车数量大于最大容量的一半，需要拿走c[i]辆自行车，否则需要补充0-c[i]辆自行车
	}
	for(int i=0;i<=n;i++){
		for(int j=0;j<=n;j++){//初始化
			if(i==j) e[i][j]=0;
			else e[i][j]=INF;
		}
	}
	while(m--){
		int u,v,w;
		cin>>u>>v>>w;
		if(w<e[u][v]){
			e[u][v]=w;
			e[v][u]=w;
		}
	}
	dij();//寻找最短路
	dfs(t);//从问题站点开始搜索
	cout<<minneed<<" 0";
	for(int i=path.size()-2;i>=0;i--){
		cout<<"->"<<path[i];
	}
	cout<<" "<<minback<<endl;
	
	
	
}