题意
现有巴萨球迷皇马球迷各
s1,s2
s
1
,
s
2
名,有
n
n
个娱乐场所,第 个娱乐场所可容纳
ai
a
i
人。当容纳的人都是某一队的球迷,或两队球迷人数相等时可行。保证
∑i=1n=s1+s2。
∑
i
=
1
n
=
s
1
+
s
2
。
求可行解的种数。
1≤s1,s2≤100000
1
≤
s
1
,
s
2
≤
100000
1≤n≤200
1
≤
n
≤
200
思路
如果用一个二维的数组 dpi,j d p i , j 表示巴萨皇马球迷各为 i,j i , j 人时的可行解数,那么只用把满足条件的 dpi,j−a,dpi−a,j,dpi−a2,j−a2 d p i , j − a , d p i − a , j , d p i − a 2 , j − a 2 转移即可。但是不难发现,已知枚举到的场所编号时,第二维的值是唯一可确定的,所以第二维可直接省去,根据01背包的一维优化,将循环倒转即可。
代码
#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
#define P 1000000007
typedef long long LL;
using namespace std;
int dp[100003];
int main()
{
int s1,s2;
while(~scanf("%d%d",&s1,&s2))
{
memset(dp,0,sizeof(dp));
dp[0]=1;
int n,a;
scanf("%d",&n);
while(n--)
{
scanf("%d",&a);
DOR(i,s1,0)
{
if(i>=a)(dp[i]+=dp[i-a])%=P;
if((~a&1)&&i>=(a>>1))(dp[i]+=dp[i-(a>>1)])%=P;
}
}
printf("%d\n",dp[s1]);
}
return 0;
}