DFS暴力枚举解决逻辑推理问题

埃琳娜莱农

1. 问题背景与理解

这道题目描述了一个有趣的逻辑推理游戏。Farmer John在n个槽中隐藏了食物，Bessie通过提出m个特定问题来试图找出哪些槽中有食物。每个问题都询问某些特定槽中食物的总数，我们需要根据这些问题的答案来推断出实际的槽状态。

问题的核心在于：给定m个约束条件（每个条件指定了某些槽中食物的数量），找出所有可能的槽状态组合（每个槽有食物或无食物），并判断解的唯一性。

2. 解题思路分析

2.1 暴力枚举的可行性

由于n的范围很小（1≤n≤20），我们可以考虑暴力枚举所有可能的槽状态组合。对于每个槽，它要么有食物（1），要么没有（0），所以总共有2^n种可能的组合。

对于n=20的情况，2^20=1,048,576种组合，这在现代计算机的处理能力下是完全可行的。因此，DFS（深度优先搜索）是一个合理的选择。

2.2 验证逻辑的设计

对于每个生成的槽状态组合，我们需要验证它是否满足所有的m个约束条件。具体来说：

对于每个问题（即每个约束条件），我们需要检查该组合中对应槽位的食物数量是否等于问题中给出的t值。
如果所有约束条件都满足，则该组合是一个有效解。
我们需要统计所有有效解的数量，以判断输出结果（唯一解、多解或无解）。

3. 代码实现详解

3.1 数据结构设计

cpp复制int n, m, tot, a[1500001], sum;
string ans[1500001], s[100];

n：槽的数量
m：问题的数量
tot：记录找到的有效解的数量
a[]：存储每个问题对应的t值（即对应槽中应有的食物数量）
s[]：存储每个问题的01序列（表示询问了哪些槽）
ans[]：存储找到的有效解

3.2 验证函数check

cpp复制bool check(string ss) {
    for(int i=1; i<=m; i++) {
        int cnt = 0;
        for(int j=0; j<n; j++)
            if((ss[j]=='1') && (s[i][j]=='1'))
                cnt++;
        if(cnt != a[i])
            return false;
    }
    return true;
}

这个函数接收一个槽状态组合ss，然后检查它是否满足所有约束条件：

对于每个问题i，计算被询问的槽中实际有食物的数量cnt
比较cnt与问题中给出的a[i]是否一致
如果所有问题都满足，返回true；否则返回false

3.3 DFS函数

cpp复制void dfs(int x, string s) {
    if(x >= n+1) {
        if(check(s)) {
            if(tot == 1) {
                cout << "NOT UNIQUE";
                exit(0);
            }
            ans[++tot] = s;
        }
        return;
    }
    dfs(x+1, s+'0');
    dfs(x+1, s+'1');
}

这个递归函数用于生成所有可能的槽状态组合：

参数x表示当前处理到第几个槽
参数s表示当前已生成的槽状态前缀
当处理完所有槽（x≥n+1）时，检查当前组合是否有效
如果找到第二个有效解，立即输出"NOT UNIQUE"并终止程序
否则继续递归生成下一个槽的两种可能状态（0或1）

3.4 主函数

cpp复制int main() {
    cin >> n >> m;
    for(int i=1; i<=m; i++)
        cin >> s[i] >> a[i];
    dfs(1, "");
    if(tot == 0)
        cout << "IMPOSSIBLE";
    else
        for(int i=0; i<n; i++)
            cout << ans[1][i];
    return 0;
}