[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【多重背包】

Ant Counting Time Limit: 1000MS   Memory Limit: 65536K Total Submissions: 4091   A...
  • liuke19950717
  • liuke19950717
  • 2016年03月30日 17:01
  • 540

poj 3046 Ant Counting

题意:T种蚂蚁,共计A只,种内没有区别,问含有x只蚂蚁的子集有多少种(x∈[S,B]) 分析 :1.类比C[i][j]相当于把每个种类增加到A只;     2.所以可以从类似组合公式的递推下手; ...
  • c337134154
  • c337134154
  • 2014年11月30日 14:29
  • 473

Ant Counting(3046)

大致题意:给n. m. l, y,n 种数,m个数, 问cpn
  • u011867401
  • u011867401
  • 2014年11月05日 13:27
  • 177

POJ 3046 Ant Counting

POJ 3046题目大意如下: 指定蚂蚁家族有T个,每个家族的蚂蚁数量最多为100只,最少当然为1了,总共有A只蚂蚁,每只蚂蚁唯一的区别就是它所对应的家族编号,相同编号的蚂蚁看成没有差别。现在将这些...
  • kornberg_fresnel
  • kornberg_fresnel
  • 2016年03月07日 00:13
  • 265

***POJ 3046 Ant Counting

原题链接思路: 首先弄清楚T,A,S,B的意思。 T是蚂蚁有多少种。 A是蚂蚁有多少个。 每组蚂蚁数在S~B之间。 output有多少组符合条件。 然后用动态规划。 dp[i][j]在第...
  • tjx41624560
  • tjx41624560
  • 2017年03月17日 17:11
  • 78

POJ 3046 Ant Counting

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

POJ - 3046 :Ant Counting

POJ - 3046 :Ant CountingBessie was poking around the ant hill one day watching the ants march to and...
  • Mitsuha_
  • Mitsuha_
  • 2017年08月28日 18:47
  • 112

dp: POJ 3046 Ant Counting

Description Bessie was poking around the ant hill one day watching the ants march to and fro while ...
  • Davenny
  • Davenny
  • 2016年07月29日 14:29
  • 111

POJ 3046 Ant Counting dp

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

POJ 3046 Ant Counting (dp)

Ant Counting Time Limit: 1000MS   Memory Limit: 65536K Total Submissions: 3111   A...
  • lwt36
  • lwt36
  • 2015年03月07日 01:54
  • 285
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:[PKU 3046] Ant Counting (计数DP+差分加数)
举报原因:
原因补充:

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