jzoj3767【BJOI2014】路径 （记忆化搜索）

【BJOI2014】路径

Description

在一个N个节点的无向图（没有自环、重边）上，每个点都有一个符号，可能是数字，也可能是加号、减号、乘号、除号、小括号。你要在这个图上数一数，有多少种走恰好K个节点的方法，使得路过的符号串起来能够得到一个算数表达式 算数表达式。路径的起点和终点可以任意选择。

所谓算数表达式 算数表达式，就是由运算符连接起来的一系列数字。括号可以插入在表达式中以表明运算顺序。

注意，你要处理各种情况，比如数字不能有多余的前导0，减号只有前面没有运算符或数字的时候才可以当成负号，括号可以任意添加（但不能有空括号），0可以做除数（我们只考虑文法而不考虑语意），加号不能当正号。

例如，下面的是合法的表达式：

-0/ 0

((0)+(((2*3+4)+(-5)+7))+(-(2*3)*6))

而下面的不是合法的表达式：

001+0

1+2(2)

3+-3

–1

+1

()

Input

第一行三个整数N,M,K，表示点的数量，边的数量和走的节点数。

第二行一个字符串，表示每个点的符号。

接下来M行，每行两个数，表示一条边连的两个点的编号。

Output

输出一行一个整数，表示走的方法数。这个数可能比较大，你只需要输出它模1000000007的余数即可。

Sample Input

6 10 3

)(1*+0

1 2

1 3

1 4

2 3

3 4

2 5

3 5

3 6

4 6

5 6

Sample Output

10

Data Constraint

对于40%的数据，N,M,K≤10

对于100%的数据，1≤N≤20，0≤M≤N×(N-1)/ 2，0≤K≤30

分析：根据数据范围不难想到搜索，设f[i][j][k][0/1]表示选到第i位时选了j个选过k个左括号时是否有前导0的答案，分情况转移即可。

代码

#include <cstring>
#include <cstdio>
#include <string>
#define mo 1000000007
#define ll long long
#define N 1000
using namespace std;

struct arr
{
    int to,nxt;
}a[N*2];
ll f[100][100][100][2];
int n,m,k,l,ls[N*2];
char c[N],c1[N];

void add(int x,int y)
{
    a[++l].to=y;
    a[l].nxt=ls[x];
    ls[x]=l;
}

bool check1(char ch)
{
    if (ch>='0'&&ch<='9') return true;
    return false;
}

bool check2(char ch)
{
    if (ch!='+'&&ch!='-'&&ch!='*'&&ch!='/') return false;
    return true;
}

ll dfs(int x,int y,int p,int x0)
{
    if (y==k)
    {
        if (!p) return 1;
            else return 0;
    }
    if (f[x][y][p][x0]>-1) return f[x][y][p][x0];
    ll sum=0;

    for (int i=ls[x];i;i=a[i].nxt)
    {
        int t=a[i].to;
        if (check1(c[x]))
        {
            if ((check1(c[t])&&!x0)||(check2(c[t])&&k-y>1)) 
                sum=(sum+dfs(t,y+1,p,0))%mo;
            if (c[t]==')'&&p) sum=(sum+dfs(t,y+1,p-1,0))%mo;
        }
        if (check2(c[x]))
        {
            if (c[t]=='(') sum=(sum+dfs(t,y+1,p+1,0))%mo;
            if (check1(c[t])) if (c[t]=='0') sum=(sum+dfs(t,y+1,p,1))%mo;
                else sum=(sum+dfs(t,y+1,p,0))%mo;
        }
        if (c[x]=='(')
        {
            if (check1(c[t])) if (c[t]=='0') sum=(sum+dfs(t,y+1,p,1))%mo;
                else sum=(sum+dfs(t,y+1,p,0))%mo;
            if (c[t]=='(') sum=(sum+dfs(t,y+1,p+1,0))%mo;
            if (c[t]=='-'&&k-y>1) sum=(sum+dfs(t,y+1,p,0))%mo; 
        }
        if (c[x]==')')
        {
            if (c[t]==')'&&p) sum=(sum+dfs(t,y+1,p-1,0))%mo; 
            if (check2(c[t])&&k-y>1) sum=(sum+dfs(t,y+1,p,0))%mo;
        }
    }
    return f[x][y][p][x0]=sum;
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    getchar();
    scanf("%s",c+1);
    for (int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    for (int i=0;i<=99;i++)
        for (int j=0;j<=99;j++)
            for (int k=0;k<=99;k++)
                f[i][j][k][0]=-1,f[i][j][k][1]=-1;
    ll ans=0;
    for (int i=1;i<=n;i++)
    {
        if ((check1(c[i])&&c[i]!='0')||c[i]=='-') ans=(ans+dfs(i,1,0,0))%mo;
        if (c[i]=='0') ans=(ans+dfs(i,1,0,1))%mo;
        if (c[i]=='(') ans=(ans+dfs(i,1,1,0))%mo;
    }
    printf("%lld",ans);
}