【BZOJ1630/2023】[Usaco2007 Demo]Ant Counting DP

【BZOJ1630/2023】[Usaco2007 Demo]Ant Counting

题意:T中蚂蚁,一共A只,同种蚂蚁认为是相同的,有一群蚂蚁要出行,个数不少于S,不大于B,求总方案数

题解:DP,先列出朴素的方程,用f[i][j]表示前i种,出行j只的方案数,v[i]代表第i中蚂蚁的个数

 

f[i][j]=∑f[i-1][j-k] (0≤k≤min(j,v[i]))

也可以表示为

f[i][j]=∑f[i-1][k] (j-min(j,v[i])≤k≤j)

发现时间复杂度为均摊O(A^2),那么我们可以用前缀和优化,空间复杂度为O(T*A),我们可以采用滚动数组

关于数据范围什么的,反正O(T*A)过了,题目描述233

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#define mod 1000000
using namespace std;
int f[2][100010],s[2][100010];
int n,m,l,r;
int v[100010];
int main()
{
    int i,j,k,a;
    scanf("%d%d%d%d",&n,&m,&l,&r);
    for(i=1;i<=m;i++)
    {
        scanf("%d",&a);
        v[a]++;
    }
    for(i=0;i<=m;i++)    s[0][i]=1;
    for(i=1;i<=n;i++)
    {
        for(j=0;j<=m;j++)
        {
            if(j<=v[i])    f[i&1][j]=s[(i&1)^1][j];
            else    f[i&1][j]=(s[(i&1)^1][j]-s[(i&1)^1][j-v[i]-1]+mod)%mod;
            if(!j)    s[i&1][j]=f[i&1][j];
            else    s[i&1][j]=(s[i&1][j-1]+f[i&1][j])%mod;
        }
    }
    printf("%d",(s[n&1][r]-s[n&1][l-1]+mod)%mod);
    return 0;
}
posted @ 2017-02-16 14:28  CQzhangyu  阅读(231)  评论(0编辑  收藏  举报