码迷,mamicode.com
首页 > 其他好文 > 详细

hihocoder 1124 : 好矩阵 dp

时间:2018-02-14 13:52:15      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:pos   main   ati   内存   data-   title   article   amp   uri   

好矩阵

时间限制:3000ms
单点时限:1000ms
内存限制:256MB

描写叙述

给定n, m。一个n?×?m矩阵是好矩阵当且仅当它的每一个位置都是非负整数,且每行每列的和?≤?2。求好矩阵的个数。模109?+?7

输入

第一行一个整数T,表示測试点个数。

以下T个測试点。

每一个測试点一行。包括两个整数n。m。

1?≤?T?≤?104. 1?≤?n,?m?≤?100.

输出

T行。

每行是相应測试点的答案。

例子输入
1
2 2
例子输出
26

题意非常easy。因为,数量非常大。假设考虑一个一个方格的放,要考虑横向的,又要考虑竖向的,非常复杂,所以不可取。所以一排一排的放,假设,m是一定的,那么。每一排仅仅须要考虑不超过2,且与前面已经排好的不冲突就能够了。

dp[i][a][b]表示,第i排,有a列0,b列1,m - a - b列2,的个数则

dp[i+1][a][b]+= dp[i][a][b];//第i+1排全放0
dp[i+1][a-1][b] += (ll)a * dp[i][a][b];//第i+1排在和为0那些列放一个2
dp[i+1][a-1][b+1] += (ll)a * dp[i][a][b];//第i+1排和为1放一个1
dp[i+1][a-1][b]+= (ll)a * (ll) b * dp[i][a][b];//第i+1排和为0 和为1的列各选一个 放两个1
dp[i+1][a][b-1] += (ll)b * dp[i][a][b];//第i+1排选一个和为1的列放一个1
dp[i+1][a-2][b+2] += (ll)(a * (a-1)/2) * dp[i][a][b];//第i+1排选两个和为0放两个1
dp[i+1][a][b-2] += (ll)(b * (b-1)/2) * dp[i][a][b];//第i+1排选两个和为1的放两个1

总复杂度为o(n^4);

#define N 105
#define M 100005
#define maxn 205
#define MOD 1000000007
int n,m,T;
ll dp[N][N][N],ans[N][N];
int main()
{
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    for(int m = 1;m<=100;m++){
        int n = 100;
        for(int i =0;i<=n;i++){
            for(int a = 0;a<=m;a++){
                for(int b = 0;b<=m;b++){
                    dp[i][a][b] = 0;
                }
            }
        }
        dp[0][m][0] = 1;
        for(int i = 0;i<n;i++){
            for(int a = 0;a<=m;a++){
                for(int b = 0;a + b<=m;b++){
                    dp[i+1][a][b]+= dp[i][a][b];
                    dp[i+1][a][b] %= MOD;
                    if(a >= 1){
                        dp[i+1][a-1][b] += (ll)a * dp[i][a][b];
                        dp[i+1][a-1][b] %= MOD;
                        dp[i+1][a-1][b+1] += (ll)a * dp[i][a][b];
                        dp[i+1][a-1][b+1] %= MOD;
                    }
                    if(a >= 1 && b >= 1){
                        dp[i+1][a-1][b]+= (ll)a * (ll) b * dp[i][a][b];
                        dp[i+1][a-1][b] %= MOD;
                    }
                    if(b >= 1 ){
                        dp[i+1][a][b-1] += (ll)b * dp[i][a][b];
                        dp[i+1][a][b-1] %= MOD;
                    }
                    if(a >= 2 ){
                        dp[i+1][a-2][b+2] += (ll)(a * (a-1)/2) * dp[i][a][b];
                        dp[i+1][a-2][b+2] %= MOD;
                    }
                    if(b >= 2 ){
                        dp[i+1][a][b-2] += (ll)(b * (b-1)/2) * dp[i][a][b];
                        dp[i+1][a][b-2] %= MOD;
                    }
                }
            }
            ans[i+1][m] = 0;
            for(int a = 0;a<=m;a++){
                for(int b = 0;b<=m;b++){
                    ans[i+1][m] += dp[i+1][a][b];
                    ans[i+1][m] %= MOD;
                }
            }
        }
    }
    while(S(T)!=EOF)
    {
        while(T--){
            int s,e;
            S2(s,e);
            printf("%lld\n",ans[s][e]);
        }
    }
    return 0;
}

hihocoder 1124 : 好矩阵 dp

标签:pos   main   ati   内存   data-   title   article   amp   uri   

原文地址:https://www.cnblogs.com/zhchoutai/p/8448223.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!