有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;
}