CF400D最短路

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxm = 100009;
const int maxn = 600;
const int inf = 1<<29;
struct node
{
	int u,v,w;
}eg[maxm];
int fa[maxm];
int dis[maxn][maxn];
int Count[maxn]; 
int hash[maxn];
int sum[maxn];
int n,m,k;
int pre[maxm];

void init()
{
	int i,j;
	for(i=0; i<= n; i++) fa[i] = -1;
	for(i =0;i<maxn;i++)
	{
		for(j=0;j<maxn;j++)
			dis[i][j] = inf;
		dis[i][i] = 0;	
	}
}

int find(int x)
{
	int p;
	for(p =x;fa[p] >= 0;p  = fa[p]);
	while(x != p)
	{
		int temp = fa[x];
		fa[x] = p;
		x = temp;
	}
	return p;
}
void Union(int a,int b)
{
	int x = find(a);
	int y = find(b);
	if(x == y) return ;
	int temp = fa[x] + fa[y];
	if(fa[x] > fa[y])
		fa[x] = y, fa[y] = temp;
	else
		fa[y] = x, fa[x] = temp;
}

void floyd()
{
	for(int z =0; z < k; z++)
	{
		for(int i= 0; i < k; i++)
		{
			for(int j = 0;j < k; j++)
			{
				if(dis[i][j] > dis[i][z] + dis[z][j])
					dis[i][j] = dis[i][z] + dis[z][j];
			}
		}
	}
}
inline int min(int a,int b){return a < b ? a : b;
}
int main()
{
	int i;
	while(~scanf("%d%d%d",&n,&m,&k))
	{
		init();	
		sum[0] = 0;
		for(i=1;i<=k;i++)
		{
			scanf("%d",&Count[i]);
			sum[i] =sum[i-1] + Count[i];
		}
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&eg[i].u,&eg[i].v,&eg[i].w);
			if(eg[i].w == 0)
				Union(eg[i].u,eg[i].v);
			int u=lower_bound(sum+1,sum+k+1,eg[i].u)-sum;
			u--;
            int v=lower_bound(sum+1,sum+k+1,eg[i].v)-sum;
			v--;
			if(u == v)continue;		
			dis[u][v] = dis[v][u] = min(dis[u][v],eg[i].w);
		}

	    memset(pre,0,sizeof(pre));
        int flag=1;
        for(i=1;i<=n;i++)
        {
            int posu=lower_bound(sum+1,sum+k+1,i)-sum;
            int u=find(i);
            if(pre[posu])
            {
                if(u!=pre[posu])
                {
                    flag=0;
                    break ;
                }
            }
            else
            {
                pre[posu]=u;
            }
        }
        if(flag) printf("Yes\n");
        else
        {
            printf("No\n");
            continue ;
        }
		floyd();

		for(i=0;i<k;i++)
		{
			for(int j=0; j < k;j++)
			{
				if(dis[i][j] >= inf)dis[i][j] = -1;
				printf("%d ",dis[i][j]);
			}
			printf("\n");
		}
	}
	return 0;
}