小算法题库及答案.docxVIP

小算法题库及答案

给定一个整数数组nums和一个整数k，找出数组中所有连续子数组的和等于k且长度为偶数的子数组数目。

解题思路：连续子数组的和问题通常用前缀和解决。定义前缀和数组preSum，其中preSum[i]表示前i个元素的和（preSum[0]=0，preSum[1]=nums[0]）。子数组nums[j...i-1]的和为preSum[i]-preSum[j]。要求和为k，即preSum[j]=preSum[i]-k。同时，子数组长度为i-j（i和j为前缀和数组索引），需为偶数，即i和j奇偶性相同。因此维护两个哈希表，分别记录前缀和为s时索引为偶数和奇数的出现次数。遍历前缀和数组时，根据当前索引的奇偶性，查询另一个哈希表中preSum[i]-k的出现次数，累加到结果中。初始时preSum[0]=0（索引0为偶数），偶数哈希表初始化为{0:1}。

代码实现（Python）：

```python

fromcollectionsimportdefaultdict

defcount_even_length_subarrays(nums,k):

pre_sum=0

even_counts=defaultdict(int)记录索引为偶数的前缀和出现次数

odd_counts=defaultdict(int)记录索引为奇数的前缀和出现次数

even_counts[0]=1初始前缀和0，索引0（偶数）

result=0

foriinrange(len(nums)):

pre_sum+=nums[i]

current_index=i+1前缀和数组的索引（从1开始）

ifcurrent_index%2==0:当前索引为偶数，需找奇数索引的j

target=pre_sumk

result+=odd_counts.get(target,0)

even_counts[pre_sum]+=1更新偶数哈希表

else:当前索引为奇数，需找偶数索引的j

target=pre_sumk

result+=even_counts.get(target,0)

odd_counts[pre_sum]+=1更新奇数哈希表

returnresult

```

给定一个仅由小写字母组成的字符串s，统计其中满足首尾字符相同且长度至少为3的子串数目。

解题思路：子串需满足首尾字符相同且长度≥3（即结束位置j≥起始位置i+2）。直接枚举所有i和j的时间复杂度为O(n2)，优化方法是按字符分组记录位置。对每个字符的位置列表positions，用双指针法统计满足j≥i+2的(i,j)对数：对每个i，找到第一个j≥i+2的位置，之后的所有j都符合条件，数目为列表长度减去右指针位置。

代码实现（Python）：

```python

fromcollectionsimportdefaultdict

defcount_same_end_substrings(s):

char_pos=defaultdict(list)

n=len(s)

foridx,charinenumerate(s):

char_pos[char].append(idx)记录每个字符的所有出现位置

result=0

forchar,positionsinchar_pos.items():

m=len(positions)

right=0

forleftinrange(m):

找到第一个right满足positions[right]≥positions[left]+2

whilerightmandpositions[right]positions[left]+2:

right+=1

result+=(mright)右侧所有位置都符合条件

r