题目链接

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;
}