POJ3046 Ant Counting

有T种蚂蚁，编号为1-T。每一种蚂蚁有Ni只蚂蚁。现在从蚂蚁中取出一些只组成大小分别为S....B的集合。问方法总数。

解法：计数Dp中的多重集组合数问题。

 模板题。只是统计取出S..B个的方法总数。最后做个相加即可。

但是需要计算数据范围蚂蚁总共有1000*100只。在dp中用到滚动数组作为优化。

#include<cstdio>
#include<iostream>
#define M 1000000
using namespace std;
int n, m, a, b, ans;
int p[100010];
int d[2][100010];
int main(){
    scanf("%d%d%d%d", &n, &m, &a, &b);
    for (int i = 1; i <= m; i++) {
        int x;
        scanf("%d", &x);
        p[x]++;
    }
    d[0][0] = 1;
    d[1][0] = 1;
    for (int i = 1; i <= n; i++){
        for (int j = 1; j <= b; j++){
            if(j - p[i] - 1 >= 0)
                d[i % 2][j] = (d[(i + 1) % 2][j] % M + d[i % 2][j - 1] % M - d[(i + 1) % 2][j - 1 - p[i]] % M + M) % M;
            else
                d[i % 2][j] = (d[(i + 1) % 2][j] % M + d[i % 2][j - 1] % M) % M;
         }
    }
    for (int i = a; i <= b; i++) ans = (ans % M + d[n % 2][i] % M) % M;
    printf("%d", ans);
    return 0;
}