九度OJ：1343 城际公路网（特定情况下的floyd优化）

题目1343：城际公路网

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：504

解决：197

题目描述：

为了加快城市之间的通行和物资流动速度，A国政府决定在其境内的N个大中型城市之间，增加修建K条公路。已知这N个城市中的任意两个都能相互连通，且已知其最短的路径长度。为了时刻监测修建新路对A国城市的影响，特任命你为观察员，负责在每修建完一条公路之后，就向该国领导汇报当前N个城市间的最短路之和。

输入：

测试数据包括多个，每个测试数据包含四个部分

首先第一行将输入整数N，其中2 <= N <= 300，代表城市的个数。

第二部分是是一个N * N大小的整数矩阵，代表A国城市之间的一个最短路长度情况。矩阵的第i行，第j列代表从城镇i到城镇j的最短路d _ij的长度。输入数据保证当i == j时，d _ij = 0；同时还保证d _ij = d _ji。

第三部分是一个整数K，1 <= K <= 300， 即计划修建的公路数量。

第四部分包括K行，每行由三个整数a、b、w组成，整数之间由空格隔开。其中a、b代表城镇的编号（1 <= a、b <= N）, w为该条公路的长度（1 <= w<=1000）。注意：所建公路都为双向公路。

输出：

对于每个测试案例，输出K行，即修建了公路之后的N个城市的最短路径之和。例如，第i行则代表修建完1、2、3 …、i条公路之后的城市最短路径之和。

样例输入：

2
0 7
7 0
1
1 2 3
3
0 4 5
4 0 9
5 9 0
2
2 3 8
1 2 1

样例输出：

3
17
12

题目大意：之前n个城市已经互通，现在又要在其间修路，每修一条路，计算出当前的最短的那条路径的长度。

解题思路：Floyd算法，三重for循环显然会超时，我们尽量把它降到二维的。当你新建设一条路时，设这条路起点为from，终点为to，长度为d，输入这条路之后，双重for循环扫描下i->from->to->j这个新的距离是否小于原先的i->j的距离，小于的话就更新。（那个d就相当于from到to的距离了）

代码如下：

#include <cstdio>
int map[310][310];
int main()
{
	int n;
	while(scanf("%d",&n)!=EOF)
	{
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				scanf("%d",&map[i][j]);
			}
		}
		int k;
		scanf("%d",&k);
		for(int i=0;i<k;i++)
		{
			int from,to,d;
			scanf("%d%d%d",&from,&to,&d);//这里d就是map[from][to] 
			for(int w=1;w<=n;w++)
			{
				for(int m=1;m<=n;m++)
				{
					if(map[w][m]>map[w][from]+d+map[to][m])//原图中w到m的距离大于w到from到to到m，则更新map 
					{
						map[w][m]=map[w][from]+d+map[to][m];
						map[m][w]=map[w][m];//双向更新 
					}
				}
			}
			int sum=0;
			for(int j=1;j<=n;j++)
			{
				for(int b=1;b<=n;b++)
				{
					sum=sum+map[j][b];
				}
			}
			sum=sum/2;//双向图，最后除以2 
			printf("%d\n",sum);
		}
	}
	return 0;
}