洛谷P1828 香甜的黄油——floyed的优化

题目：https://www.luogu.org/problemnew/show/P1828

优化技巧：

因为无向图，邻接矩阵对称，所以可以将第3层for由j<p改成j<i，并将f[i][j]=f[i][k]+f[k][j]改成f[i][j]=f[j][i]=f[i][k]+f[k][j]。

for(int k=1;k<=p;k++)
		for(int i=1;i<=p;i++)
			for(int j=1;j<i;j++)
				if(i!=k && j!=k && i!=j 
				&& f[i][j]>f[i][k]+f[k][j])
					f[i][j]=f[j][i]=f[i][k]+f[k][j];

经测试可以节省近一半时间。

不做优化，洛谷将TLE两个点。

floyed优化技巧：对称邻接矩阵，可以将第三行for写成j<i。

AC代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int n,p,m;
int a[2005],f[2005][2005];
void floyed(){
	for(int k=1;k<=p;k++)
		for(int i=1;i<=p;i++)
			for(int j=1;j<i;j++)
				if(i!=k && j!=k && i!=j 
				&& f[i][j]>f[i][k]+f[k][j])
					f[i][j]=f[j][i]=f[i][k]+f[k][j];
}
int main(){
	int num,x,y,z;
	cin>>n>>p>>m;
	for(int i=1;i<=n;i++){
		cin>>num;
		a[num]++;
	}
	
	for(int i=1;i<=p;i++)
		for(int j=1;j<=p;j++)
			if(i!=j) f[i][j]=f[j][i]=1e9;
	for(int i=1;i<=m;i++){
		cin>>x>>y>>z;
		f[x][y]=f[y][x]=z;
	}
	
	floyed();
	
	int minn=1e9;
	for(int i=1;i<=p;i++){
		int sum=0;
		for(int j=1;j<=p;j++){
			if(i==j)continue;
			sum+=a[j]*f[i][j];
		}
		if(minn>sum)minn=sum;
	}
	
	printf("%d\n",minn);
	return 0;
}