LeetCode算法网站算法题
https://leetcode-cn.com/problems/satisfiability-of-equality-equations/
思路:
并查集的应用,把所有相等的节点并入同一个集合,即拥有相同祖先,然后在遍历不相等的变量,如果两个不相等的变量仍然存在于相等的那一个集合之中,则返回错误。
class union_set
{
private:
vector<int>parents;
public:
union_set();//初始化并查集数据
int find_parents(int index);//寻找节点的最终父节点
void unite(int index1,int index2);//整合两个节点
};
union_set::union_set()
{
int i;
for(i=0; i<26; i++)
parents.push_back(i);//初始化所有节点,把所有节点的祖先设置为自己
}
int union_set::find_parents(int index)
{
if(index==parents[index])
return index;
parents[index]=find_parents(parents[index]);//所以在这里就能明白为什么要改变当前节点的祖先值了,因为如果当前节点的祖先节点都有了别的祖先节点,那么就需要把当前节点的祖先节点改变为原来祖先节点的祖先节点
//还有很重要的一点是,有可能当前节点在并查集合并的时候没有更改祖先节点,但是在后来判断的时候也可以更改自己的祖先节点,这样判断的结果仍然是正确的
return parents[index];
}
void union_set::unite(int index1,int index2)
{
parents[find_parents(index1)]=find_parents(index2);//把index1的祖先节点的祖先由原来祖先的祖先赋值为index2的祖先节点
}
class Solution
{
public:
bool equationsPossible(vector<string>& equations)
{
union_set us;
for (const string& str: equations)
{
if (str[1] == '=')
{
int index1 = str[0] - 'a';
int index2 = str[3] - 'a';
us.unite(index1, index2);//连接两个相等的节点
}
}
for (const string& str: equations)
{
if (str[1] == '!')
{
int index1 = str[0] - 'a';
int index2 = str[3] - 'a';
if (us.find_parents(index1) == us.find_parents(index2))//如果不应该相等的两个元素,在同一个相等集合中(即拥有公共祖先),就返回假
{
return false;
}
}
}
return true;
}
};