大厂秋招真题【前缀和】美团20230826秋招T5-平均数为k的最长连续子数组

本文主要是介绍大厂秋招真题【前缀和】美团20230826秋招T5-平均数为k的最长连续子数组,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

文章目录

  • 【前缀和】美团20230826秋招T5-平均数为k的最长连续子数组
  • 题目描述与示例
    • 题目描述
    • 输入描述
    • 输出描述
    • 示例
      • 输入
      • 输出
      • 说明
  • 解题思路
  • 代码
    • Python
    • Java
    • C++
    • 时空复杂度
  • 华为OD算法/大厂面试高频题算法练习冲刺训练

【前缀和】美团20230826秋招T5-平均数为k的最长连续子数组

题目描述与示例

题目描述

给定n个正整数组成的数组,求平均数正好等于k最长连续子数组的长度。

输入描述

第一行输入两个正整数nk,用空格隔开。

第二行输入n个正整数ai,用来表示数组。

1 <= n <= 200000
1 < = k, ai <= 10^9

输出描述

如果不存在任何一个连续子数组的平均数等于k,则输出-1

否则输出平均数正好等于k的最长连续子数组的长度。

示例

输入

5 2
1 3 2 4 1

输出

3

说明

取前三个数即可,平均数为2

解题思路

求连续子数组的平均数是一个比较难处理的过程,可以先做一步转换,把原数组nums中每一个元素都减去k得到一个新数组nums_new,那么题目就变成了求和为0的最长连续子数组的长度了

由于nums_new中的元素有负数也有正数,该解题过程不能够用滑动窗口来解决,而应该使用前缀和结合哈希表来解决问题。

对数组nums_new构建前缀和数组pre_sum_lst,由于要求计算和为0的连续子数组,故我们仅需要找到pre_sum_lst两个距离最远的相等元素。该过程可以通过一边遍历pre_sum_lst中的元素pre_sum,一边构建哈希表dic来进行,若

  • pre_sum不位于哈希表中,说明它是首次出现,将其下标i记录在哈希表中
  • pre_sum已位于哈希表中,说明它之前已经出现过了,第一次出现的下标为dic[pre_sum],那么当前和为0的连续子数组的长度为i-dic[pre_sum],将其与ans比较并更新。

上述过程的核心代码如下

for i, pre_sum in enumerate(pre_sum_lst):if pre_sum not in dic:dic[pre_sum] = ielse:ans = max(ans, i-dic[pre_sum])

代码

Python

# 题目:【前缀和】美团2023秋招-平均数为k的最长连续子数组
# 作者:闭着眼睛学数理化
# 算法:前缀和/哈希表
# 代码有看不懂的地方请直接在群上提问from itertools import accumulate# 输入数组长度n,平均值k
n, k = map(int, input().split())
# 对输入的数组nums中的每一个元素进行-k的预处理,得到nums_new数组
nums_new = list(map(lambda x: int(x)-k, input().split()))
# 构建前缀和数组,注意首位需要填充一个0,表示不选取任何数字的前缀和
pre_sum_lst = [0] + list(accumulate(nums_new))# 构建哈希表,储存每个前缀和首次出现的下标
dic = dict()
# 初始化答案为-1
ans = -1
# 遍历前缀和数组中的所有下标和元素
for i, pre_sum in enumerate(pre_sum_lst):# 若pre_sum没有在哈希表中出现过# 则记录其第一次出现的下标if pre_sum not in dic:dic[pre_sum] = i# 若pre_sum在哈希表中出现过# 则计算当前下标i和其第一次出现下标dic[pre_sum]之差# 用于更新答案anselse:ans = max(ans, i-dic[pre_sum])print(ans)

Java

