POJ2021 Relative Relatives

最新推荐文章于 2020-05-17 18:06:10 发布

原创最新推荐文章于 2020-05-17 18:06:10 发布 · 997 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#iterator #dataset #string #vector

POJ 专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种利用图论和C++ STL解决年龄计算问题的方法。通过建立父子关系图，并使用深度优先搜索(DFS)算法遍历计算各成员年龄，最终采用红黑树(RBT)或向量(vector)保持输出结果的有序性。

这题用到了STL的强大功能，简单图论，但是需要进行hash，我运用的是map来hash。

首先是建图，将父子关系连有向边，从父亲到孩子，这样我们就可以通过DFS一遍找到全部的年龄。注意map的定义。最后用一个rbt来维护最后的解有序，当然，你也可以运用向量vector来维护，总的时间复杂度是O(nlogn)的。

我的代码：

#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <string> #include <set> #include <map> using namespace std; struct Ans { string name; int age; bool operator<(const Ans& ans) const { if(age!=ans.age) return age>ans.age; else return name<ans.name; } }; map<string,int> age; map<string, map<string,int> > mp; set<Ans> st; void dfs(string name) { map<string,int>::iterator it; for(it=mp[name].begin();it!=mp[name].end();it++) { if(!age[(*it).first]) { age[(*it).first]=age[name]-(*it).second; dfs((*it).first); } } } int main() { int t,cnt=0,n,num; string a,b; char s[120]; map<string,int>::iterator it; set<Ans>::iterator iter; Ans ans; scanf("%d",&t); while(t--) { cnt++; mp.clear(); age.clear(); age["Ted"]=100; scanf("%d",&n); while(n--) { scanf("%s",s); a=s; scanf("%s",s); b=s; scanf("%d",&num); mp[a][b]=num; } dfs("Ted"); st.clear(); for(it=age.begin();it!=age.end();it++) { ans.name=(*it).first; ans.age=(*it).second; st.insert(ans); } printf("DATASET %d/n",cnt); for(iter=st.begin();iter!=st.end();iter++) { if((*iter).name=="Ted") continue; printf("%s %d/n",(*iter).name.c_str(),(*iter).age); } } return 0; }