Codeforces 1971 H ±1(染色法)

题目:

题目思路:

        大佬们做这道题的时候都是一眼2-SAT,直接秒了.但我木有学,所以在赛时用染色法解决了这道题,这个方法来做的时候不像2-SAT那样典,所以需要一些思考,但因为也是O(n)的时间复杂度解决的,所以就想着分享一下思路,如果真想学习这道题最好的做法,建议还是看其他博主的2-SAT做法.

       1.中间一行都是1,说明每一列中至少有两个为正的

        2.读完题,我们会发现,如果一个矩阵里面只有(或者)是一定有解的,换言而知,只有当(和)同时出现时,才可能会出错.

        3.一组有正有负.我们可以先假设为正,那么就是负的.那么根据题意,可以得出,所在的那一列的剩余两个数和必然是正的.那么可以继续推出,如果和有对立面,那么对立面所在列的剩余两个数也必然是正的...以此类推,我们会发现,所有含有负数的列全被我们染色完了,而且染色的结果必然是一个负数带两个正数.那有人就有疑问,那一开始假设是正数的那一列就不用管了吗,是的.回到问题,我们要的情况是(至少两个正的->最多一个负的->不能有两个负的->在一列有负的情况下不能再有一个负的),所以只要考虑是负的情况就行了.

        4.如果染色的过程中,已经假设为正的情况下,在某一列出现了为正才能使得这一列的负数最多有一个,就会发生冲突,染色失败.这时候需要假设是正的,重新染色.

        5.还得考虑一件事,就是如果遍历到第i个点时,发现和都染色失败了,那这时候是直接宣布答案还是要回溯,重新对前面的点染色呢.再回到第三点的讲解,会发现我们在一次染色后,形成的是一负带两正的若干列,以及一个有一正的若干列.这时候其实是一种局部的最优状态.到第i个点失败的原因无非是两种可能,一种是出现一个列,之前没有开发过,出现了两个负数的情况,这时候再怎么改变过去也没有用,因为影响不到.第二种是出现一个列,已经已知有一个正数了,那改变过去,也不会改变世界线.说明当两次染色失败后,就是彻底的失败了.

        6.第二次染色的时候不能继承第一次染色的状态,需要把第一次染过色的点重新设置成未染色.

下面附上代码.

#include <bits/stdc++.h>
using namespace std;
int a[505][3],vis[505],qq[505],cnt;
vector<pair<int,int>>h[505][3];
bool dfs(int k1,int k2){
    vis[k1]=k2;
    qq[++cnt]=k1;
    for (auto it:h[k1][3-k2]){
        int x=it.first,y=it.second;
        for (int i=0;i<=2;i++){
            if (y==i) continue;
            int xx=abs(a[x][i]),pp;
            if (a[x][i]>0) pp=2;
            else pp=1;
            if (!vis[xx]) {
                if (!dfs(xx,pp)) return false;
            }
            else if (vis[xx]==3-pp) return false;
        }
    }
    return true;
}
void solve(int g){
    memset(vis,0,sizeof vis);
    int n;
    cin>>n;
    for (int i=0;i<=n;i++) h[i][2].clear(),h[i][1].clear();
    int f=1;
    for (int j=0;j<=2;j++){
        for (int i=1;i<=n;i++){
            cin>>a[i][j];
            if (a[i][j]>0) h[a[i][j]][2].push_back({i,j});
            else h[-a[i][j]][1].push_back({i,j});
        }
    }
    for (int i=1;i<=n;i++){
        if (vis[i]==0&&h[i][2].size()&&h[i][1].size()){
            cnt=0;
            if (!dfs(i,1)){
                for (int i=1;i<=cnt;i++) vis[qq[i]]=0;
                cnt=0;
                if (!dfs(i,2)){
                    f=0;
                    break;
                }
            }
        }
    }
    if (f) cout<<"YES\n";
    else cout<<"NO\n";
}
int main(){
    int t;
    cin>>t;
    for (int i=1;i<=t;i++){
        solve(i);
    }
    return 0;
}