[PKU 3046] Ant Counting (计数DP+差分加数)

原创 2016年06月01日 18:43:17

PKU - 3046

有 T组蚂蚁,每组有 Ni只,同组蚂蚁没有区别
问从中选出一定数量的选法有多少种


计数DP
dp[i][j]表示考虑到第 i组蚂蚁,size大小为j的选法
其中 i可以滚动数组优化掉,就变成了 dp[2][j]

然后对于每组蚂蚁,枚举只数 k,更新答案
dp[cur][j]+=dp[las][jk] 其中cur为当前状态,las为前一层状态
但是这样做会 T,所以我们不必逐个更新,可以使用差分的技巧
比如要将 jj+num[i]都加上某个值,只需在 j处加,在 j+num[i]+1处减
最后扫一遍更新即可

时间复杂度 O(AT)

#pragma comment(linker, "/STACK:102400000,102400000")
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cctype>
#include <map>
#include <set>
#include <queue>
using namespace std;
typedef pair<int,int> Pii;
typedef long long LL;
typedef unsigned long long ULL;
typedef double DBL;
typedef long double LDBL;
#define MST(a,b) memset(a,b,sizeof(a))
#define CLR(a) MST(a,0)
#define Sqr(a) ((a)*(a))

const int maxt=1e3+10, maxc=1e5+10, MOD=1e6;
int T,A,S,B;
int dp[2][maxc];
int num[maxt];

int main()
{
    #ifdef LOCAL
    freopen("in.txt", "r", stdin);
//  freopen("out.txt", "w", stdout);
    #endif

    while(~scanf("%d%d%d%d", &T, &A, &S, &B))
    {
        CLR(dp); CLR(num);
        for(int i=1; i<=A; i++)
        {
            int x;
            scanf("%d", &x);
            num[x]++;
        }
        dp[0][0]=1;
        int sum=0;
        for(int i=1; i<=T; i++)
        {
            int cur=i&1, las=(i-1)&1;
            CLR(dp[cur]);
            for(int j=0; j<=sum; j++)
            {
                int r=j+num[i]+1;
                dp[cur][j] += dp[las][j];
                dp[cur][r] -= dp[las][j];
            }
            int cnt=0;
            for(int j=0; j<=sum+num[i]+1; j++)
            {
                cnt = (cnt+dp[cur][j])%MOD;
                dp[cur][j]=cnt;
            }
            sum+=num[i];
        }
        int ans=0;
        for(int i=S; i<=B; i++) ans = (ans+dp[T&1][i]) %MOD;
        printf("%d\n", (ans+MOD)%MOD);
    }
    return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

[POJ 3046]Ant Counting[dp][优化]

题目链接:[POJ 3046]Ant Counting[dp][优化] 题意分析: 蚂蚁有T个家族,每个家族里的蚂蚁没有区别,总共有A只蚂蚁,现在问:给出每个家族的蚂蚁数量,最多能组合成多少个不同...

POJ3046 Ant Counting DP 隔板法讲解 有图有真相

POJ 3046  Ant Counting 描述 Description Bessie was poking around the ant hill one day watching the ant...
  • moep0
  • moep0
  • 2016年11月02日 15:28
  • 290

POJ 3046 Ant Counting dp

定义状态为前i种蚂蚁能够形成大小为j的集合的种数 注意到每次都要求一段dp数字的和,我们可以使用一个前缀数组sum。 由于sum数组的使用,我们可以在O(1)的时间内算出dp(i, j) 同时...

挑战练习题2.3动态规划 poj3046 Ant Counting dp

题目链接:http://poj.org/problem?id=3046题意:有T种蚂蚁,共A只。同一个种的蚂蚁长得一样,但是不同种的蚂蚁牙齿颜色不同。任取n只蚂蚁(S...
  • yxg_123
  • yxg_123
  • 2017年03月10日 17:38
  • 65

poj 3046 Ant Counting dp 优化

《挑战》练习题,dp+优化。

POJ 3046 Ant Counting (dp 多重集组合数)

differ from the White Book

POJ 3046 - Ant Counting(dp多种背包变型)

Description Bessie was poking around the ant hill one day watching the ants march to and fro while ...

POJ3046——Ant Counting(dp算法)

先占个坑 不知道为啥一直 wa 简单dp #include #include #include #include using namespace std; #define MAXN 1...

POJ 3046 Ant Counting

背包水题...一开始看着觉得是多重背包,其实是分组背包... 每个蚂蚁家庭是一个分组,每个分组选一种情况,从0到分组蚂蚁数上限; 最后把从s只到b只的记录加起来就行了,真心水... 注意要求输出最后六...

Ant Counting (poj 3046 分组背包)

题意:给出T个数字,编号1~T,每个数字有N[i]个。用这些数字组成一个序列,最后问长度在S~B之间的不同序列有多少种。 思路:分组背包。每个蚂蚁家族为一个分组,在每个分组里的选择有N[i]种(选1~...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:[PKU 3046] Ant Counting (计数DP+差分加数)
举报原因:
原因补充:

(最多只允许输入30个字)