ZOJ 3682 E - Cup 3

二維DP，用a[i][j]表示前i個廣場容納j個球迷的方案數，這裏的球迷或爲Barcelona，或爲Madrid。由於題目保證兩隊人數與廣場總容量相等，則可選擇人數較少的隊DP。不過這樣的話用來存儲中間結果的二維數組過大，爲200*100000*4Byte，剛好約是內存限制的1.25倍。開始想用map<pair<int, int>, int>節約內存，結果超時，故而重新用滾動數組寫了一遍。

#include <stdio.h>
#include <memory.h>
#define Mod 1000000007
#define Max 100001
int a[2][Max];
int s1, s2, n;
int k[200];

void dp(int ind)
{
	int d = ind%2;
	int s = 1 - d;
	memset(a[d], 0, sizeof(a)/2);
	for(int i=0; i<=s1; ++i)
	{
		if(a[s][i] > 0)
		{
			if(i+k[ind]<=s1)
			{
				a[d][i+k[ind]] += a[s][i];
				if(a[d][i+k[ind]] > Mod)
					a[d][i+k[ind]] %= Mod;
			}
			if((k[ind]%2)==0 && i+k[ind]/2<=s1)
			{
				a[d][i+k[ind]/2] += a[s][i];
				if(a[d][i+k[ind]/2] > Mod)
					a[d][i+k[ind]/2] %= Mod;
			}
			a[d][i] += a[s][i];
			if(a[d][i] > Mod)
				a[d][i] %= Mod;
		}
	}
	if(ind < n)
		dp(ind+1);
}

int main()
{
	while(~scanf("%d%d", &s1, &s2))
	{
		scanf("%d", &n);
		for(int i=0; i<n; ++i)
			scanf("%d", &k[i]);
		if(s1 > s2)
			s1 = s2;
		memset(a, 0, sizeof(a)/2);
		a[0][0] = 1;
		a[0][k[0]] = 1;
		if((k[0]%2) == 0)
			a[0][k[0]/2] = 1;
		dp( 1 );
		printf("%d\n", a[1-(n%2)][s1]);
	}
	return 0;
}