UVa Problem Solution: 10044 - Erdos Number

The unclear specifications on the input format and size lead to the undeserving difficulties to solve this problem.  After numerous frustrating REs and WAs, I was forced to switch to C++. I'm still not sure about the judge's data format, but I believe its size is smaller 5000.

The follow code uses a simple BFS on the coauthor graph. The parsing of paper database is relatively simple but tricky. It could be surely much simpler if the judge's data format strictly conforming to the specification.

Code:

1. /*************************************************************************
2.  * Copyright (C) 2008 by liukaipeng                                      *
3.  * liukaipeng at gmail dot com                                           *
4.  *************************************************************************/
6. /* @JUDGE_ID 00000 10044 C++ "Erdos Number" */
8. #include <iostream>
9. #include <fstream>
10. #include <string>
11. #include <vector>
12. #include <map>
13. #include <queue>
14. #include <limits>
16. using namespace std;
18. void get_authors(vector<string> const& papers, 
19.                  vector<vector<size_t> >& authors,
20.                  map<string, size_t>&namelist)
21. {
22.   for (int i = 0; i < papers.size(); ++i) {
23.     string const&p = papers[i];
24.     for (int fb, fe = 0, lb, le = 0; p[fe] != ':' && p[le] != ':'; ) {
25.       fb = p.find_first_not_of(" ,", le);
26.       fe = p.find_first_of(" ,:", fb);
27.       lb = p.find_first_not_of(" ,", fe);
28.       le = p.find_first_of(" ,:", lb);
29.       if (fb == string::npos || fe == string::npos || 
30.           lb == string::npos || le == string::npos) {
31.         break;
32.       }
33.       string name = p.substr(fb, fe-fb) + ", " + p.substr(lb, le-lb);
34.       map<string, size_t>::iterator it = namelist.find(name);
35.       if (it == namelist.end()) {
36.         it = namelist.insert(it, make_pair(name, namelist.size()));
37.       }
38.       authors[i].push_back(it->second);
39.     }
40.   }
41. }
43. void get_coauthors(vector<vector<size_t> > const& authors,
44.                    vector<vector<size_t> >& coauthors)
45. {
46.   for (int i = 0; i < authors.size(); ++i) {
47.     vector<size_t> const& a = authors[i];
48.     if (a.empty()) {
49.       continue;
50.     }
51.     for (int a1 = 0; a1 < a.size() - 1; ++a1) {
52.       for (int a2 = a1 + 1; a2 < a.size(); ++a2) {
53.         coauthors[a[a1]].push_back(a[a2]);
54.         coauthors[a[a2]].push_back(a[a1]);
55.       }
56.     }
57.   }
58. }
60. void get_erdos_numbers(vector<vector<size_t> > const& coauthors, 
61.                        vector<int>& erdos_numbers, size_t s)
62. {
63.   vector<bool> visited(coauthors.size());
64.   queue<pair<size_t, int> > q;
65.   size_t v = s;
66.   int d = 0;
67.   q.push(make_pair(v, d));
68.   visited[v] = true;
69.   while (!q.empty()) {
70.     s = q.front().first;
71.     d = q.front().second;
72.     q.pop();
73.     erdos_numbers[s] = d;
74.     for (int i = 0; i < coauthors[s].size(); ++i) {
75.       v = coauthors[s][i];
76.       if (!visited[v]) {
77.         q.push(make_pair(v, d+1));
78.         visited[v] = true;
79.       }
80.     }
81.   }
82. }
84. void put_erdos_numbers(vector<string> const& names,
85.                        map<string, size_t> const& namelist, 
86.                        vector<int> const& erdos_numbers)
87. {
88.   for (int i = 0; i < names.size(); ++i) {
89.     if (!names[i].empty()) {
90.       map<string, size_t>::const_iterator it = namelist.find(names[i]);
91.       if (it != namelist.end() && erdos_numbers[it->second] != -1) {          
92.         cout << names[i] << " " << erdos_numbers[it->second] << "/n";
93.       } else {
94.         cout << names[i] << " infinity/n";
95.       }
96.     } else {
97.       cout << "/n";
98.     }
99.   }
100. }
102. int main(int argc, char *argv[])
103. {
104. #ifndef ONLINE_JUDGE
105.   string name = argv[0];
106.   filebuf infb, outfb;
107.   cin.rdbuf(infb.open((name + ".in").c_str(), ios_base::in));
108.   cout.rdbuf(outfb.open((name + ".out").c_str(), ios_base::out));
109. #endif
111.   int scenarios;
112.   cin >> scenarios;
113.   for (int s = 1; s <= scenarios; ++s) {
114.     int p, n;
115.     cin >> p >> n;
116.     cin.ignore(numeric_limits<streamsize>::max(), '/n');
118.     vector<string> papers(p);
119.     for (int i = 0; i < p; ++i) {
120.       getline(cin, papers[i]);
121.     }
123.     map<string, size_t> namelist; /* map author name to author id */
124.     vector<vector<size_t> > authors(p); 
125.     get_authors(papers, authors, namelist);
127.     vector<vector<size_t> > coauthors(namelist.size());
128.     get_coauthors(authors, coauthors);
130.     vector<int> erdos_numbers(namelist.size(), -1);
131.     map<string, size_t>::const_iterator it = namelist.find("Erdos, P.");
132.     get_erdos_numbers(coauthors, erdos_numbers, it->second);
134.     vector<string> names(n);
135.     for (int i = 0; i < n; ++i) {
136.       getline(cin, names[i]);
137.     }
138.     cout << "Scenario " << s << "/n";
139.     put_erdos_numbers(names, namelist, erdos_numbers);
140.   }
141.   return 0;
142. }