初见安~这个题的题目很坑,因为题面在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内所有的情况数加起来即可。
所以就很容易联想到动态规划啦~
我们设表示涉及到了i个家族,共有j只蚂蚁时的情况数。可以转移过来的状态为:
1.这只蚂蚁还是这个种族的,
2.这只蚂蚁不要。
就这样。当然,可以发现这样算的话前i - 1个种族的蚂蚁会重复算一遍【可以手动模拟一下dp的过程(本狸太蠢了真的说不清楚……)这么理解吧:
所以,中间那一部分就重复了。
再往前枚举,重复的部分j的减数会越来越大。
但其实如果再往前枚举几个,会发现其实j往前延伸的范围会受到每个种类的蚂蚁数量的限制。所以到了一定程度后,可以概括为:,再减的那个1是当前的意思。就如同上面所描述的。
所以最后真正的递推式为:
.
后面那一部分有没有必要去重,取决于是否.
重点讲完了。事后累加起来就可以了。当然中途是要取模的,公式为重点所以就省略了。
另外,因为本题的空间范围只有哦64M,如果像上述那样直接开的话空间复杂度就是,会爆空间的。所以我们选择——滚动数组。本题特殊,所以要两行一起滚。具体实现请自行脑补。
#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——