当前位置:编程学习 > C/C++ >>

poj 1837 Balance (dp背包)

 


题目大意:

有一个天平,天平左右两边的手臂长度都是15,手臂上面有些位置会有挂钩。还有G个砝码 (1 <= G <= 20),它们重量各不相同,在1~25中取值。

给出C个挂钩,它们的位置在【-15..15】,不会重叠。负号的代表在左边臂,正号的在右边。

要求把所有砝码都放在挂钩上,多个砝码可以挂在同一个挂钩上,问有多少种不同的方案使天平能够平衡?

 

 

 

思路:

天平左臂的力矩和是负数,右边的力矩和是正数,那么左边+右边的力矩之和,如果是正数,代表天平平衡倾向右边,负数代表倾向左边,为0的时候天平是平衡的。我们把 “左边力矩和+右边力矩和”叫做平衡系数


状态f[i][j]代表用前i个砝码,放置成平衡系数为j的时候共有多少种方案。
那么,f[i][j] += f[i-1][j-C[k]*G[i]],  {0<=k=<c};


因为平衡系数中有负数的,所以要所有平衡系数往右平移,即加上一个足够大的正数。可以计算出力矩之和最小负数的是把所有砝码都挂在天平-15的位置上,砝码最多20个,取值最大的情况是6...25,那么砝码之和最终为 (6+25)*20/2 = 310, 力矩之和为 -15*310 = 4650
所以加上4650即可,这是位置4650代表的是原来天平的中间位置,
初始化 f[0][4650] = 1, 表示一个砝码都不挂,这是一种平衡的方案。


最终,f[G][4650]就是答案。

 


PS: 这题最开始我是用滚动数组做的,用G++提交一直RE到死,郁闷。后来改用C++提交就可以AC了。

 


代码:


[cpp]  #include<iostream>  
#include<cstdio>  
#include<cmath>  
#include<algorithm>  
#include<cstring>  
#define SQ(x) ((x)*(x))  
#define MP make_pair  
const int INF = 0x3f3f3f3f; 
const double PI = acos(-1.0); 
typedef long long int64; 
using namespace std; 
 
const int MAXN = 22; 
const int mid = 4650; 
int pos[MAXN], w[MAXN]; 
int f[22][mid*2+10]; 
int n, m; 
 
int main(){ 
    while(~scanf("%d%d", &n, &m)){ 
 
        for(int i=0; i<n; ++i) 
            scanf("%d", &pos[i]); 
 
        for(int i=0; i<m; ++i) 
            scanf("%d", &w[i]); 
 
        memset(f, 0, sizeof(f)); 
        f[0][mid] = 1; 
 
        for(int i=0; i<m; ++i){ 
            for(int j=0; j<n; ++j){ 
 
                int add = w[i]*pos[j];   
                for(int v=mid*2; v-add>=0; --v){   
                    if(v-add <= mid*2) 
                        f[i+1][v] += f[i][v-add]; 
                } 
            } 
        } 
        printf("%d\n", f[m][mid]); 
    } 
    return 0; 

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define SQ(x) ((x)*(x))
#define MP make_pair
const int INF = 0x3f3f3f3f;
const double PI = acos(-1.0);
typedef long long int64;
using namespace std;

const int MAXN = 22;
const int mid = 4650;
int pos[MAXN], w[MAXN];
int f[22][mid*2+10];
int n, m;

int main(){
    while(~scanf("%d%d", &n, &m)){

        for(int i=0; i<n; ++i)
            scanf("%d", &pos[i]);

        for(int i=0; i<m; ++i)
            scanf("%d", &w[i]);

        memset(f, 0, sizeof(f));
        f[0][mid] = 1;

        for(int i=0; i<m; ++i){
            for(int j=0; j<n; ++j){

                int add = w[i]*pos[j]; 
                for(int v=mid*2; v-add>=0; --v){ 
                    if(v-add <= mid*2)
                        f[i+1][v] += f[i][v-add];
                }
            }
        }
        printf("%d\n", f[m][mid]);
    }
    return 0;
}

 

补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,