表达式得到期望结果的组合种数
题目描述
给定一个只由0(假)、1(真)、&(逻辑与)、|(逻辑或)和^(异或)五种字符组成的字符串express,再给定一个布尔值desired。求出express能有多少种组合方式,可以达到desired的结果。并输出你所求出的总方案数对 1 0 9 + 7 10^{9}+7 109+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^{3}) O(n3),空间复杂度 O ( n 2 ) O(n^2) O(n2)。
题解:
此题很明显需要以运算符为分界点,分别计算左半部分和右半部分的值,然后得到整体的值。我们可以用两个二维数组 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;
}