题意
给出一个初始序列
ai
和一个最终序列
bi
,要求用一个栈执行两种操作
1 从
ai
的前面取出一个数字压到栈中
2 弹栈,将栈顶放到
bi
的后面。给出的最终序列中有一些位置是位置的数字,询问有多少种方法构造出
bi
来。
长度至多16
题解
将
ai
中数字是否在栈中状压,状态表示f[压栈次数][弹栈次数][状态]
然后dp
代码
#include <cstdio>
#define maxn 16
#define maxs (1<<16)
#define mod 1000000007LL
int n, a[maxn+1], b[maxn+1], f[maxn+1][maxn+1][maxs] = {0};
/// f[in][out][exist]
int main() {
int i, j, s, k;
scanf("%d", &n);
for (i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
for (i = 1; i <= n; i ++ ) scanf("%d", &b[i]);
/// build array b from array a !
f[0][0][0] = 1;
for (i = 0; i <= n; i ++ )
for (j = 0; j <= i; j ++ )
for (s = (1<<i)-1; s >= 0; s -- ) {
if (!f[i][j][s]) continue;
if (i < n) (f[i+1][j][s|(1<<i)] += f[i][j][s]) %= mod;
if (!s) continue;
for (k = i; !((1<<k-1)&s); k -- );
if (b[j+1]!=-1 && a[k]!=b[j+1]) continue;
(f[i][j+1][s^(1<<k-1)] += f[i][j][s]) %= mod;
}
printf("%d\n", f[n][n][0]);
//while(1);
return 0;
}