hdu1546Idiomatic Phrases Game(floyd+map)

传送门

成语接龙，找每个单词都需要一点时间，问最少的时间

把字符串用map处理成数字编号，之后用floyd

#include<bits/stdc++.h>
using namespace std;
map<string,int>g;
int road[1005][1005];
int num=0;
const int inf=0x3f3f3f3f;
void floyd()
{
    for(int k=0;k<num;k++)
    {
        for(int i=0;i<num;i++)
        {
            for(int j=0;j<num;j++)
            {
                road[i][j]=road[i][j]>road[i][k]+road[k][j]?road[i][k]+road[k][j]:road[i][j];
            //    cout<<i<<" "<<j<<" "<<road[i][j]<<endl;        
            }
        }
    }
}
void init()
{
    for(int i=0;i<1005;i++)
    {
        for(int j=0;j<1005;j++)
        {
            road[i][j]=inf;
        }
        road[i][i]=0;
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n),n)
    {
        
        num=0;g.clear();init();
        int start=0,endd=0;
        for(int i=0;i<n;i++)
        {
            int t;char s[1000];
            scanf("%d %s",&t,s);
            char pre[5]={''},end[5]={''};
            for(int j=0;j<4;j++)
            {
                pre[j]=s[j];
                
            }
            //printf("%s ",pre);
            int temp=0;
            for(int j=strlen(s)-4;j<strlen(s);j++)
            {
                end[temp++]=s[j];
            }
        //    printf("%s
",end);
            if(!g[pre])g[pre]=num++;
            if(!g[end])g[end]=num++;
        //    printf("%d %d
",g[pre],g[end]);
            if(i==0)start=g[pre];
            if(i==n-1)endd=g[pre];
            if(t<road[g[pre]][g[end]])road[g[pre]][g[end]]=t;
        }
        floyd();
    //    printf("%d %d
",start,endd);
        
        if(road[start][endd]==inf)
        {
            printf("-1
");
        }
        else printf("%d
",road[start][endd]);
    }
    return 0;
}