通过交换a,b中的元素,使数组a元素的和与数组b元素的和之间的差

2020-01-09  本文已影响0人  桂老七

华为06年面试题(要求8分钟完成)

题目:

有两个数组a,b,大小都为n,数组元素的值任意,无序;
要求:

通过交换a,b中的元素,使数组a元素的和与数组b元素的和之间的差最小。

前几天华为机试,挂了...题目就是这道题,当时只有百分之三十的通过率,机试挂了面试机会都没有,郁郁寡欢,一直思索这题到底正确的方式是什么,特此记录.

网上很多代码都是直接遍历任意两个之间交换,这样只能做到局部最优无法处理a{7,8,97,9800} b{1,15,898,9000}这样的情况(必须同时交换两个);

这题比较正确的思路可能是在2n个数里取n个数,这n个数最接近sum/2的情况;
网上有讨论 n个数里取若干个数,使其和最接近值M---https://blog.csdn.net/lsjseu/article/details/11660731

也就是这个问题可以看作一个背包问题,容量为sum/2,但是有个限制条件2n个数里必须要取n个;

暂时代码如下,有更好解法的大佬还望留个言指导下,对这题是真的懵逼...

public class BagProblem {
    // 存a,b数组拼接后的数组;
    private  int[] arr;

    public BagProblem(int[] arr){
        this.arr=arr;
    }

    public  int sum(int[] arr){
        int sum=0;
        for(int i=0;i<arr.length;i++){
            sum+=arr[i];
        }
        return sum;
    }
    public  int sumz(int[] arr,int index){
        int sum=0;
        for(int i=0;i<index;i++){
            sum+=arr[i];
        }
        return sum;
    }



    // 递归:(复杂度2^len次方,舍弃)
    public  int recursion(int i,int j,int v){
        if(i<=0||j<0||v<0||i<j){
            return Integer.MIN_VALUE;
        }
        if(j==0){
            return 0;
        }
        if(arr[i-1]<=v){
            return Math.max(recursion(i-1,j,v),recursion(i-1,j-1,v-arr[i-1])+arr[i-1]);
        }else{
            return recursion(i-1,j,v);
        }
    }


    /**
     * 动态规划:
     * 复杂度:len*len/2*sum
     * 注意顺序必须前面算完了才开始算后面
     * 计算前可以先比较下len*len/2*sum和2^len哪个大再决定用哪种
     * i表示前i个数组
     * j表示前i个里选j个
     * v表示放不超过v重量
     * */

    public  int dynamic(int[] input){
        int len =input.length;
        int sum=sum(input)/2;
        // 存总重量
        int[][][] result=new int[len+1][len/2+1][sum+1];

        for(int i=1;i<=len;i++){
            for(int v=0;v<=sum;v++){
                for(int j=1;j<=len/2&&j<=i;j++){
                    // 比较i与j,i个里选j个
                    if(i>j){
                        if(input[i-1]>v){
                            result[i][j][v]=result[i-1][j][v];
                        }else{
                            result[i][j][v]=Math.max(result[i-1][j][v],result[i-1][j-1][v-input[i-1]]+input[i-1]);
                        }
                    }else{
                        // 当前i==j时,向上递归的时候result[i-1][j][v]会失去意义
                        if(sumz(input,i)>v||result[i-1][j-1][v-input[i-1]]<0){
                            result[i][j][v]=Integer.MIN_VALUE;
                        }else{
                            result[i][j][v]=result[i-1][j-1][v-input[i-1]]+input[i-1];
                        }
                    }

                }
            }
        }
        return result[len][len/2][sum];
    }







    public static void main(String[] args) {
        // 数组合并下
//        int[] input={1, 3, 5 ,49,0, 2, 4 ,18,1, 3, 5 ,49,0, 2, 4 ,18,1, 3, 5 ,49,0, 2, 4 ,18,1, 3, 5 ,49,0, 2, 4 ,18,1, 3, 5 ,49,0, 2, 4 ,18,1, 3, 5 ,49,0, 2, 4 ,18,1, 3, 5 ,49,0, 2, 4 ,18,1, 3, 5 ,49};
        int[] input={9,990,99000,99999};
//        int[] input={1,7,8,14,98,9800,9000,897};


        BagProblem bp=new BagProblem(input);

        System.out.println("2n数组的长度:"+input.length);

//        int k=bp.recursion(input.length,input.length/2,bp.sum(input)/2);
        final long start = System.currentTimeMillis();
        int k=bp.dynamic(input);
        final long end = System.currentTimeMillis();
        System.out.println("花费时间:"+(end-start));
        System.out.println("最接近sum/2的组合的和:"+k);
        System.out.println("最小差值结果:"+ (bp.sum(input)-2*k));
    }
}
上一篇下一篇

猜你喜欢

热点阅读