题目链接:http://acm.timus.ru/problem.aspx?space=1&num=1837
题目大意:
有一个 叫Isenbaev的人每个人想要计算和Isenbaev的距离,同组的为1,如果是队友的朋友为2,以此类推
思路:
用map把每个人映射成编号,在用dijkstra计算最短距离。第一次使用map记录一下
代码:
#include <cstdio>
#include <string>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <map>
using namespace std;
const int maxn = 310;
const int inf = 0x3f3f3f3f;
string a[maxn][3];
map<string, int>m;
int g[maxn][maxn], d[maxn];
bool used[maxn];
void dijkstra(int s, int V)
{
fill(used, used + V, false);
fill(d, d + V, inf);
d[s] = 0;
while(true)
{
int v = -1;
for(int u = 0; u < V; u++)
{
if(!used[u] && (v == -1 || d[u] < d[v])) v = u;
}
if(v == -1) break;
used[v] = true;
for(int u = 0; u < V; u++)
{
d[u] = min(d[u], d[v] + g[v][u]);
}
}
}
int main()
{
int n;
cin>>n;
for(int i = 0; i < n; i++)
{
cin>>a[i][0]>>a[i][1]>>a[i][2];
// m[a[i][0]] = 0;
// m[a[i][1]] = 0;
// m[a[i][2]] = 0;
m[a[i][0]] = m[a[i][1]] = m[a[i][2]] = 0;
}
int num = 0;
map<string, int>::iterator it;
for(it = m.begin(); it != m.end(); it++)
{
it->second = ++num;
}
// for(int i=0; i<maxn; i++)
// for(int j=0; j<maxn; j++) g[i][j] = inf;
memset(g, inf,sizeof(g));
for(int i = 0 ; i < n; i++)
{
int f1 = m.find(a[i][0])->second;
int f2 = m.find(a[i][1])->second;
int f3 = m.find(a[i][2])->second;
g[f1][f2] = g[f1][f3] = g[f2][f3] = 1;
g[f2][f1] = g[f3][f1] = g[f3][f2] = 1;
}
int len = m.size();
it = m.find("Isenbaev");
if(it == m.end())
{
for(it = m.begin(); it != m.end(); it++)
{
cout<<it->first<<" "<<"undefined"<<endl;
}
}
else
{
dijkstra(it -> second, len + 1);
for(it = m.begin(); it != m.end(); it++)
{
cout<<it -> first<<" ";
if(d[it -> second] == inf) cout<<"undefined"<<endl;
else cout<<d[it -> second]<<endl;
}
}
return 0;
}