[POJ 3046]Ant Counting[dp][优化]

题目链接：[POJ 3046]Ant Counting[dp][优化]
题意分析：
蚂蚁有T个家族，每个家族里的蚂蚁没有区别，总共有A只蚂蚁，现在问：给出每个家族的蚂蚁数量，最多能组合成多少个不同的集合个数为S~B的集合？
就是给你T个集合，每个集合$num[i]$个数，这些数排列组合能有多少个，个数为S~B的集合。
解题思路：
本题不能有重复情况，即{1,2}{2,1}是相同的，为了不重复，我们每一次处理一个集合。
设状态$dp[i][j]$为：前i个集合，最多组合成数量为j的集合的数量。
那么有

$$dp[i][j] = \sum_{k=0}^{min(j, num[i])}dp[i-1][j - k]$$ 新集合是由前面的元素的和转换而来的。
如果按照这个转移方程写的话，整个复杂度为 $O(TAnum)$整个三方的感觉。
注意到转移只是个对之前状态的求和，那么这个求和可以表达为 $$\sum_{k=0}^{min(j, num[i])}dp[i-1][j - k] = \sum_{k = 0}^{min(j - 1, num[i])}dp[i-1][j - 1 - k] +dp[i - 1][j] - dp[i - 1][j - 1 - num[i]]$$
简化后式子为： $dp[i][j] = dp[i][j - 1] + dp[i - 1][j] - dp[i - 1][j - 1 - num[i]]$
当 $num[i]>j-1$时后面一项不存在。
个人感受：
学到了这种求和结构还能这么优化= =。POJ数据比较水，三方也能过，平方就实力过题了。
具体代码如下：
三方代码：

#include<cstdio>
#include<iostream>
using namespace std;

const int INF = 0x7f7f7f7f;
const int MAXN = 1e5 + 111;
const int MOD = 1e6;

int dp[1011][MAXN];
int num[1011];

int main()
{
    int t, a, s, b, x;
    scanf("%d%d%d%d", &t, &a, &s, &b);
    for (int i = 0; i < a; ++i) {
        scanf("%d", &x);
        ++num[x];
    }

    dp[0][0] = 1;
    for (int i = 1; i <= t; ++i) {
        dp[i][0] = 1;
        for (int j = 1; j <= a; ++j) {
            int up = min(num[i], j);
            for (int k = 0; k <= up; ++k)
                dp[i][j] = (dp[i][j] + dp[i - 1][j - k]) % MOD;
        }
    }

    //for (int i = 1; i <= t; ++i, cout << '\n') for (int k = 0; k <= a; ++k) cout << dp[i][k] << ' ';

    int ans = 0;
    for (int i = s; i <= b; ++i) {
        ans = (ans + dp[t][i]) % MOD;
    }
    printf("%d\n", ans);
    return 0;
}

平方代码：

#include<cstdio>
#include<iostream>
using namespace std;

const int INF = 0x7f7f7f7f;
const int MAXN = 1e5 + 111;
const int MOD = 1e6;

int dp[1011][MAXN];
int num[1011];

int main()
{
    int t, a, s, b, x;
    scanf("%d%d%d%d", &t, &a, &s, &b);
    for (int i = 0; i < a; ++i) {
        scanf("%d", &x);
        ++num[x];
    }

    for (int i = 0; i <= t; ++i) dp[i][0] = 1;

    for (int i = 1; i <= t; ++i) {
        for (int j = 1; j <= a; ++j) {
            if (j - 1 - num[i] >= 0) {
                dp[i][j] = (dp[i][j - 1] + dp[i - 1][j] - dp[i - 1][j - 1 - num[i]] + MOD) % MOD;
            }
            else dp[i][j] = (dp[i][j - 1] + dp[i - 1][j]) % MOD;
        }
    }

    //for (int i = 1; i <= t; ++i, cout << '\n') for (int k = 0; k <= a; ++k) cout << dp[i][k] << ' ';

    int ans = 0;
    for (int i = s; i <= b; ++i) {
        ans = (ans + dp[t][i]) % MOD;
    }
    printf("%d\n", ans);
    return 0;
}