HDU 3715 Go Deeper【2-SAT+二分】

题目链接

题意：给了你一个递归方程，在满足条件x[a[dep]] + x[b[dep]] != c[dep]下最多能到第几层递归。其中a[ ]，b[ ]，c[ ]数组都是给定的，而x[]不是给定的。

看这个条件，正因为x[ ]不是给定的，所以这是一个2-SAT问题，而m最多是10000，直接线性复杂度的话会超时，所以要二分求dep的最大值，每次把前dep层的矛盾关系都建边。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int n, m;
int a[10005];
int b[10005];
int c[10005];

const int MAXN = 200020;
const int MAXM = 200010;
#define LL long long

struct Edge {
int to, next;
} edge[MAXM];
int head[MAXN], cntE;

void init() {
cntE = 0;
memset(head, -1, sizeof head);
}

void addedge(int u, int v) {
edge[cntE].to = v;
edge[cntE].next = head[u];
head[u] = cntE++;
}

bool vis[MAXN];
int stk[MAXN], top;

bool dfs(int u) {
if (vis[u ^ 1]) return false;
if (vis[u]) return true;
vis[u] = true; stk[top++] = u;
for (int i = head[u]; ~i; i = edge[i].next) {
if (!dfs(edge[i].to)) return false;
}
return true;
}

inline bool sat(int n) {
memset(vis, false, sizeof vis);
for (int i = 0; i < n; ++i) {
if (vis[i] || vis[i ^ 1]) continue;
top = 0;
if (!dfs(i)) {
while (top) vis[stk[--top]] = false;
if (!dfs(i ^ 1)) return false;
}
}
return true;
}

int T;

bool ok(int mid)
{
for (int i = 0; i < mid; i++)
{
int u = a[i];
int v = b[i];
u *= 2;
v *= 2;
if (c[i] == 0)
{
addedge(u ^ 1, v);
addedge(v ^ 1, u);
}
else if (c[i] == 1)
{
addedge(u ^ 1, v ^ 1);
addedge(v ^ 1, u ^ 1);
addedge(u, v);
addedge(v, u);
}
else
{
addedge(u, v ^ 1);
addedge(v, u ^ 1);
}
}
return sat(mid * 2);
}

int main()
{
scanf("%d", &T);
while (T--)
{
scanf("%d %d", &n, &m);
for (int i = 0; i < m; i++)   scanf("%d %d %d", &a[i], &b[i], &c[i]);
int low = 1, high = m;
while (low < high)
{
init();
int mid = (low + high + 1) >> 1;
if (ok(mid))    low = mid;
else    high = mid - 1;
}
printf("%d\n", low);
}
}