hdu 4665 （dfs / 2-SAT）

题目链接

多校第六场的题目， 比赛时没什么想法， 后来看GYZ神犇的解题报告里用的是O(n ^ 2)的2 - SAT解法， 但本弱菜还是没想到如何构图， 后来看别人都是用dfs过的， 自己写了一版觉得还是挺好写的， 而且速度也很快， 但复杂度不知如何分析， 一开始觉得dfs应该会T的， 毕竟需要递归2000层， 后来想了想觉得这题有一定的特殊性， 因为题目只要求输出一组解， 而且如果相同同种颜色很多的话解的个数也会多， 所以dfs速度会比较快， 但准确的理论复杂度还是不知如何分析， 有时间再研究2-SAT解法吧。。。。。

#include <iostream>#include <cstring>#include <cstdio>#include <algorithm>#include <vector>#include <queue>#include <string>#include <cctype>#include <set>#include <deque>#include <map>using namespace std;inline int readint() {char c = getchar();while (!isdigit(c)) c = getchar();int x = 0;while (isdigit(c)) {x = x * 10 + c - '0';c = getchar();}return x;}int buf[10];inline void writeint(int x) {int p = 0;if (x == 0) p++;else while (x) {buf[p++] = x % 10;x /= 10;}for (int j = p - 1; j >= 0; j--)putchar('0' + buf[j]);}const int N = 2005;int cnt0[N], cnt1[N], sum[N];int res[N], s0[N], s1[N], str[N];int n;bool ok = 0;void dfs(int id, int sz0, int sz1) {if (id == n + 1) {for (int i = 1; i <= n; i++) printf("%d", res[i]);puts("");ok = 1;return;}if (cnt0[str[id]] < sum[str[id]] / 2 && (sz0 >= sz1 || str[id] == s1[sz0])) {res[id] = 0;cnt0[str[id]]++;s0[sz0] = str[id];dfs(id + 1, sz0 + 1, sz1);if (ok) return;cnt0[str[id]]--;}if (cnt1[str[id]] < sum[str[id]] / 2 && (sz1 >= sz0 || str[id] == s0[sz1])) {res[id] = 1;cnt1[str[id]]++;s1[sz1] = str[id];dfs(id + 1, sz0, sz1 + 1);if (ok) return;cnt1[str[id]]--;}}int main() {int test;test = readint();while (test--) {n = readint();for (int i = 1; i <= n; i++)cnt0[i] = 0, cnt1[i] = 0, sum[i] = 0;for (int i = 1; i <= n; i++)str[i] = readint(), sum[str[i]]++;ok = 0;s0[1] = str[1];cnt0[str[1]]++;dfs(2, 2, 1);}return 0;}