PAT 1018 Public Bike Management

参考链接：柳婼 の blog—1018. Public Bike Management (30)-PAT甲级真题（Dijkstra + DFS）
    题目就不贴了。
    个人感觉这道题挺有价值的，做一下学习记录。
    一开始只用的Dijkstra，读入自行车数量后将每个结点的自行车数量减去Cmax/2。Dijkstra算法过程中，使用两个数组，一个数组dis存距离，一个need数组存需要从管理中心拿出的直行车的数量（用正负区分是拿出还是拿回）。这种思路的错误在于，题目中要求是从管理中心出发到目的地点的路上能分配自行车给结点，返回时不能分配，但是使用Dijkstra算法时更新need值时则有可能不满足这个要求。除此之外这个思路无法满足题目中要求的距离都是最短的情况下选择从管理中心拿出自行车最少的路径，若拿出的自行车还相同则选拿回自行车最少的路径。
    所以本题采用的方法是使用Dijkstra算法，记录下所有的最短路径，然后从管理中心出发模拟分配自行车的过程，最后得到结果。
    本题最大的收获是最短路径的记录，采用dfs遍历结点记录路径，在dfs开头的时候将结点push_back到存路径的vector中，然后检查该路径是否已经到达终点，然后执行相应的操作，最后pop_back，然后返回。

代码：

#include <bits/stdc++.h>
using namespace std;

const int maxn = 505;
const int inf = 99999999;
int min_need = inf,min_back = inf;
int CMAX,N,Sp,M,u,v,len;
int num[maxn];
int mark[maxn];
int dis[maxn];
vector<int> pre[maxn],temppath,path;
int G[maxn][maxn];
int debug = 2;

void dijkstra()
{
	// If there are more than one shortest path,
	// the one that requires the least number of bikes sent from PBMC 
	// will be chosen.
	
	// Note that if such a path is not unique, 
	// output the one that requires minimum number of bikes 
	// that we must take back to PBMC. 
	// The judge's data guarantee that such a path is unique.
	
	for(int i = 0;i <= N; i++){
		dis[i] = inf;
		mark[i] = 0;
	}	
	
	dis[0] = 0;
	
	while(1) 
	{
		//选出当前距离0最近的点 
		int v = -1;
		for(int i = 0;i <= N; i++)
		{
			if((v == -1 && mark[i] == 0) || (v != -1 && mark[i] == 0 &&  dis[i] < dis[v]))
				v = i;
		}
		if(v == -1) 
			break;
		
		mark[v] = 1;
		
		for(int i = 0;i <= N; i++) 
		{
			if(dis[i] > dis[v] + G[v][i])
			{
				dis[i] = dis[v] + G[v][i];
				pre[i].clear();
				pre[i].push_back(v);
			}
			else if(dis[i] == dis[v] + G[v][i])
			{
				pre[i].push_back(v);
			}
		}
	}
	
	if(debug == 1)
	{
		for(int i = 0;i <= N; i++)
		{
			for(auto value: pre[i])
				cout << value << " ";
			cout << endl;
		}	
	}
}

void dfs(int v) {
    temppath.push_back(v);
    if(v == 0)   //从Sp已经到达0 整个路径的信息都已存入temppath 
    {
    	int back = 0,need = 0;
    	for(int i = temppath.size()-1;i >= 0; i--)   
		// 回去的路上不能调整，所以不能直接叠加 
    	{
    		int id = temppath[i];
    		if(num[id] < 0)    //需要自行车 
    		{
    			if(back >= -num[id])
    				back -= -num[id];
    			else if(back < -num[id])
    			{
    				need += -num[id] - back;
    				back = 0;
				}
			}
			else if(num[id] > 0)    //有多余的自行车 
			{
				back += num[id];
			}
		}
		if(min_need > need) 
		{
			min_need = need;
			min_back = back;
			path = temppath;
		}
		else if(min_need == need && min_back > back)
		{
			min_back = back;
			path = temppath;
		}
		temppath.pop_back();
		return ;
	}
	
	for(int i = 0;i < pre[v].size(); i++)
		dfs(pre[v][i]);
	
	temppath.pop_back();
}

int main(int argc, char const *argv[])
{
	cin >> CMAX >> N >> Sp >> M;
	
	for(int i = 0;i <= N; i++)
		for(int j = 0;j <= N; j++)
			G[i][j] = inf;

	for(int i = 1;i <= N; i++)
	{
		cin >> num[i];
		num[i] -= CMAX / 2;
	}
	
	for(int i = 0;i < M; i++)	
	{
		cin >> u >> v >> len;
		G[u][v] = G[v][u] = len;
	}
	
	dijkstra();
	
	dfs(Sp);
	
	cout << min_need << " ";
	for(int i = path.size()-1;i >= 0; i--)
	{
		cout << path[i];
		if(i != 0)	
			cout << "->";
	}
	cout << " " << min_back << endl;
		
	return 0;
}