POJ 3046 DP

题意:
这里写图片描述
这里写图片描述
思路:先说O(A^2T)的做法
f[i][j]表示前i个家族选j个蚂蚁的方案数 f[i][j]+=f[i-1][k] j-vis[i]<=k<=j

然后前缀和优化一哈 就成O(AT)的了

我们发现空间还是会炸 这样子就压掉一维好了 空间是O(A)的

嗯就酱

//By SiriusRen
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int T,A,S,B,f[100005],sum[1000*105],vis[1005],jy,ans;
int main(){
    scanf("%d%d%d%d",&T,&A,&S,&B);
    for(int i=1;i<=A;i++){
        scanf("%d",&jy);
        vis[jy]++;
    }
    f[0]=1;
    for(int i=1;i<=A+1;i++)sum[i]=1;
    for(int i=1;i<=T;i++){
        for(int j=1;j<=A;j++)
            f[j]=(sum[j+1]-sum[max(j-vis[i],0)])%1000000;
        for(int j=1;j<=A;j++)sum[j+1]=sum[j]+f[j];
    }
    for(int i=S;i<=B;i++){
        ans=(ans+f[i])%1000000;
    }
    printf("%d\n",ans);
}

这里写图片描述

posted @ 2016-10-28 12:54  SiriusRen  阅读(85)  评论(0编辑  收藏  举报