PAT A1018

最新推荐文章于 2020-02-24 14:10:42 发布

Focus9

最新推荐文章于 2020-02-24 14:10:42 发布

阅读量622

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT 甲级文章标签： PAT A1018

本文链接：https://blog.youkuaiyun.com/weixin_37565706/article/details/78985552

PAT 甲级专栏收录该内容

130 篇文章

订阅专栏

本文介绍了一种使用Dijkstra算法寻找最短路径的方法，并通过深度优先搜索确定最佳路径来最小化单车调配成本的问题解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

方法：应用dijkstra算法算出单源路径，并用pre（vector容器）记录其路径，然后利用深度遍历，计算不同路径的需要带出的单车数和带回的单车数，找到最优解并输出。

注意点：1，三个关键字优先级：路径长短，带出的单车数，带回的单车数；2，path深度遍历时，记得添加pop_back。

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=520;
const int INF=10000000;
int cmax,N,sp,M;
int station[maxn];
struct edge{
	int v,time;
};
vector<edge> ver[maxn];
vector<int> pre[maxn];

bool vis[maxn]={0};
int dis[maxn];
void Dijkstra(){
	fill(dis,dis+maxn,INF);
	dis[0]=0;
	for(int i=0;i<maxn;i++){
		int min=INF,minv=-1;
		for(int j=0;j<maxn;j++){
			if(dis[j]<min&&!vis[j]){
				min=dis[j];
				minv=j;
			}
		}
		if(minv==sp||minv==-1) break;
		vis[minv]=true;
		int len=ver[minv].size();
		for(int j=0;j<len;j++){
			int tempv=ver[minv][j].v;
			if(vis[tempv]==false){
				if(dis[tempv]>dis[minv]+ver[minv][j].time){
					dis[tempv]=dis[minv]+ver[minv][j].time;
					pre[tempv].clear();
					pre[tempv].push_back(minv);
				}else if(dis[tempv]==dis[minv]+ver[minv][j].time){
					pre[tempv].push_back(minv);
				}
			}
		}//minv
	}//所有节点 
}
vector<int> temppath,fpath;
int frequire=INF,freturn=INF;
void DFS(int root){
	if(root==0){
		int lent=temppath.size(),tsum=0,trequire=0;
		for(int i=lent-1;i>=0;i--){
			if(station[temppath[i]]>=cmax/2){
				tsum+=station[temppath[i]]-cmax/2;
			}else{
				if(tsum<cmax/2-station[temppath[i]]){
					trequire+=(cmax/2-station[temppath[i]]-tsum);
					tsum=0;
				}else{
					tsum-=(cmax/2-station[temppath[i]]);
				}
			}
		}
		if(trequire<frequire){
			frequire=trequire;
			fpath=temppath;
			freturn=tsum;
		}else if(trequire==frequire&&tsum<freturn){
			fpath=temppath;
			freturn=tsum;
		}
		return;    //此句别忘了写 
	}
	temppath.push_back(root);
	int len=pre[root].size();
	for(int j=0;j<len;j++){
		DFS(pre[root][j]);
	}
	temppath.pop_back();   //此句别忘了写 
}
int main(){
	scanf("%d%d%d%d",&cmax,&N,&sp,&M);
	for(int i=1;i<=N;i++){
		scanf("%d",&station[i]);
	}
	for(int i=0;i<M;i++){
		int v1,v2;
		edge temp;
		scanf("%d%d%d",&v1,&v2,&temp.time);
		temp.v=v2;
		ver[v1].push_back(temp);
		temp.v=v1;
		ver[v2].push_back(temp);
	}
	Dijkstra();
	DFS(sp);
	printf("%d 0",frequire);
	int lenf=fpath.size();
	for(int i=lenf-1;i>=0;i--){
		printf("->%d",fpath[i]);
	}
	printf(" %d",freturn);
	return 0;
}