问题描述(来源于以前上过的一门课)
一些家史复杂的家族会记录家谱。家谱是同一个姓的族人的一个集合,从最原始的祖先开始记录,之后每次新诞生的、与祖先有血缘关系的同姓族人都会被记录在案。
一个家庭,我们定义为一个族人和其直属同姓子女这两代人的集合。比如男性族人A,生了一个女儿B,接着生了一个儿子C,然后生了一个女儿D。那么A-BCD可以被称为一个以A为祖先的家庭。两个家庭称为相似,当且仅当两个家庭有相同的结构。比如男性族人E,生了一个女儿F,接着生了一个儿子G,然后生了一个女儿H,那么家庭E-FGH可以和家庭A-BCD称为相似。但比如男性族人I,生了两个女儿J、K,再生了一个儿子L,那么家庭I-JKL和家庭A-BCD就不能称为相似。或一个男性族人M,生了一个女儿N,接着生了一个儿子O,那家庭M-NO也不能和家庭A-BCD称为相似。或一个女性族人P,生了一个女儿Q,接着生了一个儿子R,然后生了一个女儿S,那家庭P-QRS也不能和家庭A-BCD称为相似。如果一个族人没有后代,那么他/她自己一个人也形成一个家庭。
一个子族,我们定义为一个族人和其所有直属同姓后代的集合。比如B生了一个儿子T,C生了两个女儿U、V,T生了一个女儿W,那么A-B=T~W-C=UV-D就可以称为以A为祖先的子族。同样子族判断相似的标准和家庭相同,当且仅当两个子族拥有相同的结构才能被认为相似。
现在告知一个家族的家谱,请你回答一些关于相似性的问题。
输入格式
第一行两个正整数N、Q,表示家谱的长度和询问的个数。
接下来N行描述家谱,每行的格式为两个只包含小写字母的字符串parent、child和一个字符’M’或’F’,表示名为parent的族人生了一个孩子叫做child,child的性别为’M’(代表男)或’F’(代表女)。家谱的描述以日期顺序,即同一家庭中,先描述的是哥哥/姐姐,后描述的是弟弟/妹妹。
接下来是Q行询问,每行有可能是:
1.一个字符’F’和一个字符串parent,表示询问有多少家庭和以parent为祖先的家庭相似(不包括parent自身)。
2.一个字符’S’和一个字符串parent,表示询问有多少子族和以parent为祖先的子族相似(不包含parent自身)。
最原始的祖先名字为字符串’root’,性别为男性。
族人的名字两两不同。
数据保证合法性,且数据默认没有族人之间结婚的情况。
输出格式
对于每个询问输出一行一个整数表示答案。
#include <iostream>
#include <string>
#include <queue>
#include <map>
using namespace std;
const int q1 = 999999937;
const int q2 = 999992221;
int a = 2, b = 3;
void Generate(map<string, int> &t, string parent, string child, char gender) { //生成第一个map,人名->家庭
int g;
if (gender == 'F') g = 2;
else g = 1;
t[child] = g;
t[parent] = (a*t[parent] + b*g) % q1;
}
void Family(map<string, int> t, map<int, int> &r) { //生成第二个map,家庭->拥有与该家庭一样的结构的人数
map<string, int>::iterator iter;
iter = t.begin();
while (iter != t.end()) {
r[iter->second]++;
iter++;
}
}
int main(int argc, const char * argv[]) {
int N, Q;
string parent, child, order, name;
char gender;
map<string, int> tree;
map<int, int> result;
cin >> N >> Q;
tree["root"] = 1;
for (int i = 0; i < N; i++) {
cin >> parent >> child >> gender;
Generate(tree, parent, child, gender);
}
Family(tree, result);
for (int i = 0; i < Q; i++) {
cin >> order >> name;
if (order == "F") {
cout << result[tree[name]] - 1 << endl;
}
if (order == "Q") {
//待解决
}
}
}
目前只解决了F类的询问,往后尝试解决Q类的询问。