砝码称重,砝码可以两边放

阅读更多

Description

　　设有已知m个砝码，每个砝码的重量分别为g1、g2、…gm，现有一架天平，问用这些砝码能称出多少种不同的重量。（1<=m<=100，1<=每种砝码重量<=255）。（注意，砝码可以放在天平的两边）。

Input

　第一行：整数 m、n，两整数间用空格分开；
　接下来有m行：每行一个整数，第i+1行表示第I个砝码的重量，

Output

　能称出的重量个数

Sample Input

　2
　3
　5

Sample Output

　4

#include 
#include 
//可以左右放,所以数组的长度要扩大一倍
#define MAX 100 * 255 * 2 + 5

int dp[MAX], buf[MAX];

//母函数的应用,表达式为(1 + x^3 + x^-3)(1 + x^5 + x^-5) 例子的表达式
void Generation(int value[], int length)
{
	int i, j, k;
	memset(dp, 0, sizeof(dp));
	dp[0] = 1;
	for(i = 0; i < length; i++)
	{
		memset(buf, 0, sizeof(buf));
		for(j = -value[i]; j <= value[i]; j += value[i]) //
		{
			for(k = -value[i]; k + j + value[i] <= MAX; k++)
			{
				buf[k + j + value[i]] += dp[k];
			}
		}
		for(j = 0; j <= MAX; j++)
		{
			dp[j] = buf[j];
		}
	}
}

int main()
{
	int m, i, cnt;
	int value[101];
	while(scanf("%d", &m) != EOF)
	{
		for(i = 0; i < m; i++)
		{
			scanf("%d", &value[i]);
		}
		Generation(value, m);
		cnt = 0;
		for(i = 1; i < MAX; i++)
		{
			if(dp[i] != 0)
				cnt++;
		}
		printf("%d\n", cnt / 2);
	}
	return 0;
}