ICM Technex 2017 and Codeforces Round 400 D. The Door Problem 【2-SAT】

吵闹的人群保持笑容多冷静

已于 2024-04-09 09:04:09 修改

阅读量910

点赞数 11

分类专栏： codeforces 练习文章标签：算法

于 2024-04-09 08:44:39 首次发布

本文链接：https://blog.csdn.net/m0_73500785/article/details/137534679

版权

codeforces 练习专栏收录该内容

51 篇文章 0 订阅

订阅专栏

D. The Door Problem

题意

有 $n$ 个门和 $m$ 个开关，每个门恰好由 $2$ 个开关控制
起初每个门可能处于开启( $1$ ) 状态或关闭( $0$ ) 状态，每按下一个控制这扇门的开关，这扇门的状态都会翻转

问是否存在一种按下开关的方案，使得所有门都开启？

思路

考虑 $2 - S A T$ ，每个开关只会有两种状态：按下 $(1)$ 和不按 $(0)$ （按两下等价于不按）
可以发现：

对于已经开启的门，其两个开关要么同时不按，要么同时按下
对于关闭的门，其两个开关有且仅有一个开关被按下（两个开关状态不同）

假设其两个开关为 $u 、 v$ ，
对于限制 $1$ ：连边 $\lrarr v$ ， $\bar u \lrarr \bar v$
对于限制 $2$ ：连边 $\lrarr \bar v$ ， $\bar u \lrarr v$

跑 $2 - S A T$ 即可

时间复杂度： $O (n + m)$

#include<bits/stdc++.h>
#define fore(i,l,r)	for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n'
#define ull unsigned long long
#define ALL(v) v.begin(), v.end()
#define Debug(x, ed) std::cerr << #x << " = " << x << ed;

const int INF=0x3f3f3f3f;
const long long INFLL=1e18;

typedef long long ll;

struct TwoSat {
    int n; //属性数量
    std::vector<std::vector<int>> e;
    std::vector<bool> ans;
    TwoSat(int n) : n(n), e(2 * n), ans(n) {} //下标从0开始

    /* 建边表示 u为f 且 v为g */
    // void addedge(int u, bool f, int v, bool g) { //原变量和反变量相邻放
        // e[2 * u + !f].push_back(2 * v + g); //反变量在偶数位置，原变量在奇数位置
        // e[2 * v + !g].push_back(2 * u + f);

    // }

    void addedge(int u, int v, int same){
        e[2 * u + 1].push_back(2 * v + same);
        e[2 * v + 1].push_back(2 * u + same);
        e[2 * u].push_back(2 * v + (same ^ 1));
        e[2 * v].push_back(2 * u + (same ^ 1));
    }

    bool satisfiable() {
        std::vector<int> id(2 * n, -1), dfn(2 * n, -1), low(2 * n, -1);
        std::vector<int> stk;
        int now = 0, cnt = 0;
        std::function<void(int)> tarjan = [&](int u) {
            stk.push_back(u);
            dfn[u] = low[u] = now++;
            for (auto v : e[u]) {
                if (dfn[v] == -1) {
                    tarjan(v);
                    low[u] = std::min(low[u], low[v]);
                } else if (id[v] == -1) {
                    low[u] = std::min(low[u], dfn[v]);
                }
            }
            if (dfn[u] == low[u]) {
                int v;
                do {
                    v = stk.back();
                    stk.pop_back();
                    id[v] = cnt;
                } while (v != u);
                ++cnt;
            }
        };
        for (int i = 0; i < 2 * n; ++i) if (dfn[i] == -1) tarjan(i);
        for (int i = 0; i < n; ++i) {
            if (id[2 * i] == id[2 * i + 1]) return false;
            ans[i] = id[2 * i] > id[2 * i + 1]; //取依赖性更高的那个
        }
        return true;
    }

    std::vector<bool> get_ans() { return ans; }
};


int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    int n, m;
    std::cin >> n >> m;
    TwoSat twosat(m);
    std::vector<int> a(n);
    std::vector<std::vector<int>> b(n);
    fore(i, 0, n) std::cin >> a[i];
    fore(i, 0, m){
        int k;
        std::cin >> k;
        while(k--){
            int x;
            std::cin >> x;
            --x;
            b[x].push_back(i);
        }
    }

    fore(i, 0, n){
        int u = b[i][0];
        int v = b[i][1];
        twosat.addedge(u, v, a[i]);
    }

    if(twosat.satisfiable()) std::cout << "YES";
    else std::cout << "NO";

    return 0;
}