bzoj·Ant Counting

初见安~这个题的题目很坑,因为题面在bzoj P2023,而样例却在bzoj P1630呢【=============】。当然最后代码交哪一个都一样。这里就很良心地整合出最终的正确题目啦~

Description

    有一天,贝茜无聊地坐在蚂蚁洞前看蚂蚁们进进出出地搬运食物.很快贝茜发现有些蚂蚁长得几乎一模一样,于是她认为那些蚂蚁是兄弟,也就是说它们是同一个家族里的成员.她也发现整个蚂蚁群里有时只有一只出来觅食,有时是几只,有时干脆整个蚁群一起出来.这样一来,蚂蚁们出行觅食时的组队方案就有很多种.作为一头有数学头脑的奶牛,贝茜注意到整个蚂蚁群由T(1≤T≤1000)个家族组成,她将这些家族按1到T依次编号.编号为i的家族里有Ni(1≤Ni≤100)只蚂蚁.同一个家族里的蚂蚁可以认为是完全相同的.

    如果一共有S,S+1….,B(1≤S≤B≤A)只蚂蚁一起出去觅食,它们一共能组成多少种不同的队伍呢?注意:只要两支队伍中所包含某个家族的蚂蚁数不同,我们就认为这两支队伍不同.由于贝茜无法分辨出同一家族的蚂蚁,所以当两支队伍中所包含的所有家族的蚂蚁数都相同时,即使有某个家族换了几只蚂蚁出来,贝茜也会因为看不出不同而把它们认为是同一支队伍.    比如说,有个由3个家族组成的蚂蚁群里一共有5只蚂蚁,它们所属的家族分别为1,1,2,2,3.于是出去觅食时它们有以下几种组队方案:

  ·1只蚂蚁出去有三种组合:(1)(2)(3)

  ·2只蚂蚁出去有五种组合:(1,1)(1,2)(1,3)(2,2)(2,3)

  ·3只蚂蚁出去有五种组合:(1,1,2)(1,1,3)(1,2,2)(1,2,3)(2,2,3)

  ·4只蚂蚁出去有三种组合:(1,2,2,3)(1,1,2,2)(1,1,2,3)

  ·5只蚂蚁出去有一种组合:(1,1,2,2,3)

    你的任务就是根据给出的数据,计算蚂蚁们组队方案的总数.

Input

    第1行:4个用空格隔开的整数T,A,S,B.

    第2到A+1行:每行是一个正整数,为某只蚂蚁所在的家族的编号.

Output

    输出一个整数,表示当S到B(包括S和B)只蚂蚁出去觅食时,不同的组队方案数.

    注意:组合是无序的,也就是说组合1,2和组合2,1是同一种组队方式.最后的答案可能很大,你只需要输出答案的最后6位数字.注意不要输出前导0以及多余的空格.

Sample Input

3 5 2 3
1
2
2
1
3

Sample Output

10
//2只蚂蚁外出有5种组合,3只蚂蚁外出有5种组合.共有10种组合

Sol

不太好想啊……这么说吧,我们先思考确认了有i只蚂蚁的时候的情况数量,而后再把s~b内所有的情况数加起来即可。

所以就很容易联想到动态规划啦~

我们设dp[i][j]表示涉及到了i个家族,共有j只蚂蚁时的情况数。可以转移过来的状态为:

1.这只蚂蚁还是这个种族的,dp[i][j] += dp[i][j - 1]

2.这只蚂蚁不要。dp[i][j] += dp[i - 1][j]

就这样。当然,可以发现这样算的话前i - 1个种族的蚂蚁会重复算一遍【可以手动模拟一下dp的过程(本狸太蠢了真的说不清楚……)这么理解吧:

dp[i - 1][j] = dp[i - 2][j] + dp[i - 1][j - 1]

dp[i][j - 1] = dp[i - 1][j - 1] + dp[i][j - 2]

所以dp[i][j] = dp[i - 2][j] + 2 * dp[i - 1][j - 1] + dp[i][j - 2],中间那一部分就重复了。

再往前枚举,重复的部分j的减数会越来越大。

但其实如果再往前枚举几个,会发现其实j往前延伸的范围会受到每个种类的蚂蚁数量的限制。所以到了一定程度后,可以概括为:dp[i][j - num[i - 1] - 1],再减的那个1是当前的意思。就如同上面所描述的。

所以最后真正的递推式为:

dp[i][j] = dp[i -1][j]+dp[i][j -1]-dp[i-1][j-num[i-1]-1].

后面那一部分有没有必要去重,取决于是否j-num[i-1]>0.

重点讲完了。事后累加起来就可以了。当然中途是要取模的,公式为重点所以就省略了。

另外,因为本题的空间范围只有哦64M,如果像上述那样直接开的话空间复杂度就是O(T*T*N),会爆空间的。所以我们选择——滚动数组。本题特殊,所以要两行一起滚。具体实现请自行脑补。

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 1000005
using namespace std;
const int mod = 1e6;
int read() {
	int x = 0, f = 1, ch = getchar();
	while(!isdigit(ch)) {if(ch == '-') f = -1; ch = getchar();}
	while(isdigit(ch)) x = (x << 1) + (x << 3) + ch - '0', ch = getchar();
	return x * f;
}

int num[maxn], dp[2][maxn];//这里num其实用不了那么多,但是也不会爆空间的
int T, a, b, s;
long long ans = 0;
signed main() {
	T = read(), a = read(), s = read(), b = read();
	for(int i = 1, x; i <= a; i++) x = read(), num[x - 1]++;
	dp[1][0] = dp[0][0] = 1;//初始化——一只蚂蚁都不选算是一种情况
	
	register int k = 0;
	for(int i = 1; i <= T; i++) {//枚举各个种族
		k ^= 1;//滚动
		for(int j = 1; j <= b; j++) {//枚举蚂蚁数量。下面都是上文的公式
			if(j - num[i - 1] - 1 >= 0)
				dp[k][j] = (dp[k ^ 1][j] + dp[k][j - 1] - dp[k ^ 1][j - num[i - 1] - 1] + mod) % mod;
            //避免因为取模出现负数,所以要+mod再%mod
			else
				dp[k][j] = (dp[k ^ 1][j] + dp[k][j - 1]) % mod;
		}		
	}
	for(int i = s; i <= b; i++) ans = (ans + dp[k][i]) % mod;//累加答案
	printf("%lld\n", ans);
	return 0;
}

迎评:)
——End——

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值