CodeForces - 489F Special Matrices（DP）

最新推荐文章于 2018-01-09 20:56:14 发布

glq007

最新推荐文章于 2018-01-09 20:56:14 发布

阅读量1.3k

点赞数

分类专栏： DP 文章标签： dp codeforces

本文链接：https://blog.csdn.net/glqac/article/details/41246123

版权

DP 专栏收录该内容

66 篇文章 0 订阅

订阅专栏

第一次做CF，D题好多人过没看直接做F了。结果想了好久，后来过了还是挺开心的，最后过了4题rating到1715，紫了，虽说估计以后会掉。。但还是蛮开心的！

题意很简单就不说了。

直接说做法，我比赛时是o(n^3)过的，跑了460ms，先说这个，dp[x][i][j]代表填到x行，列上总的1为0的列数为i，列上总的1为2的列数为j，的方案数。因为太大开不下，所以可以用滚动数组去优化。

递推的话分3种情况，1. i>=2, dp[x+1][i-2][j+2] += dp[i][j]*i*(i-1)/2，2. j >= 2 dp[x+1][i][j-2] += dp[i][j] * j * (j-1)/2。

3. i >= 1 && j >= 1，dp[x+1][i-1][j] += dp[i][j] * i * j。

其实可以优化成n^2的，首先你会发觉有大量状态被浪费，因为i只能变小，j可能变大或者变小，但是如果i到0了，那么j只能变小了。（因为最后都要变成列上为2的）。

所以可以逆向递推。不用管行数，最后到0，0这个状态就说明到最后一行了。优化后只需要15ms了。

AC代码：

//#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<ctype.h>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
#include<cstdlib>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
#include<ctime>
#include<string.h>
#include<string>
#include<sstream>
#include<bitset>
using namespace std;
#define ll __int64
#define ull unsigned long long
#define eps 1e-11
#define NMAX 200005
#define MOD 1000000007
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI acos(-1)
template<class T>
inline void scan_d(T &ret)
{
    char c;
    int flag = 0;
    ret=0;
    while(((c=getchar())<'0'||c>'9')&&c!='-');
    if(c == '-')
    {
        flag = 1;
        c = getchar();
    }
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
    if(flag) ret = -ret;
}
ll dp[505][505];
int cal[505];

int main()
{
#ifdef GLQ
    freopen("input.txt","r",stdin);
//    freopen("o4.txt","w",stdout);
#endif // GLQ
    int n,m;
    ll mod;
    scanf("%d%d%I64d",&n,&m,&mod);
    char tmp[505];
    for(int i = 1; i <= m; i++)
    {
        scanf("%s",tmp);
        for(int j = 0; j < n; j++) if(tmp[j] == '1')
                cal[j]++;
    }
    int t0=0,t1=0;
    for(int i = 0; i < n; i++)
    {
        if(cal[i] == 0) t0++;
        if(cal[i] == 1) t1++;
    }
    dp[t0][t1] = 1;
    for(int i = t0; i >= 0; i--)
        for(int j = n; j >= 0; j--) if(dp[i][j] != 0)
        {
            ll d = dp[i][j];
            if(i >= 2) dp[i-2][j+2] = (dp[i-2][j+2]+(d*i*(i-1)/2LL)%mod)%mod;
            if(i >= 1 && j >= 1) dp[i-1][j] = (dp[i-1][j]+((d*(ll)i)%mod*(ll)j)%mod)%mod;
            if(j >= 2) dp[i][j-2] = (dp[i][j-2]+(d*j*(j-1)/2LL)%mod)%mod;
        }
    printf("%I64d\n",dp[0][0]);
    return 0;
}