题解 | #称砝码,背包问题,动态规划#

称砝码

http://www.nowcoder.com/practice/f9a4c19050fc477e9e27eb75f3bfd49c

import java.util.* ;
public class Main{
    public static void main(String...args) {
        Scanner sc = new Scanner(System.in) ;
        while(sc.hasNextLine()) {
            Integer n = Integer.parseInt(sc.nextLine()) ;
            int W[] = new int[n] ;
            int T[] = new int[n] ;
            for(int i = 0 ; i < n ; i ++) {
                W[i] = sc.nextInt();
            }
            sc.nextLine() ;
            for(int i = 0 ; i < n ; i ++) {
                T[i] = sc.nextInt();
            }
            sc.nextLine() ;
            int count = 0 ;//总个数
            int sum = 0 ;//总重量(背包的总沉重)
            for(int i = 0 ; i < n ; i ++) {
                sum += W[i]*T[i] ;
                count += T[i] ;
            }
            //把砝码平摊开表示,每个砝码当成互相独立的砝码
            int W2[] = new int[count] ;//1,1,2
            int t = 0 ;
            for(int i = 0 ; i < n ; i ++) {
                for(int j = 0 ; j < T[i] ; j ++) {
                    W2[t++] = W[i] ;
                }
            }
            System.out.println(haoManyDiff(W2,sum)) ;
        }
    }
    public static int haoManyDiff(int[] W , int all) {
        int len = W.length ; 
        boolean f[][] = new boolean[len+1][all+1] ;//f[i][j]表示前i个砝码,能否拼出重量j
        f[0][0] = true ;
        for(int i = 1 ; i <= all ; i ++) {//注意:从1开始,因为f[0][0]已经初始化了
            f[0][i] = false ;
        }
        for(int i = 1 ; i <= len ; i ++) {
            for(int j = 0 ; j <= all ; j ++) {
                //转移方程,第i-1个物品,可能放得下,可能放不下
                //f[i][j] = f[i-1][j]&nbs***bsp;f[i-1][j-W[i-1]]
                f[i][j] = f[i-1][j] ;
                if(j >= W[i-1]) {
                     f[i][j] = f[i][j] || f[i-1][j-W[i-1]] ;
                }
            }
        }
        int ret = 0 ;
        for(int i = 0 ; i <= all ; i ++) {
            if(f[len][i]) {
                ret ++ ;
            }
        }
        return ret ;
    }
} 

一个菜鸟的算法刷题记录 文章被收录于专栏

分享一个菜鸟的成长记录

全部评论

相关推荐

07-11 15:12
门头沟学院 Java
别人在上班,我就在工位上看看视频啥的,这正常吗?
程序员小白条:实习就是摸鱼,只是公司指标,把你进来了,可能那时候客户很多,但等你进来的时候,已经是淡季了,根本没多少需求,或者说根本不适合实习生去完成,因此你就每天干坐着就行,可能1,2个月都没需求
实习生的蛐蛐区
点赞 评论 收藏
分享
点赞 评论 收藏
分享
Lorn的意义:你这标个前端是想找全栈吗?而且项目确实没什么含金量,技术栈太少了,边沉淀边找吧 现在学院本想就业好一点四年至少得高三模式两年加油吧
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务