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 }