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 }

 

posted @ 2013-01-20 00:56  zhsl  阅读(292)  评论(0编辑  收藏  举报