贝壳找房秋招机器学习笔试卷

奇特区间数

给出一个大小为n的数组a和整数t,定义区间[l,r](0<=l<r<=n-1),若存在下标i,j(l<=i<j<=r)属于区间[l,r],且a[i]异或a[j]=t,那么称[l,r]是非奇特区间,如不存在,则[l,r]是奇特区间,求a数组里的奇特区间个数。

示例1

输入例子:[2,4,8],6
输出例子:1
例子说明:因为2异或4为6,等于t,所以只有区间[1,2]是奇特区间,对应的数组是[4,8],数组不能为[2,4],数组也不能为[2,4,8],因为里面包含了2和4

代码:

class Solution:
    def section(self , a , t ):
        # write code here
        n = len(a)
        dp = [-1] * n
        res = 0
        maps_index = {} #记录当前出现的数最后一个下标
        maps_index[a[0]] = 0
        for i in range(1,n):
            maps_index[a[i]] = i
            target = a[i] ^ t # 异或是可恢复的
            dp[i] = dp[i-1]
            if target in maps_index:
                dp[i] = max(dp[i],maps_index[target]) # 比较前一个的最早的下标和自己的奇异数下标哪个大
            res += i-dp[i]-1
        return res

使用动态规划,dp[i]代表以a[i]结尾的区间左端点最多在哪,这个值一定大于等于dp[i-1]所代表的值,那么就用两数之和的想法,维护一个hashmap找到在i左端的最近的一个奇异值,与dp[i-1]相比较。最后以a[i]结尾的区间最长为i-dp[i]最短为2

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务