Leetcode 1343题解决(java)

这道题是求解大小为k,且平均值大于等于阀值的子数组的数目,然后求出均值大于threshold的数组。看到题目的第一感觉是,应该先列出所有长度为k的子数组,尝试了两种方法:

第一种是,求出所有长度为k的数组,分别求出这些子数组的均值,如果大于阈值,进行累加。话不多说,直接上代码。

public int getSubArray(int[] array, int k, int threshold) {
        int length = array.length;
        int tempSum = 0;
        int result = 0;
        if (length < k) return 0;
        for (int i = 0; i <= array.length - k; ++i) {
            int[] tempData = new int[k];
            System.arraycopy(array, i, tempData, 0, k);
            if (tempData.length == 0) return 0;
            for (int m : tempData) {
                tempSum += m;
            }
            if (tempSum / k >= threshold) {
                result += 1;
            }
            tempSum = 0;
        }
        return result;
    }

本地运行ok,于是信心满满的提交了,结果没等到AC,却报了一个运行时间超时,觉得应该是两层for循环嵌套,所以运行时间很慢。所以优化一下,先得到所有的子数组,然后再去计算子数组的均值大于阈值的数组的个数(好像也没优化多少),直接上代码,欢迎各位大佬指正。

public int getSubArray1(int[] array, int k, int threshold) {
        int length = array.length;
        int tempSum = 0;
        int result = 0;
        List<int[]> tempList = new ArrayList<>();
        if (length < k) return 0;
        for (int i = 0; i <= array.length - k; ++i) {
            int[] tempData = new int[k];
            System.arraycopy(array, i, tempData, 0, k);

            if (tempData.length == 0) return 0;
            tempList.add(tempData);
        }
        if (tempList.size() == 0) return 0;
        for (int[] temp : tempList) {
            for (int n: temp) {
                tempSum += n;
            }
            if (tempSum / k >= threshold) {
                result += 1;
            }
            tempSum = 0;
        }
        return result;
    }

还是运行时间超时,好尴尬,好希望AC啊,无奈,去学习了一下大佬的代码,哇咔咔,好生简洁。

public static int numOfSubarrays3(int[] arr, int k, int threshold) {
        int sum = 0, result = 0;
        int sumTarget = k*threshold;
        for (int i = 0; i < k; i++) {
            sum += arr[i];
        }

        int adder = sum - sumTarget;

        if (adder >= 0) {
            result++;
        }

        int pos = k;
        for (int i = 0; i < arr.length-k; i++) {
            adder = adder + arr[pos] - arr[i];
            if (adder >= 0) {
                result ++;
            }
            pos++;
        }
        return result;

    }

先解释一下,当时为了找java如何得到固定长度的子数组,还用到了这个函数 arraycopy(Object src,int srcPos,Object dest, int destPos,int length),感兴趣的童鞋可以去了解一下,整个运行效率好像还ok。

再看大佬的代码,压根没有想到得到子数组,直接逆向思维,如果长度为k的数组阈值为threshold,那么如果从从整体上来看,是不是数组的和值大于k*threshold就ok了,牛皮啊,觉得想的很有意思,瞬间简单了不少啊!之后就是常规的指针一点点往前移动了,直到指针移动到数组的长度减去k,然后呢,通过判断是否大于阈值就ok,好生简单。

不过,用我那种笨办法也有一丢丢好处,就是,如果求出所有的满足条件的子数组,可能用我的笨办法好些吧!

向优秀的大佬们致敬!

猜你喜欢

转载自blog.csdn.net/weixin_45806131/article/details/106610005