51nod 1934 受限制的排列——笛卡尔树

题目：http://www.51nod.com/Challenge/Problem.html#!#problemId=1934

根据给出的信息，可以递归地把笛卡尔树建出来。一个点只应该有 0/1/2 个孩子，不然就是无解。

dp[ cr ] 表示把 1~siz[cr] 填进 cr 这个子树的方案数。那么 \( dp[cr]=C_{siz[cr]-1}^{siz[ls]}*dp[ls]*dp[rs] \) 。

注意在各种地方判断无解！如果是 l , cr , r 的话，左孩子应该是 l , ls , cr-1 ，右孩子应该是 cr+1 , rs , r 这样的。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#include<vector>
#define ls Ls[cr]
#define rs Rs[cr]
#define pb push_back
#define ll long long
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}

const int N=1e6+,mod=1e9+;
int pw(int x,int k)
{int ret=;while(k){if(k&)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=;}return ret;}

int n,rt,l[N],Ls[N],Rs[N],siz[N],dp[N],jc[N],jcn[N]; bool flag;
struct Node{
  int r,p;
  Node(int r=,int p=):r(r),p(p) {}
  bool operator< (const Node &b)const
  {return r<b.r;}
};
vector<Node> st[N];
vector<int> vt[N];
void init()
{
  int n=1e6;
  jc[]=;for(int i=;i<=n;i++)jc[i]=(ll)jc[i-]*i%mod;
  jcn[n]=pw(jc[n],mod-);
  for(int i=n-;i>=;i--)jcn[i]=(ll)jcn[i+]*(i+)%mod;
}
int C(int n,int m)
{return (ll)jc[n]*jcn[m]%mod*jcn[n-m]%mod;}
void solve(int L,int R,int cr)
{
  int tl=L,v,tr;
  vt[cr].clear();//
  if(cr>L)
    {
      if(!l[L]){flag=;return;}
      Node v=st[L][--l[L]]; int bh=v.p;
      if(v.r!=cr-){flag=;return;}
      solve(L,cr-,bh); if(flag)return;
      ls=bh;
    }
  else ls=;
  if(cr<R)
    {
      if(!l[cr+]){flag=;return;}
      Node v=st[cr+][--l[cr+]]; int bh=v.p;
      if(v.r!=R){flag=;return;}
      solve(cr+,R,bh); if(flag)return;
      rs=bh;
    }
  else rs=;
  siz[cr]=siz[ls]+siz[rs]+;
  dp[cr]=(ll)C(siz[cr]-,siz[ls])*dp[ls]%mod*dp[rs]%mod;
}
int main()
{
  int T=; init(); dp[]=;//
  while(scanf("%d",&n)==)
    {
      for(int i=;i<=n;i++)st[i].clear();//
      for(int i=;i<=n;i++)l[i]=rdn();
      for(int i=,r;i<=n;i++)
    {
      r=rdn();st[l[i]].pb(Node(r,i));
    }
      for(int i=;i<=n;i++)
    {
      sort(st[i].begin(),st[i].end());
      l[i]=st[i].size();
    }
      T++; printf("Case #%d: ",T);
      if(!l[])puts("");
      else
    {
      Node rt=st[][--l[]]; int bh=rt.p;
      if(rt.r!=n)puts("");
      else {flag=; solve(,n,bh); printf("%d\n",flag?:dp[bh]);}
    }
    }
  return ;
}