POJ-1837 Balance DP

  题目链接:http://poj.org/problem?id=1837

  题意是给定一个平衡杆,平衡杆的两端某些位置有钩子可以放置砝码,现在给定一些砝码,要你求出有多少种方法能使之保持平衡。方法是用平衡度的方法来进行状态转移,f[i][j]表示前i个物品的平衡度为j时的状态总数,则转移方程为f[i][j+c[k]*g[i]]+=f[i-1][j],显然f[m][0]就是最终答案,不过数组下标不能为负数,映射一下就可以了。

 1 //STATUS:C++_AC_16MS_1632KB

 2 #include<stdio.h>

 3 #include<stdlib.h>

 4 #include<string.h>

 5 #include<math.h>

 6 #include<iostream>

 7 #include<string>

 8 #include<algorithm>

 9 #include<vector>

10 #include<queue>

11 #include<stack>

12 #include<map>

13 using namespace std;

14 #define LL __int64

15 #define pii pair<int,int>

16 #define Max(a,b) ((a)>(b)?(a):(b))

17 #define Min(a,b) ((a)<(b)?(a):(b))

18 #define mem(a,b) memset(a,b,sizeof(a))

19 #define lson l,mid,rt<<1

20 #define rson mid+1,r,rt<<1|1

21 const int N=25,INF=0x3f3f3f3f,MOD=1999997;

22 const LL LLNF=0x3f3f3f3f3f3f3f3fLL;

23 

24 int c[N],g[N],f[N][15010];

25 int n,m;

26 

27 int main()

28 {

29  //   freopen("in.txt","r",stdin);

30     int i,j,k;

31     while(~scanf("%d%d",&n,&m))

32     {

33         mem(f,0);

34         f[0][7500]=1;

35         for(i=1;i<=n;i++)

36             scanf("%d",&c[i]);

37         for(i=1;i<=m;i++)

38             scanf("%d",&g[i]);

39         for(i=1;i<=m;i++){

40             for(j=0;j<=15000;j++){

41                 if(f[i-1][j]){

42                     for(k=1;k<=n;k++)

43                         f[i][j+c[k]*g[i]]+=f[i-1][j];

44                 }

45             }

46         }

47 

48         printf("%d\n",f[m][7500]);

49     }

50     return 0;

51 }

 

你可能感兴趣的:(poj)