Codeforces Gym 101158 F. Three Kingdoms of Bourdelot

题意

有多个文档。 每个文档都有多行家谱信息。 有两种类型的文档: 正和负, 每行是一对 x, y,

• 如果文档为正: x 是 y 的祖先
• 如果文档为负: x 不是 y 的祖先

每个文档都可以有正面或负面的理解。 如果文档数量 n 和两个人 p, q 是给定的, 确定是否有一种组合（确定每篇文档的正负），使得满足 p 是 q 的祖先。

解题思路

数据结构。

• 存储 x, y 属于哪几篇文档，
• 存储每篇文档包含的 x, y 。
• 记录二维数组 isAnc[i][j]=1 表示 i 是 j 的祖先

对于已确定 x, y， 其对应的文档都应该是正面的，即对应文档中的所有记录均为正确的。根据这些记录的正确性及原本的祖孙关系，产生新的 x, y ，判断是否存在 isAnc[i][j] = isAnc[j][i] = 1 的情况，存在为 No ，否则一定为 Yes 。

代码

#include<bits/stdc++.h>using namespace std;const int NAME = 300+10;const int N = 100000 + 10;string p, q, a, b;int n, idx;vector<pair<int, int> > g[1010];map<string, int> str2int;vector<int> belg[NAME][NAME];bool isAnc[NAME][NAME], vis[1010];bool dfs(int ancestor, int son);bool setAncestor(int ancestor, int son){    if(isAnc[son][ancestor] == 1)   return false;    isAnc[ancestor][son] = 1;    for(int i=1;i<=idx;i++)    {        if(isAnc[i][ancestor] == 1)        {            if(isAnc[son][i] == 1)                return false;            else {                isAnc[i][son] = 1;                if(dfs(i, son) == false)    return false;            }        }        if(isAnc[son][i] == 1)        {            if(isAnc[i][ancestor] == 1) return false;            else {                isAnc[ancestor][i] = 1;                if(dfs(ancestor, i) == false)   return false;            }        }    }    return true;}bool dfs(int ancestor, int son){    for(int i=0, id;i<belg[ancestor][son].size();i++)    {        id = belg[ancestor][son][i];        if(vis[id]) continue;        vis[id] = 1;        for(int j=0;j<g[id].size();j++)        {            if(setAncestor(g[id][j].first, g[id][j].second) == false)                return false;            if(dfs(g[id][j].first, g[id][j].second) == false)                return false;        }    }    return true;}int main(){    cin>>p>>q;    str2int[p] = ++idx;    str2int[q] = ++idx;    scanf("%d", &n);    for(int i=1, m;i<=n;i++)    {        scanf("%d", &m);        for(int j=0;j<m;j++)        {            cin>>a>>b;            if(str2int.find(a) == str2int.end())                str2int[a] = ++idx;            if(str2int.find(b) == str2int.end())                str2int[b] = ++idx;            belg[str2int[a]][str2int[b]].push_back(i);            g[i].push_back(make_pair(str2int[a], str2int[b]));        }    }    isAnc[str2int[p]][str2int[q]] = 1;    if(dfs(str2int[p], str2int[q]))    {        for(int i=1;i<=idx;i++)        for(int j=i+1;j<=idx;j++)            if(isAnc[i][j] == isAnc[j][i] && isAnc[i][j] == 1)            {                printf("No\n");                return 0;            }        printf("Yes\n");    }    else    printf("No\n");}