2026年AI算法工程师职位面试问题及答案参考.docxVIP

第PAGE页共NUMPAGES页

2026年AI算法工程师职位面试问题及答案参考

一、编程与算法基础（共5题，每题10分）

题目1（10分）：

请用Python实现快速排序算法，并说明其时间复杂度和空间复杂度。假设输入数组为`[8,3,1,7,0,10,2]`，输出排序后的数组。

答案：

python

defquick_sort(arr):

iflen(arr)=1:

returnarr

pivot=arr[len(arr)//2]

left=[xforxinarrifxpivot]

middle=[xforxinarrifx==pivot]

right=[xforxinarrifxpivot]

returnquick_sort(left)+middle+quick_sort(right)

arr=[8,3,1,7,0,10,2]

sorted_arr=quick_sort(arr)

print(sorted_arr)#输出:[0,1,2,3,7,8,10]

解析：

快速排序的平均时间复杂度为O(nlogn)，最坏情况为O(n2)。空间复杂度为O(logn)，因为递归调用栈的深度为logn。上述实现采用列表切片，空间效率较低，实际应用中可优化为原地排序以降低空间复杂度。

题目2（10分）：

给定一个字符串`s`和一个目标字符串`t`，请编写函数返回`t`在`s`中第一次出现的起始索引。如果未找到，返回`-1`。例如，`s=helloworld`,`t=world`，输出应为`6`。

答案：

python

deffind_substring(s,t):

ifnott:

return0

len_s,len_t=len(s),len(t)

foriinrange(len_s-len_t+1):

ifs[i:i+len_t]==t:

returni

return-1

s=helloworld

t=world

print(find_substring(s,t))#输出:6

解析：

该算法采用滑动窗口方法，时间复杂度为O(n)，空间复杂度为O(1)。通过逐个比较子串与目标字符串，找到首次匹配的位置。

题目3（10分）：

请实现一个函数，输入一个正整数`n`，返回其二进制表示中`1`的个数。例如，`n=11`（二进制为`1011`），输出应为`3`。

答案：

python

defcount_bits(n):

count=0

whilen:

count+=n1

n=1

returncount

n=11

print(count_bits(n))#输出:3

解析：

该方法利用位运算，每次将`n`右移一位并统计最低位的`1`的数量，直到`n`为`0`。时间复杂度为O(logn)，空间复杂度为O(1)。

题目4（10分）：

给定一个无重复元素的数组`nums`和一个目标值`target`，请返回所有相加等于`target`的`nums`的组合。例如，`nums=[2,3,6,7]`,`target=7`，输出应为`[[2,5],[7]]`。

答案：

python

deffour_sum(nums,target):

nums.sort()

res=[]

n=len(nums)

foriinrange(n-3):

ifi0andnums[i]==nums[i-1]:

continue

forjinrange(i+1,n-2):

ifji+1andnums[j]==nums[j-1]:

continue

left,right=j+1,n-1

whileleftright:

total=nums[i]+nums[j]+nums[left]+nums[right]

iftotal==target:

res.append([nums[i],nums[j],nums[left],nums[right]])

whileleftrightandnums[left]==nums[left+1]:

left+=1

whileleftrightandnums[right]==nums[right-1]:

right-=1

left+=1

rig