统计蚂蚁(ants)【DP】

>Description
蚂蚁山上有T(1<=T<=1,000)种蚂蚁，标记为1…T,每种蚂蚁有N_i只蚂蚁(1<=N_i<=100),现有A(A<=5000)只蚂蚁，从中选出S,S+1,…,B(1<=S<=B<=A)只蚂蚁一共有多少种选法？
如有5只蚂蚁分别为{1,1,2,2,3},一共有3种蚂蚁，每一种蚂蚁的数量分别为2，2，1，以下是选不同数量蚂蚁的方法：

1个蚂蚁3种选法:{1}{2}{3}

2个蚂蚁5种选法:{1,1}{1,2}{1,3}{2,2}{2,3}

3个蚂蚁5种选法:{1,1,2}{1,1,3}{1,2,2}{1,2,3}{2,2,3}

4个蚂蚁3种选法:{1,2,2,3}{1,1,2,2}{1,1,2,3}

5个蚂蚁1种选法:{1,1,2,2,3}

你的任务是从中选S…B只蚂蚁的方法总和。

---

>Input
第一行: 4个空格隔开的整数: T, A, S和B；
第2到A+1行：每行一个整数表示蚂蚁的种类。

>Output
输出从A只蚂蚁中选出S…B只蚂蚁的方法数，答案保留后6位。

对于30%的数据：T<=30,A<=100;
对于50%的数据：T<=100,A<=400;
对于100%的数据：T<=1000,A<=5000;

---

>解题思路
一道DP题

设$f[i][j]$为前$i$种蚂蚁选$j$只的总方案
我们只需预处理一个桶，记录第$i$种蚂蚁的总数
很容易得出状态转移方程：
$f[i][j]=\Sigma f[i-1][j-(0t[i])]$（默默枚举第i种蚂蚁选多少只，因为答案求的只是组合数）

我们又会发现这是一个熟悉的前缀和优化，所以我们再预处理一下前缀和就可以了（虽然我没有预处理前缀和就过了，不过我还是打了一下前缀和）

---

>代码

#include <iostream>
#include <cstdio>
#include <cstring>
#define p 1000000
#define N 5005
using namespace std;

int t, n, a, b, s[1005], f[1005][N], sum[1005][N], ans;

int main()
{
	scanf ("%d%d%d%d", &t, &n, &a, &b);
	int x;
	for (int i = 1; i <= n; i++)
	  scanf ("%d", &x), s[x]++;
	for (int i = 0; i <= n; i++)
	  sum[1][i] = min (i, s[1]) + 1; //第1种蚂蚁最多只能选s[1]只
	for (int i = 2; i <= t; i++)
	{
		f[i][0] = sum[i][0] = 1;
		for (int j = 1; j <= b; j++)
	  	{
	  		f[i][j] = (p + sum[i - 1][j] - sum[i - 1][j - min (j, s[i]) - 1]) % p; //min(j,s[i])防止越界
	  		sum[i][j] = (sum[i][j - 1] + f[i][j]) % p;
	  	}
	}
	printf ("%d", (p + sum[t][b] - sum[t][a - 1]) % p);
	return 0;
}