字节跳动0825笔试场第二题合法表达式（动态规划）

 很明显是一道动态规划的题，刚开始天真想用一维dp[n]来解决，结果发现想的太简单，比如(40)4就是不合法的，所以要用二维动态规划分情况来考虑，dp[i][0]表示长度为i且最后一位为数字（0-9）的个数，dp[i][1]表示长度为i且最后一位为右括号的个数。状态转移方程为:

dp[i][0]=(dp[i-1][0]*10+(dp[i-2][0]+dp[i-2][1])*10*2LL)//倒数第二位是“+”或“-”

当最后一位为右括号时，它在前面必有对应的左括号，当左括号在第一个位置时:

dp[i][1]=(dp[i-2][0]+dp[i-2][1]);

当左括号不在第1和第2个位置时（左括号如果在第2个位置则无论第1个位置是什么都不合法），左括号前面只能是“+”或者“-”，然后移动“+（”或“-（”的位置，得到dp[i][1]总个数为:

for(int j=1;i-j-3>=1;j++)//减3是减去“+（）”3个字符的长度，j是左边合法表达式的长度，i-j-3是右边合法表达式的长度
            dp[i][1]=(dp[i][1]+((dp[j][0]+dp[j][1])*(dp[i-j-3][0]+dp[i-j-3][1])*2LL)%mod)%mod;

代码如下:(仅以此文记录自己笔试时脑残将i-j-3写成了n-j-3，一度怀疑自己是dp转移错了)

#include <iostream>
#include<stdio.h>
#include<cstring>
using namespace std;
long long dp[1000+10][2];
const long long mod=1000000007;
int main()
{
    int n;
    cin>>n;
    memset(dp,0,sizeof(dp));
    dp[1][0]=10;
    dp[1][1]=0;
    for(int i=2;i<=n;i++)
       {
           dp[i][0]=(dp[i-1][0]*10+(dp[i-2][0]+dp[i-2][1])*20LL)%mod;
           dp[i][1]=(dp[i-2][0]+dp[i-2][1])%mod;
           for(int j=1;i-j-3>=1;j++)
            dp[i][1]=(dp[i][1]+((dp[j][0]+dp[j][1])*(dp[i-j-3][0]+dp[i-j-3][1])*2LL)%mod)%mod;
       }
    cout<<(dp[i][0]+dp[i][1])%mod<<endl;
}