Codeforces Round #441 E. National Property (2-sat tarjan后输出任意一组合法解)


题目链接:http://codeforces.com/contest/876/problem/E


题意:现在我们想要让所有行的字典序升序排列,且对于每一种数字,我们可以采用一种变换,比如 x->x' ,所有加 ' 的都比不加小,问是否可以找到一种变换使得原序列升序。

 

思路:

只要相邻两行满足升序则原序列升序。

于是我们考虑相邻的两行 a,b ,找到第一个 ai!=bi ,则这两个序列之间的大小由这两个字母来决定,

既然要升序,假如 ai<bi ,则此时已经满足升序的条件,选择 ai 则必须选择 bi ,选择 bi 则必须选择 ai ,连边 ai>bi,bi>ai

假如 ai>bi ,要使原序列升序则必须选择 ai,bi ,连边 ai>ai,bi>bi 


然后运用tarjan输出解,我原以为要在拓扑排序的过程选择哪些color的块可以选,选完某一个color的块,就要把其他的对立的color块标记上不可选。实际上,根本不需要标记对立的color块不可选。可以证明在2-sat经过tarjan后的DAG中,两个对立的点(i * 2, i * 2 + 1)中拓扑序小的那个点必然可以选,以这种规则产生的方案一定合法。 而且tarjan的过程中就能把每个点的拓扑序顺便处理出来了,代码量就很短了。

代码:

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

#define REP(i,n) for ( int i=1; i<=int(n); i++ )  
#define MP make_pair
#define PB push_back
#define SZ(x) (int((x).size()))
#define ALL(x) (x).begin(), (x).end()
#define X first
#define Y second
template inline bool chkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template inline bool chkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

typedef long long LL;
typedef long double LD;
const int INF = 0x3f3f3f3f;

template   
inline bool RD(T &ret) {  
        char c; int sgn;  
        if (c = getchar(), c == EOF) return 0;  
        while (c != '-' && (c<'0' || c>'9')) c = getchar();  
        sgn = (c == '-') ? -1 : 1 , ret = (c == '-') ? 0 : (c - '0');  
        while (c = getchar(), c >= '0'&&c <= '9') ret = ret * 10 + (c - '0');  
        ret *= sgn;  
        return 1;  
}  
template   
inline void PT(T x) {  
        if (x < 0) putchar('-') ,x = -x;  
        if (x > 9) PT(x / 10);  
        putchar(x % 10 + '0');
}
typedef pair pii;


const int N = 2e5 + 100;  
vector g[N];  
  
inline void add_edge(int u,int v)  
{  
    g[u].push_back(v);  
}  

