PAT 1030. Travel Plan (30)

37 篇文章 0 订阅
#include<iostream>
#include<vector>
#include<cstring>
#define MAX 501
#define MAXWEIGHT 250005
using namespace std;
int n, m, s, d;
int adj_dis[MAX][MAX] = { 0 };
int adj_cost[MAX][MAX] = { 0 };
int min_len = MAXWEIGHT, min_cost = MAXWEIGHT;
int vst[MAX] = { 0 };
vector<int> path, final_path;
void dfs(int v)
{
	if (v == d)
	{
		int len = 0, cost = 0;
		for (int i = 1; i < path.size(); i++)
		{
			len += adj_dis[path[i]][path[i - 1]];
			cost += adj_cost[path[i]][path[i - 1]];
		}
		if (len < min_len)
		{
			min_len = len; min_cost = cost;
			final_path.resize(path.size());
			memcpy(&final_path[0], &path[0], path.size()*sizeof(int));
		}
		else if (len == min_len&&min_cost > cost)
		{
			min_cost = cost;
			final_path.resize(path.size());
			memcpy(&final_path[0], &path[0], path.size()*sizeof(int));
		}
		vst[v] = 0;
		path.pop_back(); 
		return;
	}
	for (int i = 0; i < n; i++)
	{
		if (adj_dis[v][i] != 0 && !vst[i])
		{
			path.push_back(i);
			vst[i] = 1;
			dfs(i);
		}
	}
	vst[v] = 0;
	path.pop_back();
	return;
}
int main()
{
	cin >> n >> m >> s >> d;
	for (int i = 0; i < m; i++)
	{
		int a, b, dis, cost;
		cin >> a >> b >> dis >> cost;
		adj_dis[a][b] = adj_dis[b][a] = dis;
		adj_cost[a][b] = adj_cost[b][a] = cost;
	}
	path.push_back(s); vst[s] = 1;
	dfs(s);
	cout << final_path[0];
	for (int i = 1; i < final_path.size(); i++)
	{
		cout << " " << final_path[i];
	}
	cout << " " << min_len << " " << min_cost;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值