[枚举] aw3711. 方格涂色(二进制枚举+模拟)

1. 题目来源

链接：3711. 方格涂色

2. 题目解析

二进制枚举题目，按题模拟即可。

四角比较特殊，每个角有两种情况，放、不放。则共有 2^4=16 种情况，则共枚举这 16 种情况，每个角上点涂色会影响它所在的两条边，判断四条边上中间 n-2 个点数量是否合法即可。

---

注意这题是多组数据…在做的过程中不注意修改了原数据…一直错。

---

时间复杂度：$O(2^4)$

空间复杂度：$O(1)$

---

#include <bits/stdc++.h>

using namespace std;

const int N = 4;

int n;
int w[N];
int a[N];
bool b[N];      // bool 来判断四角

bool check(int x) {
    memset(b, 0, sizeof b);         // 多组数据...
    memcpy(a, w, sizeof w);
    
    for (int i = 0; i < 4; i ++ ) 
        if (x >> i & 1) 
            b[i] = true;
    
    // 处理四角对边的影响
    if (b[0]) a[0] -- , a[3] -- ;
    if (b[1]) a[0] -- , a[1] -- ;
    if (b[2]) a[1] -- , a[2] -- ;
    if (b[3]) a[2] -- , a[3] -- ;

	// 判断四边中间点
    for (int i = 0; i < 4; i ++ ) 
        if (a[i] < 0 || a[i] > n - 2)
            return false;
            
    return true;
}

int main() {
    int T;
    scanf("%d", &T);
    
    while (T -- ) {
        scanf("%d", &n);
        
        for (int i = 0; i < 4; i ++ ) scanf("%d", w + i);
        
        bool flag = true;
        for (int i = 0; i < 1 << 4; i ++ ) {
            if (check(i)) {
                puts("YES");
                flag = false;
                break;
            }
        }
        
        if (flag) puts("NO");
    }
    
    return 0;
}