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
Hint
【样例解释】
一共有10条路径,构成的表达式依次是101, (1), 1+1, 1+0, 1*1, 1*0, 0+0, 0+1, 0*0, 0*1
思路
神奇的记忆化搜索。
设f[x][k][p][t]表示走到点x,用的步数为k,括号内有c个数,是否有前导0。
初始时只可能从数字,括号,负号出发。
然后枚举所有情况转移即可。
代码
#include<iostream>
#include<cstring>
#include<cstdio>
#define mo 1000000007
using namespace std;
int f[25][35][35][2],a[50][50],n,m,x,y,ans,num;
char c[30];
bool fuhao(int x){return c[x]=='+'||c[x]=='-'||c[x]=='*'||c[x]=='/';}
bool shuzi(int x){return c[x]>='0'&&c[x]<='9';}
int dp(int x,int k,int p,int t)
{
if (k==num) return ((p==0)&&(!fuhao(x)));
int now=f[x][k][p][t];
if (now>=0) return now;
now=0;
for (int i=1;i<=n;i++)
if (a[x][i])
{
if (shuzi(i))
{
if (shuzi(x))
{
if (!t) (now+=dp(i,k+1,p,0))%=mo;
}
else if (fuhao(x)||c[x]=='(') (now+=dp(i,k+1,p,c[i]=='0'))%=mo;
}
else if (c[i]=='('||c[i]==')'){
if (c[i]=='('&&(c[x]=='('||fuhao(x))) (now+=dp(i,k+1,p+1,0))%=mo;
else if (c[i]==')'&&p&&(shuzi(x)||c[x]==')')) (now+=dp(i,k+1,p-1,0))%=mo;
}
else
{
if (shuzi(x)) (now+=dp(i,k+1,p,0))%=mo;
if (c[x]==')'||(c[x]=='('&&c[i]=='-')) (now+=dp(i,k+1,p,0))%=mo;
}
}
f[x][k][p][t]=now;
return now;
}
int main()
{
scanf("%d%d%d",&n,&m,&num);
scanf("%s",c+1);
for (int i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
a[x][y]=a[y][x]=1;
}
memset(f,-1,sizeof(f));
for (int i=1;i<=n;i++)
{
if (c[i]=='(') (ans+=dp(i,1,1,0))%=mo;
if (c[i]=='-') (ans+=dp(i,1,0,0))%=mo;
if (c[i]>='0'&&c[i]<='9')(ans+=dp(i,1,0,c[i]=='0'))%=mo;
}
printf("%d",ans);
}