POJ 2491 Scavenger Hunt 解题报告 再次利用map判重

这道题说的是一个路径（A->B->C->…）被拆散成很多step（B->C;A->B;…），要根据step还原路径。节点都是用string表示的。

如果用map来表示step的话，那么只要能找到路径的起点，就很容易根据step之间的关系还原路径，只需不断在map里将上一个step的value作为key来查找即可得到下一step的value。所以关键就是找到起点，起初我想同样利用此map来查找起点，结果发现找终点很容易（以终点作为key查不到value），但是起点就很难，要O(N*N)。后来想到一个很简单的办法可以在O(N)内找到起点：由于已知条件是所有step（形容A->B），所以如果记录每个step里的两个string，那么最终下来，除了起点和终点只出现一次外，其余中间各点都应该出现过两次，只需找到这两个点再稍加判断就能得到起点，典型的判重问题，或者说查找出现过指定次数的点，自然就想到要用map。于是，问题迎刃而解。

代码如下：

#include <iostream> #include <map> #include <string> #include <vector> using namespace std; map<string,string> ssmap; map<string,int> simap; int main() { int count; cin>>count; for (int i = 1;i <= count;i++) { ssmap.clear(); simap.clear(); vector<string> vs; int stepNum; cin>>stepNum; string str1,str2; for (int j = 0;j < stepNum-1;j++) { cin>>str1>>str2; ssmap[str1] = str2; simap[str1]++; simap[str2]++; } string head_tail[2]; int k = 0; for (map<string,int>::iterator it = simap.begin();it != simap.end();it++) { if(it->second==1) { head_tail[k++]=it->first; } } string head; for (int j = 0;j < 2;j++) { if(ssmap[head_tail[j]].size()==0) { head = head_tail[1-j]; } } cout<<"Scenario #"<<i<<":"<<endl<<head<<endl; while (ssmap[head].size()!=0) { cout<<ssmap[head]<<endl; head = ssmap[head]; } cout<<endl; } }