牛客NC343 和大于等于K的最短子数组【困难 前缀和 Java/Go】

题目

题目链接:

https://www.nowcoder.com/practice/3e1fd3d19fb0479d94652d49c7e1ead1

思路

复制代码
本答案利用前缀和解答,Java,Go答案通过,但是同样的代码用PHP的话有一个测试用例超时
应该还有更优秀的答案,后面找到更优答案再更新博客

Java代码

java 复制代码
import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param nums int整型ArrayList
     * @param k int整型
     * @return int整型
     */
    public int shortestSubarray (ArrayList<Integer> nums, int k) {
        //前缀和
        int n = nums.size();
        int[] sum = new int[n + 1];
        for (int i = 0; i < n; i++) {
            sum[i + 1] = sum[i] + nums.get(i);
             sum[i + 1]%=1000000007;  //关键
            if (nums.get(i) == k) return 1;
        }

        int ans = 0x7fffffff;
        for (int i = 0; i <= n ; i++) {
            if (sum[i] >= k) {
                ans = Math.min(ans, i);
                for (int j = i ; j >= 0 ; j--) {
                    int diff = sum[i] - sum[j];
                    if (diff >= k) {
                        ans = Math.min(ans, i - j);
                        break;
                    }
                }
            }

        }
        if (ans == 0x7fffffff)
            return -1;

        return ans;
    }
}

Go代码

go 复制代码
package main

/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 *
 *
 * @param nums int整型一维数组
 * @param k int整型
 * @return int整型
 */
func shortestSubarray(nums []int, k int) int {
	//前缀和
	n := len(nums)
	ans := 0x7fffffff

	sum := make([]int, n+1)
	for i := 0; i < n; i++ {
		sum[i+1] = sum[i] + nums[i]
		if nums[i] == k {
			return 1
		}
	}

	for i := 0; i <= n; i++ {
		if sum[i] >= k { //剪枝
			if sum[i] >= k {
				if i < ans {
					ans = i
				}
			}
			for j := i; j >= 0; j-- {
				diff := sum[i] - sum[j]
				if diff >= k {
					if i-j < ans {
						ans = i - j
					}
					break
				}
			}
		}
	}

	if ans == 0x7fffffff {
		return -1
	}

	return ans
}
相关推荐
gaoshou451 小时前
代码随想录训练营第三十一天|LeetCode56.合并区间、LeetCode738.单调递增的数字
数据结构·算法
自信的小螺丝钉1 小时前
Leetcode 240. 搜索二维矩阵 II 矩阵 / 二分
算法·leetcode·矩阵
KING BOB!!!2 小时前
Leetcode高频 SQL 50 题(基础版)题目记录
sql·mysql·算法·leetcode
我是渣哥3 小时前
Java String vs StringBuilder vs StringBuffer:一个性能优化的探险故事
java·开发语言·jvm·后端·算法·职场和发展·性能优化
THMAIL3 小时前
机器学习从入门到精通 - 机器学习调参终极手册:网格搜索、贝叶斯优化实战
人工智能·python·算法·机器学习·支持向量机·数据挖掘·逻辑回归
lytk993 小时前
矩阵中寻找好子矩阵
线性代数·算法·矩阵
珊瑚怪人4 小时前
算法随笔(一)
算法
晚安里4 小时前
JVM相关 4|JVM调优与常见参数(如 -Xms、-Xmx、-XX:+PrintGCDetails) 的必会知识点汇总
java·开发语言·jvm·后端·算法
THMAIL4 小时前
机器学习从入门到精通 - 集成学习核武器:随机森林与XGBoost工业级应用
人工智能·python·算法·随机森林·机器学习·集成学习·sklearn
要做朋鱼燕4 小时前
【C++】迭代器详解与失效机制
开发语言·c++·算法