Coding a Dijkstra is not hard. %70 of my time spent on tackling TLE, as my last post.
Dijkstra works in BFS manner, but at each step, it picks the shortest child greedily and then relax all other neighbors.
Data Structure: since there could be up to 10000 cities, storing distance for all pairs of cities are not allowed - 300+M space exceeds SPOJ's 256M limit. So I simply used std::map (my another version used GNU's hash_map but not tried its performance) as the adjacent linked list to store graph; and map<string, int> to store city name - index mapping. Again, hash_map could be better.
Anyway, here's my AC code, 11.34s:
// 440 #include <vector> #include <cstdio> #include <iostream> #include <queue> #include <map> using namespace std; /* ////////////////////////////////// // Components for GNU's hash_map #include <backward/hash_map> using namespace __gnu_cxx; namespace __gnu_cxx { template<> struct hash<std::string> { hash<char*> h; size_t operator()(const std::string &s) const { return h(s.c_str()); }; }; } */ ////////////////////////////////// const int MaxCity = 10000; const int MaxDist = 0xFFFFFFF; map<string, int> hmap; typedef map<int, map<int, int> > DHash; DHash distMap; bool visited[MaxCity] = {false}; int dist[MaxCity] = {MaxDist}; ///////////////////////////////////// void add_dist(int srcInx, int toInx, int cost) { distMap[srcInx][toInx] = cost; } int get_dist(int srcInx, int toInx) { DHash::iterator iter = distMap.find(srcInx); if(iter != distMap.end()) { map<int, int>::iterator iter2 = iter->second.find(toInx); if(iter2 != iter->second.end()) { return iter2->second; } } return MaxDist; } ///////////////////////////////////// struct nComp { int operator() (const pair<int,int>& p1, const pair<int,int> &p2) { return p1.second > p2.second; } }; /////////////////////////////////////// void clearGraph() { hmap.clear(); distMap.clear(); } void clearRuntime(int citycnt) { for(int i = 0; i < citycnt; i ++) { visited[i] = false; dist[i] = MaxDist; } } /////////////////////////////////////////// int find_shortest_path(int cinx1, int cinx2) { int vcnt = hmap.size(); priority_queue<pair<int, int>, vector<pair<int, int> >, nComp> q; // Init for(int i = 0; i < vcnt; i ++) { dist[i] = MaxDist; visited[i] = false; } dist[cinx1 - 1] = 0; q.push(make_pair(cinx1-1, 0)); // Go while(!q.empty()) { pair<int,int> n = q.top(); q.pop(); // Greedy, find the unvisited min-cost node int minInx = n.first; int minCost = n.second; if(minCost == MaxDist) break; visited[minInx] = true; if(minInx == cinx2 - 1) return dist[cinx2 - 1]; // Relax neighbors DHash::iterator iter = distMap.find(minInx); map<int,int> &child = iter->second; for(map<int,int>::iterator iMap = child.begin(); iMap != child.end(); iMap ++) { int i = iMap->first; int gDist = iMap->second; int alt = dist[minInx] + gDist; if(alt < dist[i]) { dist[i] = alt; //cout << "\t\trelaxed to " << alt << " " << minInx << " to" << i <<endl; q.push(make_pair(i, dist[i])); } } } return MaxDist; } ///////////////////////// #define gc getchar int read_int() { char c = gc(); while(c<'0' || c>'9') c = gc(); int ret = 0; while(c>='0' && c<='9') { ret = 10 * ret + c - 48; c = gc(); } return ret; } ///////////////////////// int main() { int runcnt = read_int(); //cout << "Run " << runcnt << endl; while(runcnt --) { clearGraph(); int ccnt = read_int(); for(int ic = 1; ic <= ccnt; ic ++) { // cityName -> cityInx string city_name; cin >> city_name; int nnbr = read_int(); hmap.insert(map<string, int>::value_type(city_name, ic)); //cout << city_name << "->" << ic << endl; // Fill out adjacency matrix while(nnbr--) { int cinx = read_int(); int cost = read_int(); add_dist(ic-1, cinx-1, cost); //cout << ic << " <-> " << cinx << " = " << cost << endl; } } int qcnt = read_int(); while(qcnt --) { string c1, c2; cin >> c1 >> c2; int cinx1 = hmap[c1]; int cinx2 = hmap[c2]; // cout << c1 << "-" << c2 << " = " << cinx1 << " - " << cinx2 << endl; clearRuntime(ccnt); cout << find_shortest_path(cinx1, cinx2) << endl; } getchar(); getchar(); // should return 0xA (enter) } return 0; }
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步