JZOJ 5384. 【NOIP2017提高A组模拟9.23】四维世界

Description

众所周知，我们常感受的世界是三维的。
Polycarp突然对四维空间产生了兴趣，他想对四维空间进行一些研究。但是在此之前，他必须先对三维世界了解透彻。
于是Polycarp决定从零维，也就是一个点，开始他的研究。我们把一个点放在三维空间中，Polycarp把这个点视为原点，并确定了三个正方向。他可以把这个点往三个方向之一拉伸一个单位，那么这个点就变为了一维的一条长度为一的线段。然后如果他把这条线段往另一方向拉伸一个单位，那么这条线就变为了二维的一个矩形。如果继续拉伸可能就会进入三维世界，也就是变为直四棱柱。
Polycarp认为矩形、线段甚至点都可以看作某一维或某几维为丰的直四棱柱。
现在Polycarp想演示把一个点一步一步拉伸为边长为n的正六面体的过程，但他缺失了m种形态的直四棱柱模具（Polycarp拥有其他的所有直四棱柱模具），他想知道共有多少种演示方案。
Polycarp的演示过程需要每拉伸一个单位时对应形态的直四棱柱。
因为方案数很大，所以输出答案对10^9+7的结果。

Input

从文件poly.in中读入数据。
第一行两个整数n;m，分别表示直四棱柱的边长和他缺失的模具数量。
接下来m行，第i行三个整数x; y; z，表示第i个缺失模具的长、宽、高。

Output

输出到文件poly.out中
一个整数，即答案。

Sample Input

2 3
1 0 1
1 1 1
0 2 0

Sample Output

36

Data Constraint

Solution

• 先考虑 $m=0$ 的情况，即计算 $(0,0,0)$ 走到 $(n,n,n)$ 的方案数。

• 那么显然答案即为：$C_{3n}^{n}*C_{2n}^{n}$ ，即从 $3n$ 步中选 $n$ 步走 X ，在从剩下 $2n$ 步中选 $n$ 步走 Y。

• 现在加入很多的障碍点，于是我们针对这些点进行递推。

• 设 $F[i]$ 表示经过第 $i$ 个障碍点且不经过其他障碍点的方案数。

• 我们发现如果可以经过其他障碍点时答案就是：

  $$F[i]=\frac{(x_i+y_i+z_i)!}{x_i!*y_i!*z_i!}$$

• 同理可以得出：

  $$F[i]=\frac{(x_i+y_i+z_i)!}{x_i!*y_i!*z_i!}-\sum_{x_j\leq x_i,y_j\leq y_i,z_j\leq z_i}F[j]*\frac{(x_i-x_j+y_i-y_j+z_i-z_j)!}{(x_i-x_j)!*(y_i-y_j)!*(z_i-z_j)!}$$

• 于是我们再加入 $(n,n,n)$ 这个点作为“第 $m+1$ 障碍点”进行处理，则答案就是 $F[m+1]$ 。

• 时间复杂度 $O(M^2)$ 。

Code

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100001,M=5002,mo=1e9+7;
struct data
{
    int x,y,z;
}a[M];
long long f[M],g[N*3],h[N*3];
inline int read()
{
    int X=0,w=1; char ch=0;
    while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
    return X*w;
}
inline long long ksm(long long x,int y)
{
    long long s=1;
    while(y)
    {
        if(y&1) s=s*x%mo;
        x=x*x%mo;
        y>>=1;
    }
    return s;
}
inline bool cmp(data x,data y)
{
    return x.x<y.x || x.x==y.x && x.y<y.y || x.x==y.x && x.y==y.y && x.z<y.z;
}
int main()
{
    int n=read(),m=read();
    g[0]=h[0]=1;
    for(int i=1;i<=n*3;i++) g[i]=g[i-1]*i%mo;
    h[n]=ksm(g[n],mo-2);
    for(int i=n-1;i;i--) h[i]=h[i+1]*(i+1)%mo;
    if(!m) 
    {
        printf("%lld",g[n*3]*ksm(ksm(g[n],3),mo-2)%mo);
        return 0;
    }
    for(int i=1;i<=m;i++) a[i].x=read(),a[i].y=read(),a[i].z=read();
    a[m].x=a[m].y=a[++m].z=n;
    sort(a+1,a+1+m,cmp);
    for(int i=1;i<=m;i++)
    {
        f[i]=g[a[i].x+a[i].y+a[i].z]*h[a[i].x]%mo*h[a[i].y]%mo*h[a[i].z]%mo;
        for(int j=1;j<i;j++)
            if(a[j].x<=a[i].x && a[j].y<=a[i].y && a[j].z<=a[i].z)
                f[i]=(f[i]+mo-f[j]*g[a[i].x-a[j].x+a[i].y-a[j].y+a[i].z-a[j].z]%mo*
                    h[a[i].x-a[j].x]%mo*h[a[i].y-a[j].y]%mo*h[a[i].z-a[j].z]%mo)%mo;
    }
    printf("%lld",f[m]);
    return 0;
}