CCF行车路线

CCF 行车路线

题目链接
注意事项

1. dev C++声明 文件才能使用memset函数
2. 最短路算法一定要先初始化好相关的数组变量，将相应的值初始化为最大值
3. 题目最终所求的结果一般都很大，最好使用 long long 类型。
4. 一定要先测试一遍程序再提交

最开始这道题我想用两张图同时用Floyd算法过一遍，最后发现这种想法是错误的
下面的这个方法采用了类似spfa算法的思想，

dis[i]表示从起点到达i点的路径的最后一段路程是大路。
dis1[i]表示从起点到达i点的路径的最后一段路程是小路。

#include<vector>
#include<iostream>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
#define DEBUG
#define MAX 550
#define INF 0x3f3f3f3f
typedef long long LL;
int M, N;
vector<LL>dis(MAX), dis1(MAX);
LL G[MAX][MAX], G1[MAX][MAX];
vector<bool>visit(MAX, false);
//计算所有以小路为连接的两点之间的距离
void Floyd()
{
	for (int k = 1; k <= N; k++)
		for (int i = 1; i <= N; i++)
			for (int j = 1; j <= N; j++)
			{
				if (i == j)
				{
					G1[i][j] == 0;
					continue;
				}
				if (i == k || j == k)continue;
				if (G1[i][j] > G1[i][k] + G1[k][j])
				{
					G1[i][j] = G1[i][k] + G1[k][j];
				}

			}
}


//dis1[i]第i条路为小路
//dis[i] 第i条路为大路
void Dij(int s)
{

	fill(dis.begin(), dis.end(), INF);
	fill(dis1.begin(), dis1.end(), INF);
	dis[s] = 0;
	dis1[s] = 0;
	queue<int>que;
	que.push(s);
	visit[s] = 1;
	while (!que.empty())
	{
		int top = que.front();
		que.pop();
		visit[top] = false;
		for (int i = 1; i <= N; i++)
		{
			//都是大路
			if (dis[i] > dis[top] + G[top][i])
			{
				dis[i] = dis[top] + G[top][i];
				if (visit[i] == false)
				{
					que.push(i);
					visit[i] = true;
				}
			}
			//上一条是小路，这一条是大路
			if (dis[i] > dis1[top] + G[top][i])
			{
				dis[i] = dis1[top] + G[top][i];
				if (visit[i] == false)
				{
					que.push(i);
					visit[i] = true;
				}

			}
			if (G1[top][i] < INF)//说明这一条路是小路
			{ //注意这里并不存在前一段为小路，这一段也为小路的情况。
			//只要点与点之间的连通全是小路，直接一口气算完
			//所以在这里计算的时候，满足前一段为大路，后一段为小路的情况。就能直接计算两个点之间全是小路的情况
				if (dis1[i] > dis[top] + G1[top][i] * G1[top][i])
				{
					dis1[i] = dis[top] + G1[top][i] * G1[top][i];
					if (visit[i] == false)
					{
						que.push(i);
						visit[i] = true;
					}
				}
			}
		}
	}
}


int main()
{
#ifdef DEBUG
	freopen("sb.txt", "r", stdin);
#endif // DEBUG
	cin >> N >> M;

	memset(G, INF, sizeof(G));
	memset(G1, INF, sizeof(G1));

	for (int i = 0; i < M; i++)
	{
		int type, from, to;
		LL cap;
		scanf("%d %d %d %lld", &type, &from, &to, &cap);
		if (type == 1)
		{
			if (cap < G1[from][to])
				G1[from][to] = G1[to][from] = cap;
		}
		else
		{
			if (cap < G[from][to])
				G[from][to] = G[to][from] = cap;
		}
	}
	Floyd();
	Dij(1);
	cout << min(dis[N], dis1[N]);
//以大路结尾和以小路结尾哪一个距离更小
	return 0;
}