POJ 3046 Ant Counting(递推,和号优化)

计数类的问题,要求不重复,把每种物品单独考虑。

将和号递推可以把转移优化O(1)。

f[i = 第i种物品][j = 总数量为j] = 方案数

f[i][j] = sigma{f[i-1][j-k],(k = [0,min(j,c[i])])}
把和号展开 
f[i][j]  :    j-0,j-1,...,j-a[i]
f[i][j-1] :       j-1,j-2,...,j-a[i]-1
中间部分是一样的可以避免重复计算。

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<queue>
#include<vector>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<algorithm>
//#include<bits/stdc++.h>
using namespace std;

int T, A, S, B;
const int maxt = 1e3+5, maxa = 1e5+5, mod = 1e6;
int f[2][maxa];
int c[maxt];

//#define LOCAL
int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
#endif
    scanf("%d%d%d%d",&T,&A,&S,&B);
    for(int i = A; i--;){
        int x; scanf("%d",&x);
        c[x]++;
    }
    f[0][0] = f[1][0] = 1; //不选方案数为1
    for(int i = 1; i <= T; i++){
        int a = i&1, b = a^1;
        for(int j = 1; j <= B; j++){
            if(j-c[i]>0){
                f[a][j] = ( f[a][j-1] + f[b][j] - f[b][j-1-c[i]] ) % mod;
            }else {
                f[a][j] = ( f[a][j-1] + f[b][j] ) % mod;
            }
        }
    }
    int sum = 0, *F = f[T&1];
    for(int i = S; i <= B; i++){
        sum = (sum + F[i]) % mod;
    }
    printf("%d\n",sum<0?sum+mod:sum);
    return 0;
}

 

posted @ 2015-10-17 12:38  陈瑞宇  阅读(207)  评论(0编辑  收藏  举报