import java.util.HashMap;
import java.util.Scanner;public class Main {public static void main(String[] args) {Scanner scanner = new Scanner(System.in);int n = scanner.nextInt();int k = scanner.nextInt();int[] nums = new int[n];for (int i = 0; i < n; i++) {nums[i] = scanner.nextInt() - k; // 预处理,将每个元素减去 k}HashMap<Integer, Integer> prefixSumIndices = new HashMap<>();prefixSumIndices.put(0, -1); // 初始化前缀和为0的下标为-1int prefixSum = 0;int maxLength = -1;for (int i = 0; i < n; i++) {prefixSum += nums[i];if (prefixSumIndices.containsKey(prefixSum)) {int startIndex = prefixSumIndices.get(prefixSum) + 1; // 子数组的起始下标int currentLength = i - startIndex + 1; // 当前子数组长度maxLength = Math.max(maxLength, currentLength);} else {prefixSumIndices.put(prefixSum, i);}}System.out.println(maxLength);}
}

C++

#include <iostream>
#include <vector>
#include <unordered_map>using namespace std;int main() {int n, k;cin >> n >> k;vector<int> nums(n);for (int i = 0; i < n; ++i) {cin >> nums[i];nums[i] -= k; // 预处理,将每个元素减去 k}unordered_map<int, int> prefixSumIndices;prefixSumIndices[0] = -1; // 初始化前缀和为0的下标为-1int prefixSum = 0;int maxLength = -1;for (int i = 0; i < n; ++i) {prefixSum += nums[i];if (prefixSumIndices.find(prefixSum) != prefixSumIndices.end()) {int startIndex = prefixSumIndices[prefixSum] + 1; // 子数组的起始下标int currentLength = i - startIndex + 1; // 当前子数组长度maxLength = max(maxLength, currentLength);} else {prefixSumIndices[prefixSum] = i;}}cout << maxLength << endl;return 0;
}

时空复杂度

时间复杂度:O(N)。构建nums_new,前缀和数组pre_sum_lst,遍历前缀和数组pre_sum_lst,均只需一次遍历。

空间复杂度:O(N)。主要为前缀和数组pre_sum_lst和哈希表dic所占空间。


华为OD算法/大厂面试高频题算法练习冲刺训练

  • 华为OD算法/大厂面试高频题算法冲刺训练目前开始常态化报名!目前已服务100+同学成功上岸!

  • 课程讲师为全网50w+粉丝编程博主@吴师兄学算法 以及小红书头部编程博主@闭着眼睛学数理化

  • 每期人数维持在20人内,保证能够最大限度地满足到每一个同学的需求,达到和1v1同样的学习效果!

  • 60+天陪伴式学习,40+直播课时,300+动画图解视频,300+LeetCode经典题,200+华为OD真题/大厂真题,还有简历修改、模拟面试、专属HR对接将为你解锁

  • 可上全网独家的欧弟OJ系统练习华子OD、大厂真题

  • 可查看链接 大厂真题汇总 & OD真题汇总(持续更新)

  • 绿色聊天软件戳 od1336了解更多

这篇关于大厂秋招真题【前缀和】美团20230826秋招T5-平均数为k的最长连续子数组的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/252818

相关文章

C++初始化数组的几种常见方法(简单易懂)

《C++初始化数组的几种常见方法(简单易懂)》本文介绍了C++中数组的初始化方法,包括一维数组和二维数组的初始化,以及用new动态初始化数组,在C++11及以上版本中,还提供了使用std::array... 目录1、初始化一维数组1.1、使用列表初始化(推荐方式)1.2、初始化部分列表1.3、使用std::

C++ Primer 多维数组的使用

《C++Primer多维数组的使用》本文主要介绍了多维数组在C++语言中的定义、初始化、下标引用以及使用范围for语句处理多维数组的方法,具有一定的参考价值,感兴趣的可以了解一下... 目录多维数组多维数组的初始化多维数组的下标引用使用范围for语句处理多维数组指针和多维数组多维数组严格来说,C++语言没

最长公共子序列问题的深度分析与Java实现方式

《最长公共子序列问题的深度分析与Java实现方式》本文详细介绍了最长公共子序列(LCS)问题,包括其概念、暴力解法、动态规划解法,并提供了Java代码实现,暴力解法虽然简单,但在大数据处理中效率较低,... 目录最长公共子序列问题概述问题理解与示例分析暴力解法思路与示例代码动态规划解法DP 表的构建与意义动

关于最长递增子序列问题概述

《关于最长递增子序列问题概述》本文详细介绍了最长递增子序列问题的定义及两种优化解法:贪心+二分查找和动态规划+状态压缩,贪心+二分查找时间复杂度为O(nlogn),通过维护一个有序的“尾巴”数组来高效... 一、最长递增子序列问题概述1. 问题定义给定一个整数序列,例如 nums = [10, 9, 2

Java 字符数组转字符串的常用方法

《Java字符数组转字符串的常用方法》文章总结了在Java中将字符数组转换为字符串的几种常用方法,包括使用String构造函数、String.valueOf()方法、StringBuilder以及A... 目录1. 使用String构造函数1.1 基本转换方法1.2 注意事项2. 使用String.valu

JAVA中整型数组、字符串数组、整型数和字符串 的创建与转换的方法

《JAVA中整型数组、字符串数组、整型数和字符串的创建与转换的方法》本文介绍了Java中字符串、字符数组和整型数组的创建方法,以及它们之间的转换方法,还详细讲解了字符串中的一些常用方法,如index... 目录一、字符串、字符数组和整型数组的创建1、字符串的创建方法1.1 通过引用字符数组来创建字符串1.2

vue如何监听对象或者数组某个属性的变化详解

《vue如何监听对象或者数组某个属性的变化详解》这篇文章主要给大家介绍了关于vue如何监听对象或者数组某个属性的变化,在Vue.js中可以通过watch监听属性变化并动态修改其他属性的值,watch通... 目录前言用watch监听深度监听使用计算属性watch和计算属性的区别在vue 3中使用watchE

hdu2241(二分+合并数组)

题意:判断是否存在a+b+c = x,a,b,c分别属于集合A,B,C 如果用暴力会超时,所以这里用到了数组合并,将b,c数组合并成d,d数组存的是b,c数组元素的和,然后对d数组进行二分就可以了 代码如下(附注释): #include<iostream>#include<algorithm>#include<cstring>#include<stack>#include<que

poj2406(连续重复子串)

题意:判断串s是不是str^n,求str的最大长度。 解题思路:kmp可解,后缀数组的倍增算法超时。next[i]表示在第i位匹配失败后,自动跳转到next[i],所以1到next[n]这个串 等于 n-next[n]+1到n这个串。 代码如下; #include<iostream>#include<algorithm>#include<stdio.h>#include<math.

poj3261(可重复k次的最长子串)

题意:可重复k次的最长子串 解题思路:求所有区间[x,x+k-1]中的最小值的最大值。求sa时间复杂度Nlog(N),求最值时间复杂度N*N,但实际复杂度很低。题目数据也比较水,不然估计过不了。 代码入下: #include<iostream>#include<algorithm>#include<stdio.h>#include<math.h>#include<cstring