Leetcode 990. 等式方程的可满足性 Floyd/并查集+模拟

最新推荐文章于 2023-11-14 20:09:37 发布

自信的小螺丝钉

最新推荐文章于 2023-11-14 20:09:37 发布

阅读量141

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法 c++ 并查集数据结构

本文链接：https://blog.csdn.net/qq_45791939/article/details/128029548

版权

Leetcode 专栏收录该内容

324 篇文章 0 订阅

订阅专栏

原题链接：Leetcode 990. 等式方程的可满足性

在这里插入图片描述

Floyd

class Solution {
public:
    bool equationsPossible(vector<string>& equations) {
        vector<vector<int>> g(26,vector<int>(26));
        for(int i=0;i<26;i++) g[i][i]=1;
        for(auto x:equations)
        {
            int a=x[0]-'a';
            int b=x[3]-'a';
            char ch=x[1];
            if(ch=='=')
            {
                g[a][b]=1;  g[b][a]=1;
            }
            if(ch=='!' && g[a][b]) return false;
        }
        for(int k=0;k<26;k++)
        {
            for(int i=0;i<26;i++)
            {
                for(int j=0;j<26;j++)
                {
                    if(g[i][k] && g[k][j]) g[i][j]=1;
                }
            }
        }
        for(auto x:equations)
        {
            int a=x[0]-'a';
            int b=x[3]-'a';
            char ch=x[1];
            if(ch=='!' && g[a][b]) return false;
            if(ch=='=' && !g[a][b]) return false;
        }
        return true;
    }
};

并查集

class Solution {
public:
    map<int,int> fa;
    int findfather(int x)
    {
        int a=x;
        while(x!=fa[x]) x=fa[x];
        while(a!=fa[a])
        {
            int z=a;
            a=fa[a];
            fa[z]=x;
        }
        return x;
    }
    void Union(int x,int y)
    {
        int fx=findfather(x);
        int fy=findfather(y);
        if(fx!=fy)
        {
            if(fx<fy) fa[fy]=fx;
            else fa[fx]=fy;
        }
    }
    bool equationsPossible(vector<string>& equations) {
        for(int i=0;i<26;i++) fa[i]=i;
        for(auto x:equations)
        {
            int a=x[0]-'a';
            int b=x[3]-'a';
            char ch=x[1];
            if(ch=='=') Union(a,b);
        }
        for(auto x:equations)
        { 
            char ch=x[1];
            if(ch=='!')
            {
                int a=x[0]-'a';
                int b=x[3]-'a';
                int fx=findfather(a);
                int fy=findfather(b);
                if(fx==fy) return false;
            } 
        }
        return true;
    }
};