Educational Codeforces Round 105 (Rated for Div. 2) B. Berland Crossword 题解

题意

给出五个数 $n，U，R，D，L$，代表一个 $n\times n$ 的网格最上层黑格个数$（U）$，最右列的黑格个数$（R）$，最下层的黑格个数$（D）$和最左列的黑格个数$（L）$。求这样的网格是否存在。

思路

对四个角的网格的所有可能判定是否存在。四个角的网格所有可能的枚举可以用 $bitset$ 来实现。

Accepted code

/*
 * @Autor: CofDoria
 * @Date: 2021-03-06 09:43:23
 * @LastEditTime: 2021-03-06 14:23:21
 */
#include <bits/stdc++.h>
using namespace std;

#define db double
#define ll long long
#define inf 0x3f3f3f3f
#define s(a, n) memset(a, n, sizeof(a))
#define debug(a) cout << '#' << a << '#' << endl
#define rep(l, a, b) for (register ll l = a; l < b; ++l)
#define per(l, a, b) for (register ll l = a; l >= b; --l)
#define _ios ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define _forit(i, c) \
    for (__typeof__((c).begin()) i = (c).begin(); i != (c).end(); ++i)

bool fi = true;
const unsigned long long MOD = 1e9 + 7;

inline ll gcd(ll a, ll b) { return (b == 0 ? a : gcd(b, a % b)); }

int t, n, u, r, d, l, s;
bool ok;

bool f(int x, int l, int r) {
    if (l) --x; // 右边顶角存在黑格，传入值减一
    if (r) --x; // 左边同理
    if (x >= 0 && x <= n - 2) return true; // 判断是否合法
    return false;
}

int main() {
    // freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
    _ios;
    cin >> t;
    while (t--) {
        cin >> n >> u >> r >> d >> l;
        s = 0;
        ok = false;
        for (; s < (1 << 4); ++s) {
            bitset<4> b(s); // 将s按二进制的每位数填入b数组
            // 不合法就进入下一循环，即下一种情况的枚举
            if (!f(u, b[0], b[1])) continue;
            if (!f(r, b[1], b[2])) continue;
            if (!f(d, b[2], b[3])) continue;
            if (!f(l, b[3], b[0])) continue;
            // 合法即标记，并退出循环
            ok = true;
            break;
        }
        if (ok)
            puts("YES");
        else
            puts("NO");
    }
    // fclose(stdin);
    // fclose(stdout);
    return 0;
}