蚂蚁0313笔试

蚂蚁0313笔试

#蚂蚁求职进展汇总#

算法大模型方向 选择好多不确定的（特别是多选）

算法题部分倒还行

第一题

01序列问所有的子序列的mex之和，分mex=0，mex=1，mex=2动态规划即可

第二题

决策树问特征重要性，调库
X,y= transform(eval(input()))
clf = sklearn.tree.DecisionTreeClassifier(random_state=42)
clf.fit(X,y)
clf.tree_.compute_feature_importances(normalize=False)
这个代码提示时有时无的，很搞心态

第三题

01序列问所有的子序列的min(0cnt,1cnt)之和
首先定义变量x=1的数目-0的数目，如果序列中x>=0则对答案贡献0cnt否则贡献1cnt
使用树状数组记录每个前缀和的x，0的数目，序列个数，就能O（nlogn）计算x>=0时候对答案贡献
反之，让序列中的0变1，1变0重新计算就能计算x<0时候对答案贡献（注意取等条件）