HDU 3631 Shortest Path Floyd

来源:互联网 发布:java sdk是什么 编辑:程序博客网 时间:2024/04/30 00:07

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=3631

题意

给定一含n个结点的有向带权图,编号0~n-1,每个点都有一个标记,初始时每个点都没被标记。然后给q个询问,每个询问的格式有两种。
1. 0 x:代表标记x结点。
2. 1 x y 代表求x到y的最短路,最短路只能走被标记的点。

思路

主要是求最短路,其他的注意细节。每个点被标记,那么就用这个点去松弛原来的图,用floyd算法思想。

#include<cstdio>#include<queue>#include<iostream>#include<vector>#include<map>#include<cstring>#include<string>#include<set>#include<stack>#include<algorithm>#define cle(a) memset(a,0,sizeof(a))#define inf(a) memset(a,0x3f,sizeof(a))#define ll long long#define Rep(i,a,n) for(int i=a;i<=n;i++)using namespace std;#define INF2 9223372036854775807llconst int INF = ( 2e9 ) + 2;const ll maxn = 310;const int maxm = 1e5+10;int mp[maxn][maxn],mark[maxn];int n,m,q;void floyd(int k){    for(int i=0; i<n; i++)    {        if(mp[i][k]==INF)continue;        for(int j=0; j<n; j++)            if(i!=k&&k!=j&&mp[k][j]!=INF&&mp[i][j]>mp[i][k]+mp[k][j])                mp[i][j]=mp[i][k]+mp[k][j];    }}int main(){    int cas=0;    while(scanf("%d%d%d",&n,&m,&q)&&(n+m+q))    {        memset(mark,0,sizeof(mark));        for(int i=0; i<n; i++)        {            mp[i][i]=0;            for(int j=i+1; j<n; j++)                mp[i][j]=mp[j][i]=INF;        }        for(int i=0; i<m; i++)        {            int u,v,w;            scanf("%d%d%d",&u,&v,&w);            mp[u][v]=min(mp[u][v],w);        }        if(cas!=0)puts("");        printf("Case %d:\n",++cas);        for(int T=0; T<q; T++)        {            int op,u,v;            scanf("%d",&op);            if(op==0)            {                scanf("%d",&u);                if(mark[u])                {                    printf("ERROR! At point %d\n",u);                    continue;                }                mark[u]=1;                floyd(u);            }            else            {                scanf("%d%d",&u,&v);                if(!mark[u]||!mark[v])                {                    printf("ERROR! At path %d to %d\n",u,v);                    continue;                }                if(mp[u][v]!=INF)                    printf("%d\n",mp[u][v]);                else                    printf("No such path\n");            }        }    }}
原创粉丝点击