【UNR #6】机器人表演

【UNR #6】机器人表演

by AmanoKumiko

Description

有一个长为/(n/)的/(01/)串，你需要计算/(t/)次操作后能得到多少不同的/(01/)串。

一次操作的定义为：在串中选两个位置插入一对/(01/)使得/(0/)在/(1/)前。

对/(998244353/)取模

Input

第一行两个数/(n,t/)

第二行一个长为/(n/)的/(01/)串

Output

一行一个整数表示答案

Sample Input

3 1
101

Sample Output

4

Data Constraint

/(1/le n/le 300,1/le t/le 300/)

Solution

感觉很自然但又很巧妙的一道题

首先有一个朴素的思想

设/(f_{i,j,k}/)表示填了/(i/)个，匹配了/(j/)个，插入的/(01/)串的和（即把/(0/)看做/(1/)，/(1/)看做/(-1/)）为/(k/)的方案数

然后匹配每次匹配最前面的

但是这样是假的

比如输入为/(1001/)，那么/(101001/)这种情况不会被记入

即当/(k/)为/(0/)时，再填/(1/)不一定不合法

我们考虑缩减匹配长度，思考缩减到什么样的位置是符合要求的

首先一定会剩下一个/(0/)来匹配新的/(1/)

然后剩下的字符可以看成是插入了一个合法的括号序中

那么此时新插入的也必须是一个合法括号序

枚举一下每次取最后的满足上面条件的位置就行了

Code

#include<bits/stdc++.h>
using namespace std;
#define F(i,a,b) for(int i=a;i<=b;i++)
#define Fd(i,a,b) for(int i=a;i>=b;i--)
#define mo 998244353
#define N 310

int f[N*3][N][N],n,t,len,to[N],a[N];
char s[N];

int mod(int x){return x>=mo?x-mo:x;}

void upd(int&x,int y){x=mod(x+y);}

bool pd(int l,int r){
	int sum=0;
	F(i,l,r){
		sum+=(a[i]==0?1:-1);
		if(sum<0)return 0;
	}
	return (sum==1);
}

int main(){
	scanf("%d%d",&n,&t);
	scanf("%s",s+1);
	F(i,1,n)a[i]=s[i]-'0';
	len=n+t*2;
	f[0][0][0]=1;
	memset(to,127,sizeof(to));
	F(i,1,n) Fd(j,i-1,0)if(pd(j+1,i)){to[i]=j;break;}
	F(i,0,len-1) F(j,0,n) F(k,0,t)if(f[i][j][k]){
		if(a[j+1]==0&&j+1<=n)upd(f[i+1][j+1][k],f[i][j][k]);
		else upd(f[i+1][j][k+1],f[i][j][k]);
		if(a[j+1]==1&&j+1<=n)upd(f[i+1][j+1][k],f[i][j][k]);
		else{
			if(k-1>=0)upd(f[i+1][j][k-1],f[i][j][k]);
			else if(to[j]<j)upd(f[i+1][to[j]][k],f[i][j][k]);
		}
	}
	printf("%d",f[len][n][0]);
	return 0;
}