BZOJ3495 PA2010 Riddle 【2-sat】

题目链接

BZOJ3495

题解

每个城市都有选和不选两种情况，很容易考虑到2-sat
边的限制就很好设置了，主要是每个郡只有一个首都的限制
我们不可能两两之间连边，这样复杂度就爆炸了

于是乎就有了一个非常厉害的方法：
前缀后缀和
我们令\(1\)表示选，\(0\)表示不选，维护一个郡的前缀和、后缀和
一个点如果作为首都，那么它和它对应前缀后缀和的位置一定为\(1\)，且之前的位置一定为\(0\)
然后再加上一些前缀后缀和固有的限制

就可以\(A\)啦
这样的建图太神了

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<map>
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define mp(a,b) make_pair<int,int>(a,b)
#define cls(s) memset(s,0,sizeof(s))
#define cp pair<int,int>
#define LL long long int
using namespace std;
const int maxn = 6000005,maxv = 1000005,maxm = 15000005,INF = 1000000000;
inline int read(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
    while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
    return out * flag;
}
int n,m,K,N,pre[maxv][2],sa[maxv][2],p[maxv][2];
int dfn[maxn],low[maxn],Scc[maxn],st[maxn],top,scci,cnt;
int h[maxn],ne;
struct EDGE{int to,nxt;}ed[maxm];
inline void build(int u,int v){
    ed[++ne] = (EDGE){v,h[u]}; h[u] = ne;
}
void dfs(int u){
    dfn[u] = low[u] = ++cnt;
    st[++top] = u;
    Redge(u){
        if (!dfn[to = ed[k].to]){
            dfs(to);
            low[u] = min(low[u],low[to]);
        }
        else if (!Scc[to]) low[u] = min(low[u],dfn[to]);
    }
    if (dfn[u] == low[u]){
        scci++;
        do{
            Scc[st[top]] = scci;
        }while (st[top--] != u);
    }
}
int main(){
    n = read(); m = read(); K = read();
    REP(i,n) p[i][0] = ++N,p[i][1] = ++N;
    int a,b;
    while (m--){
        a = read(); b = read();
        build(p[a][0],p[b][1]);
        build(p[b][0],p[a][1]);
    }
    int x,u;
    while (K--){
        x = read();
        REP(i,x){
            pre[i][0] = ++N,pre[i][1] = ++N;
            sa[i][0] = ++N,sa[i][1] = ++N;
        }
        REP(i,x){
            u = read();
            build(p[u][1],pre[i][1]); build(pre[i][0],p[u][0]);
            build(p[u][1],sa[i][1]); build(sa[i][0],p[u][0]);
            if (i < x){
                build(p[u][1],sa[i + 1][0]); build(sa[i + 1][1],p[u][0]);
                build(pre[i][1],pre[i + 1][1]); build(pre[i + 1][0],pre[i][0]);
            }
            else build(pre[i][0],pre[i][1]);
            if (i > 1){
                build(p[u][1],pre[i - 1][0]); build(pre[i - 1][1],p[u][0]);
                build(sa[i][1],sa[i - 1][1]); build(sa[i - 1][0],sa[i][0]);
            }
            else build(sa[i][0],sa[i][1]);
        }
    }
    for (int i = 1; i <= N; i++) if (!dfn[i]) dfs(i);
    for (int i = 1; i <= N; i += 2)
        if (Scc[i] == Scc[i + 1]){
            puts("NIE");
            return 0;
        }
    puts("TAK");
    return 0;
}