CodeForces 489 F.Special Matrices（组合数学+dp）

Description

一个$n\times n$的矩阵是特别的当且仅当其每个元素都是$0$或$1$且每行每列$1$的个数均为$2$，现在给出前$m$行，要求填后面的行使得这个矩阵的是特别的，问方案数

Input

第一行三个整数$n,m,mod$，之后输入$m\times n$的$01$矩阵，保证该矩阵每行两个$1$且每列$1$的个数不超过$2$$(2\le n\le 500,0\le m\le n,2\le mod\le 10^9)$

Output

输出方案数，结果模$mod$

Sample Input

3 1 1000
011

Sample Output

2

Solution

以$dp[x][y]$表示放好到当前行时，有$0$个$1$的列数为$x$，有$1$个$1$的列数为$y$的方案数，假设当前在放第$i$行，枚举前$i-1$行放好后没有$1$的列数$j$，由于前面$i-1$行已经放好，剩下$n-i+1$行必须有$2\cdot (n-i+1)$个$1$，但是这$j$列每列之后必须要再放两个$1$才行，故得到还需要放一个$1$的列数$k=2\cdot(n-i+1)-2\cdot j$，第$i$行必须放两个$1$，有三种情况

1.放在两个还需放两个$1$的列，即从$j$列从选两个放，有$dp[j-2][k+2]+=C_j^2\cdot dp[j][k]$

2.放在两个还需放一个$1$的列，即从$k$列中选两个放，有$dp[j][k-2]+=C_k^2\cdot dp[j][k]$

3.一个放在还需放两个$1$的列，另一个放在还需放一个$1$的列，即从$j$列中选一个从$k$列中选一个，有$dp[j-1][k]+=j\cdot k\cdot dp[j][k]$

放好$n$行后$dp[0][0]$即为答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=505;
int n,m,mod,dp[maxn][maxn];
char s[maxn][maxn]; 
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}
int main()
{
    scanf("%d%d%d",&n,&m,&mod);
    for(int i=1;i<=m;i++)scanf("%s",s[i]+1);
    int num0=0,num1=0;
    for(int j=1;j<=n;j++)
    {
        int num=0;
        for(int i=1;i<=m;i++)if(s[i][j]=='1')num++;
        if(num==0)num0++;
        else if(num==1)num1++;
    }
    dp[num0][num1]=1;
    for(int i=m+1;i<=n;i++)
        for(int j=0;j<=n;j++)
            {
                int k=2*(n-i+1)-2*j;
                if(k<0||k>n)continue;
                if(j>=2)add(dp[j-2][k+2],(ll)j*(j-1)/2*dp[j][k]%mod);
                if(k>=2)add(dp[j][k-2],(ll)k*(k-1)/2*dp[j][k]%mod);
                if(j&&k)add(dp[j-1][k],(ll)j*k*dp[j][k]%mod);
            }
    printf("%d\n",dp[0][0]);
    return 0;

}