表达式得到期望结果的组合种数

表达式得到期望结果的组合种数

题目描述

给定一个只由0（假）、1（真）、&（逻辑与）、|（逻辑或）和^（异或）五种字符组成的字符串express，再给定一个布尔值desired。求出express能有多少种组合方式，可以达到desired的结果。并输出你所求出的总方案数对$10^9+7$取模后的值。

输入描述：

输入两行，第一行包含一个只有0、1、&、|和^组成的字符串。其长度小于500，第二行只有一个布尔值，代表desired。

输出描述：

输出一个整数，表示取模后的答案。

示例1

输入

1^0|0|1
false

输出

2

说明

1^((0|0)|1)和1^(0|(0|1))可以得到false

示例2

输入

1
false

输出

0

备注：

时间复杂度$O(n^3)$,空间复杂度$O(n^2)$。

---

题解：

此题很明显需要以运算符为分界点，分别计算左半部分和右半部分的值，然后得到整体的值。我们可以用两个二维数组 f 和 t 分别记录 express[j…i] 为 true 和 false 的方案数，最后根据运算符合并即可。（非常像区间 dp）

代码：

#include <cstdio>
#include <cstring>

using namespace std;

const int N = 510;
const int MOD = 1e9 + 7;

char str[N];
char desired[6];
long t[N][N];
long f[N][N];
int len;

bool valid() {
    if (!(len & 1)) return false;
    for (int i = 0; i < len; ++i) {
        if (i & 1) {
            if (str[i] != '^' && str[i] != '&' && str[i] != '|')
                return false;
        } else {
            if (str[i] != '0' && str[i] != '1')
                return false;
        }
    }
    return true;
}

int main(void) {
    scanf("%s", str);
    scanf("%s", desired);
    len = strlen(str);
    if (!valid()) return 0 * puts("0");
    for (int i = 0; i < len; i += 2) {
        f[i][i] = (str[i] == '0' ? 1 : 0);
        t[i][i] = (str[i] == '1' ? 1 : 0);
    }
    for (int i = 2; i < len; i += 2) {
        for (int j = i - 2; j >= 0; j -= 2) {
            for (int k = j; k < i; k += 2) {
                if (str[k + 1] == '&') {
                    (t[j][i] += t[j][k] * t[k + 2][i]) %= MOD;
                    (f[j][i] += (f[j][k] + t[j][k]) * f[k + 2][i] + f[j][k] * t[k + 2][i]) %= MOD;
                } else if (str[k + 1] == '|') {
                    (f[j][i] += f[j][k] * f[k + 2][i]) %= MOD;
                    (t[j][i] += (f[j][k] + t[j][k]) * t[k + 2][i] + t[j][k] * f[k + 2][i]) %= MOD;
                } else {
                    (t[j][i] += f[j][k] * t[k + 2][i] + t[j][k] * f[k + 2][i]) %= MOD;
                    (f[j][i] += f[j][k] * f[k + 2][i] + t[j][k] * t[k + 2][i]) %= MOD;
                }
            }
        }
    }
    printf("%ld\n", desired[0] == 'f' ? f[0][len - 1] : t[0][len - 1]);
    return 0;
}