***POJ 3046 Ant Counting

原题链接

思路：
首先弄清楚T，A，S，B的意思。
T是蚂蚁有多少种。
A是蚂蚁有多少个。
每组蚂蚁数在S~B之间。
output有多少组符合条件。
然后用动态规划。
dp[i][j]在第i种蚂蚁前，已经取了j种蚂蚁。
dp[i+1][j+k] = dp[i][j] + dp[i][j+1] + …… + dp[i][j+k]
（已经从i种之前选了j~j+k只蚂蚁，要再选取k~0只第i种蚂蚁）
要求j<=B且k<=Ni且j+k<=A。
在此范围内j和k循环，dp[i+1][j+k] += dp[i][j]。
因为dp计算始终是i和i+1之间的关系，所以可以简化一下，把dp[i][j]设为c1[j]，把dp[i+1][j]设为c2[j]，i每增加一，则c1=c2，c2
清零。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cmath>

using namespace std;

int T,A,S,B,i,j,k,sum;
int ant[1001],c1[1000000],c2[1000000];

int main()
{
    scanf("%d %d %d %d",&T,&A,&S,&B);
    for(i = 0; i < A; i++){
        int temp;
        scanf("%d",&temp);
        ant[temp]++;  //计算每种蚂蚁的个数 
    }
    c1[0] = 1;
    for(i = 1; i <= T; i++)  //历遍每种蚂蚁
    {
        for(j = 0; j <= B; j++){  //在第i种蚂蚁之前，已经取了j只蚂蚁 
            for(k = 0; k <= ant[i] && k+j <= B ; k++){  //取k只第i种蚂蚁 
                c2[j+k] = (c2[j+k] + c1[j]) % 1000000;  
            }
        }
        for(j = 0;j <= B; j++){
            c1[j] = c2[j];
            c2[j] = 0;
        } 
    }
    for(i=S;i<=B;i++){
        sum = (sum + c1[i]) % 1000000;
    }
    printf("%d\n",sum);
    return 0;
}