问题:
滑动窗口中位数
中位数是有序序列最中间的那个数。如果序列的大小是偶数,则没有最中间的数;此时中位数是最中间的两个数的平均数。
例如:
[2,3,4],中位数是 3
[2,3],中位数是 (2 + 3) / 2 = 2.5
给你一个数组 nums,有一个大小为 k 的窗口从最左端滑动到最右端。窗口中有 k 个数,每次窗口向右移动 1 位。你的任务是找出每次窗口移动后得到的新窗口中元素的中位数,并输出由它们组成的数组。
这个问题第一次解答的时候没有理解, 只给出了求中位数
的解.
实际上还要复杂一些, 而且这个问题的核心就在我没有理解的那一部分.
画一个图更容易理解这道题的意思:
给出 nums = [1,3,-1,-3,5,3,6,7\],以及 k = 3。
中括号内即为窗口数组
[1 3 -1] -3 5 3 6 7 1
1 [3 -1 -3] 5 3 6 7 -1
1 3 [-1 -3 5] 3 6 7 -1
1 3 -1 [-3 5 3] 6 7 3
1 3 -1 -3 [5 3 6] 7 5
1 3 -1 -3 5 [3 6 7] 6
我目前的解题思路很简单, 很暴力:
1, 遍历nums的所有窗口数组
2, 对每一个窗口数组排序, 取中位值
3, 取得所有中位值, 并打印
代码:
public class 滑动窗口中位数 {
static List tempList = new ArrayList<>();
static List midNumList = new ArrayList<>();
static int[] nums = new int[]{3,4,6,8,2,7};
static int k = 2;
private static void getArrays(int[] nums, int k){
int left = 0;
int right = k-1;
getArraysUtil(nums, left, right, k);
}
private static void getArraysUtil(int[] nums, int left, int right, int k){
int temp[] = new int[k];
for (int i = 0; i < k; i++) {
temp[i] = nums[left+i];
}
tempList.add(temp);
left++;
right++;
if (right<=(nums.length-1)) {
getArraysUtil(nums, left, right, k);
}
}
private static double getMidFromArray(int args[]){
double midNum = 0.0;
List intList = new ArrayList();
for (int i = 0; i < args.length; i++) {
intList.add(args[i]);
}
Collections.sort(intList);
for (int i = 0; i < intList.size(); i++) {
}
int arraySize = intList.size();
if (arraySize%2 == 1) {
midNum = intList.get((arraySize-1)/2);
}else {
midNum = (intList.get(arraySize/2 - 1) + intList.get(arraySize/2) + 0.0) / 2 ;
}
return midNum;
}
public static void main(String[] args) {
// get all arrays
getArrays(nums, k);
// get Mid nums
for (int i = 0; i < tempList.size(); i++) {
midNumList.add(getMidFromArray(tempList.get(i)));
}
// print
System.out.println(Arrays.toString(midNumList.toArray()));
}
}
输出内容: [3.5, 5.0, 7.0, 5.0, 4.5]
此方法是最简单最基础的解答.
查询了一下其他人的解释, well.. 学习一下.
Leetcode 480.滑动窗口中位数
通过这道题倒是找到Leetcode这个网站, 有很多不错的算法题.