题目大意
对
n
个事件有
将所有约束视为无向边,满足
n
个事件在一个联通块内。
题目分析
一眼看去,题目条件和树有关,但是又有边的方向的限制,一下子好像变得很棘手。这是容易将题目看做拓扑图来解决,无异于给自己添麻烦。
其实我们将其看成一棵树,做树形dp,对于不同方向的边不同处理就行了。
约定:
fx,i
表示节点
x
在以
size(x)
为以
x
为根的子树的节点个数。
对于节点
现在我们要合并
x
的部分和
对于
x
先于
设
i
表示
于是我们可以列出递推方程:
对于 x 后于
设 i 表示
同样可以列出递推方程:
然后我们就可以将 g 赋值给
然后最终答案即是 preroot,n 或 sufroot,1 。
时间复杂度理论是 O(n3) ,但是实际上枚举的状态达不到那么多,于是就飞起了。
代码实现
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cctype>
using namespace std;
int read()
{
int x=0,f=1;
char ch=getchar();
while (!isdigit(ch))
{
if (ch=='-')
f=-1;
ch=getchar();
}
while (isdigit(ch))
{
x=x*10+ch-'0';
ch=getchar();
}
return x*f;
}
const int N=1000;
const int M=N-1;
const int E=M<<1;
const int P=1000000007;
long long f[N+1][N+1],pre[N+1][N+1],suf[N+1][N+1],g[N+1],fact[N+1],invf[N+1];
int size[N+1],last[N+1];
int tov[E+1],next[E+1];
int tot,n,m,ans,T;
bool cst[E+1];
void insert(int x,int y,bool z)
{
tov[++tot]=y;
cst[tot]=z;
next[tot]=last[x];
last[x]=tot;
}
long long quick_power(int x,int y)
{
if (!y)
return 1;
if (y==1)
return x;
long long ret=quick_power(x,y>>1);
ret=(long long)ret*ret%P;
if (y&1)
ret=(long long)ret*x%P;
return ret;
}
void preparation()
{
fact[0]=1;
for (int i=1;i<=N;i++)
fact[i]=(long long)fact[i-1]*i%P;
invf[N]=quick_power(fact[N],P-2);
for (int i=N-1;i>=0;i--)
invf[i]=(long long)invf[i+1]*(i+1)%P;
}
long long combination(int n,int m)
{
return (long long)fact[n]*invf[m]%P*invf[n-m]%P;
}
void dfs(int x,int fa)
{
size[x]=1;
int i=last[x],y;
bool found=false;
while (i)
{
y=tov[i];
if (y!=fa)
{
dfs(y,x);
size[x]+=size[y];
found=true;
}
i=next[i];
}
if (!found)
{
f[x][1]=1;
pre[x][1]=f[x][1];
for (i=2;i<=n;i++)
pre[x][i]=(pre[x][i-1]+f[x][i])%P;
suf[x][n]=f[x][n];
for (i=n-1;i>=1;i--)
suf[x][i]=(suf[x][i+1]+f[x][i])%P;
return;
}
int s=1;
for (int i=1;i<=n;i++)
f[x][i]=0;
f[x][1]=1;
i=last[x];
while (i)
{
y=tov[i];
if (y!=fa)
{
for (int k=1;k<=s+size[y];k++)
g[k]=0;
for (int k=1;k<=s;k++)
for (int l=1;l<=size[y];l++)
if (cst[i])
(g[k+l-1]+=(long long)combination(k+l-2,k-1)*combination(s-k+size[y]-l+1,s-k)%P*f[x][k]%P*suf[y][l]%P)%=P;
else
(g[k+l]+=(long long)combination(k+l-1,k-1)*combination(s-k+size[y]-l,s-k)%P*f[x][k]%P*pre[y][l]%P)%=P;
for (int k=1;k<=s+size[y];k++)
f[x][k]=g[k];
s+=size[y];
}
i=next[i];
}
pre[x][1]=f[x][1];
for (i=2;i<=n;i++)
pre[x][i]=(pre[x][i-1]+f[x][i])%P;
suf[x][n]=f[x][n];
for (i=n-1;i>=1;i--)
suf[x][i]=(suf[x][i+1]+f[x][i])%P;
}
int main()
{
preparation();
freopen("sao.in","r",stdin);
freopen("sao.out","w",stdout);
T=read();
while (T--)
{
memset(last,0,sizeof last);
memset(next,0,sizeof next);
memset(f,0,sizeof f);
tot=0;
n=read();
for (int i=1,x,y;i<n;i++)
{
x=read()+1;
char sign=getchar();
while (sign!='<'&&sign!='>')
sign=getchar();
y=read()+1;
insert(x,y,sign=='<');
insert(y,x,sign!='<');
}
dfs(1,0);
ans=0;
for (int i=1;i<=n;i++)
(ans+=f[1][i])%=P;
printf("%d\n",ans);
}
fclose(stdin);
fclose(stdout);
return 0;
}