int dfn[N],low[N], col[N], tmp[N],sta[N],top;
int topo_index[N], tot;
int indx, scc; 
void tarjan(int u)  
{  
    low[u]=dfn[u]=++indx;  
    tmp[u]=1;  
    sta[++top]=u;  
    int sz=g[u].size();  
    for(int i=0;i vec[N];
vector res;
int main() {
    int n, m;
    cin >> n >> m;
    bool flag = false;
    for(int i = 1; i <= n; i ++) {
        int k, num;
        scanf("%d", &k);
        bool f = true;
        for(int j = 0; j < k; j ++) {
            scanf("%d", &num);
            vec[i].push_back(num);
            if(i != 1 && f && vec[i - 1].size() >= j + 1) {
                if(vec[i - 1][j] > vec[i][j]) {
                    add_edge(vec[i - 1][j] << 1, vec[i - 1][j] << 1 | 1);
                    add_edge(vec[i][j] << 1 | 1, vec[i][j] << 1); 
                    f = false;
                } else if(vec[i - 1][j] < vec[i][j]) {
                    add_edge(vec[i - 1][j] << 1, vec[i][j] << 1);
                    add_edge(vec[i][j] << 1 | 1, vec[i - 1][j] << 1 | 1);
                    f = false;
                }
            }
        }
        if(i != 1 && f && SZ(vec[i - 1]) > SZ(vec[i])) flag = true;
    }

    if(flag) return puts("No") * 0;

    for(int i = 2; i < 2 * (m + 1); i ++) 
        if(dfn[i] == 0) tarjan(i);

    for(int i = 2; i < 2 * (m + 1); i ++) {
        if(col[i] == col[i ^ 1]) {
            puts("No");
            return 0;
        }
    }

    for(int i = 1; i <= m; i ++) {
        if(topo_index[i << 1 | 1] < topo_index[i << 1]) 
            res.push_back(i);
    }
    puts("Yes");
    printf("%d\n", SZ(res));
    for(int v : res) printf("%d\n", v);

}


不知道这种套路后写出来的sb的代码:


#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

#define REP(i,n) for ( int i=1; i<=int(n); i++ )  
#define MP make_pair
#define PB push_back
#define SZ(x) (int((x).size()))
#define ALL(x) (x).begin(), (x).end()
#define X first
#define Y second
template inline bool chkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template inline bool chkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

typedef long long LL;
typedef long double LD;
const int INF = 0x3f3f3f3f;

template   
inline bool RD(T &ret) {  
        char c; int sgn;  
        if (c = getchar(), c == EOF) return 0;  
        while (c != '-' && (c<'0' || c>'9')) c = getchar();  
        sgn = (c == '-') ? -1 : 1 , ret = (c == '-') ? 0 : (c - '0');  
        while (c = getchar(), c >= '0'&&c <= '9') ret = ret * 10 + (c - '0');  
        ret *= sgn;  
        return 1;  
}  
template   
inline void PT(T x) {  
        if (x < 0) putchar('-') ,x = -x;  
        if (x > 9) PT(x / 10);  
        putchar(x % 10 + '0');
}
typedef pair pii;


const int N = 2e5 + 100;  
vector g[N];  
  
inline void add_edge(int u,int v)  
{  
    g[u].push_back(v);  
}  

int dfn[N],low[N], col[N], tmp[N],sta[N],top;  
int indx, scc; 
void tarjan(int u)  
{  
    low[u]=dfn[u]=++indx;  
    tmp[u]=1;  
    sta[++top]=u;  
    int sz=g[u].size();  
    for(int i=0;i vec[N];
vector G[N];
int deg[N];
bool print[N];
vector book[N];
int que[N], head, tail;
vector res;
int main() {
    int n, m;
    cin >> n >> m;
    bool flag = false;
    for(int i = 1; i <= n; i ++) {
        int k, num;
        scanf("%d", &k);
        bool f = true;
        for(int j = 0; j < k; j ++) {
            scanf("%d", &num);
            vec[i].push_back(num);
            if(i != 1 && f && vec[i - 1].size() >= j + 1) {
                if(vec[i - 1][j] > vec[i][j]) {
                    add_edge(vec[i - 1][j] << 1, (vec[i - 1][j] << 1) ^ 1);
                    add_edge((vec[i][j] << 1) ^ 1, vec[i][j] << 1); 
                    f = false;
                } else if(f && vec[i - 1][j] < vec[i][j]) {
                    add_edge(vec[i - 1][j] << 1, vec[i][j] << 1);
                    add_edge((vec[i][j] << 1) ^ 1, (vec[i - 1][j] << 1) ^ 1);
                    f = false;
                }
            }
        }
        if(i != 1 && f && SZ(vec[i - 1]) > SZ(vec[i])) flag = true;
    }

    if(flag) return puts("No") * 0;

    for(int i = 2; i < 2 * (m + 1); i ++) 
        if(dfn[i] == 0) tarjan(i);

    for(int i = 2; i < 2 * (m + 1); i ++) {
        book[col[i]].push_back(col[i ^ 1]);
        if(col[i] == col[i ^ 1]) {
            puts("No");
            return 0;
        }
    }

    for(int u = 2; u < 2 * (m + 1); u ++) {
        for(int j = 0; j < SZ(g[u]); j ++) {
            int v = g[u][j];
            if(col[u] == col[v]) continue;
            deg[col[u]] ++;
            G[col[v]].push_back(col[u]);
        }
    }

    for(int i = 1; i <= scc; i ++) 
        if(deg[i] == 0) que[tail ++] = i;

    while(head < tail) {
        int u = que[head ++];
        if(print[u] == 0) {
            for(int v : book[u]) print[v] = 1;
        }
        for(int v : G[u]) if(--deg[v] == 0) que[tail ++] = v;
    }
    
    for(int i = 1; i <= m; i ++) {
        if(print[col[(i << 1) ^ 1]] == 0) res.push_back(i);
    }
    puts("Yes");
    printf("%d\n", SZ(res));
    for(int v : res) printf("%d\n", v);

}

 

你可能感兴趣的:(2-sat)