poj 3046 数蚂蚁 dp

题意：

蚂蚁们有T个家庭，共有A只蚂蚁。

用只数来分组，问从s只到b只总共有多少中分配方案。

解析：

用一维的想法一直想，真是糊涂了。

dp[ i ] [ j ] 表示 前 i 个家庭，凑成每j只蚂蚁一组总共有多少种分法。

状态转移方程：

dp[ i ] [ j ] =  sigema( dp[ i - 1 ] [ j - k ] )，其中k从0到family[ i ]。

状态转移方程的意思是，前 i - 1 个家族配出 j - k 个的组合，再在当前 i 家族出 k 个的个数。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#include <vector>
#include <queue>
#include <map>
#include <climits>
#include <cassert>
#define LL long long
#define lson lo, mi, rt << 1
#define rson mi + 1, hi, rt << 1 | 1

using namespace std;
const int mod = 1000000;
const int maxn = 10000 + 10;
const int maxt = 1000 + 10;
const int inf = 0x3f3f3f3f;
const double eps = 1e-8;
const double pi = acos(-1.0);
const double ee = exp(1.0);

int family[maxt];
int dp[maxt][maxn];

int main()
{
    #ifdef LOCAL
    freopen("in.txt", "r", stdin);
    #endif // LOCAL
    int T, A, S, B;
    while (~scanf("%d%d%d%d", &T, &A, &S, &B))
    {
        memset(family, 0, sizeof(family));
        for (int i = 0; i < A; i++)
        {
            int x;
            scanf("%d", &x);
            family[x]++;
        }
        memset(dp, 0, sizeof(dp));
        dp[0][0] = 1;
        int sum = 0;
        for (int i = 1; i <= T; i++)
        {
            sum += family[i];
            for (int k = 0; k <= family[i]; k++)
            {
                for (int j = sum; j >= k; j--)
                {
                    dp[i][j] = (dp[i][j] + dp[i - 1][j - k]) % mod;
                }
            }
        }
        int ans = 0;
        for (int i = S; i <= B; i++)
        {
            ans = (ans + dp[T][i]) % mod;
        }
        printf("%d\n", ans);
    }
    return 0;
}