经典算法题:连续段的中数

题目

一个长度为n的正整数序列,想要从里面取出一段连续的长度大于等于k的序列。定义一个序列的“中数”为最大的整数x,使得序列中至少一半的数字大于等于x,求这个取出来的序列的中数最大值?

思路

滑动窗口的方式处理,获取最大值、最小值作为中数左右边界,然后取中间值,遍历下标0-》k对应的数据判断mid是否是中数(至少一半的数字大于等于mid),如果mid是中位数,则将中数最小值改为mid + 1,继续遍历数组。如果不是中位数,从下标k开始继续遍历(维持)。如果mid是中位数,则将中数最小值改为mid + 1,否则中数最大值为mid -1,重新遍历直到左右边界重合。

实现

public int solve (int n, int k, int[] a) {
        // write code here
        int max = 0;
        int min = 0;
        
        for (int i = 0; i < n; i++) {
            max=Math.max(max,a[i]);
            min=Math.min(min,a[i]);
        }
        
        int l=min;
        int r=max;
        
        int ans=0;
        
        while (l <= r) {
            int mid = (l + r) / 2;
            int s = 0;
            
            for (int i = 0; i < k; i++) {
                if (a[i]>=mid) {
                    s++;
                } else {
                    s--;
                }
            }
            
            if (s >= 0) {
                ans=mid;
                l=mid+1;
                
                continue;
            }
            
            boolean flag=false;
            
            int cur=0;
            
            int ps=0;
            
            for(int j=k; j<n; j++){
                if(a[j]>=mid){
                    s++;
                }else{
                    s--;
                }
                if(a[j-k]>=mid){
                    cur++;
                }else{
                    cur--;
                }
                ps = Math.min(ps, cur);
                if(s - ps >= 0){
                    flag=true;
                    break;
                }
            }
            if(flag){
                ans=mid;
                l=mid+1;
            }else{
                r=mid-1;
            }
            
        }
        
        return ans;
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值