题目
一个长度为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;
}