Top K II

问题: 给定一维整型数组和一个整数K,找出和不小于K的数目最少的子数组。

思路: 改进快排分区思路(见Top K)。对数组进行降序排列,分区同时计算左边数之和。

java代码:

int partitionTwo(int[] data,int start,int end,ArrayList<Integer> currSum){
        if(start>end)
            return -1;
        int sum=0;
        int index=start;
        int tmp=data[index];
        data[index]=data[end];
        for(int i=start;i<end;i++){
            if(data[i]>=data[end]){
                if(i!=index){
                    tmp=data[index];
                    data[index]=data[i];
                    data[i]=tmp;
                }
                index++;
            }
        }
        tmp=data[end];
        data[end]=data[index];
        data[index]=tmp;
        for(int i=0;i<index;i++){
            sum+=data[i];
        }
        currSum.set(0,sum);
        return index;
    }
    void getTopKII(int[] data,int k){
        int start=0,end=data.length-1;
        int last=-1;
        ArrayList<Integer> currSum=new ArrayList<Integer>();
        int index=partitionTwo(data,start,end,currSum);
        while(index>=0){
            if(currSum.get(0)>=k){
                last=index;
                end=index-1;
                index=partitionTwo(data,start,end,currSum);
            }
            else{
                start=index+1;
                k=k-currSum.get(0);
                index=partitionTwo(data,start,end,currSum);
            }
        }
        for(int i=0;i<last;i++){
            System.out.println(data[i]+"\t");
        }
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值