POJ-1837 Balance

5 篇文章 0 订阅

题目大意:有一个天平,单臂长15。天平上有C个位置可以用来挂砝码。有G个砝码,求将所有砝码全部挂上时使天平平衡的方法有多少种。
题目链接:http://poj.org/problem?id=1837
注意题目中几个变量的范围都不大。暴力的话是O(C^G)复杂度,最坏为20^20,肯定不行。
寻找题目的递推规律,可以知道挂一个砝码时影响天平的状态,且只与挂之前的天平状态有关。使用二维dp数组做动态规划,第一维是当前挂的砝码数量,第二维是天平状态。天平状态使用合力矩来表示。合力矩范围为[-7500,7500],可平移到[0,15000]。
初始条件:
dp[0][7500]=1 其他均初始化为0
递推式:
dp[i][j+weight[i]*loc[k]]+=dp[i-1][j]
复杂度为C*G*15000

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <stdlib.h>
#include <math.h>

using namespace std;

#define MAXN 21
#define MAXM 15001

int dp[MAXN][MAXM];
int loc[MAXN];
int weight[MAXN];

int main()
{
    int c, g;
    while (~scanf("%d%d", &c, &g))
    {
        memset(dp, 0, sizeof(dp));
        for (int i = 1; i <= c; i++)
            scanf("%d", &loc[i]);
        for (int i = 1; i <= g; i++)
            scanf("%d", &weight[i]);
        dp[0][7500] = 1;

        for (int i = 1; i <= g; i++)
        {
            for (int j = 1; j <= 15000; j++)
            {
                for (int k = 1; k <= c; k++)
                {
                    dp[i][j + weight[i] * loc[k]] += dp[i - 1][j];
                }
            }
        }
        printf("%d\n", dp[g][7500]);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值