PAT 1018. Public Bike Management (30)

题目说明：有公共自行车网，从原点0出发到目标点sp，要把所有点的自行车数都置为half-full，找到最优路径，优先级依次为：路径长度最短，send最少，back最少。

算法：最短路径，dfs回溯。求出所有0到sp的路径，然后比较找出最优解。

不用dijkstra算法，因为不适用于有负权值的问题。题目要求的条件在求解过程中，路径长度递增，send递增，而back值可能变小。题目意思一不小心就理解错了，从原点到sp的过程中修正所有点，但是不会把多余的自行车再送回之前的点，比如一条路上bike数为 3 10，是send2，back5，而不是back3。

最后代码是参照网上大神的：

#include<iostream>
#include<fstream>
#include<vector>
using namespace std;
#define MAX 505
#define INF 10000

int cap,N,sp,M;
int vex;
int dist[MAX][MAX];
int bike[MAX];
#define PF cap/2

vector<int> curpath;
vector<int> shortpath;
int minsend=INF,minback=INF;
int minlen=INF;
int cursend=0,curback=0;
int curlen=0;
bool visit[MAX]={0};

void dfs(int cur){
	if(curlen>minlen) 
		return;
	if(cur==sp){
	//到达目标点，看是否最优 
		if(curlen<minlen){
			minlen=curlen;
			minsend=cursend;
			minback=curback;
			shortpath=curpath;
		}
		else if(curlen==minlen){
			if(cursend<minsend||(cursend==minsend&&curback<minback)){
				minsend=cursend;
				minback=curback;	
				shortpath=curpath;
			}	
		}
		return;
	}
	for(int i=1;i<vex;i++){
		if(visit[i]==true||dist[cur][i]==INF)
			continue;
		int lastsend=cursend;
		int lastback=curback;
		//计算到达当前点的send和back数 
		if(bike[i]+curback<PF){
			cursend+=PF-bike[i]-curback;
			curback=0;
		}
		else{
			curback=bike[i]+curback-PF;
		}
		visit[i]=true;
		curpath.push_back(i);
		curlen+=dist[cur][i];
		dfs(i);
		curpath.pop_back();
		visit[i]=false;
		curlen-=dist[cur][i];
		cursend=lastsend;
		curback=lastback;
	}
}
int main(){
	cin>>cap>>N>>sp>>M;
	//初始化,距离置为INF 
	vex=N+1;
	for(int i=0;i<vex;i++){
		for(int j=0;j<vex;j++){
			dist[i][j]=dist[j][i]=INF;
		}
	}

	for(int i=1;i<vex;i++){
		cin>>bike[i];
	}
	for(int k=0;k<M;k++){
		int i,j;
		cin>>i>>j;
		cin>>dist[i][j];
		dist[j][i]=dist[i][j];
	}
	
	dfs(0);
	
	printf("%d 0",minsend);	
	for(int i=0;i<shortpath.size();i++){
		printf("->%d",shortpath[i]);
	}
	printf(" %d",minback);
	return 0;
}