LeetCode 2657.找到两个数组的前缀公共数组

给你两个下标从 0 开始长度为 n 的整数排列 A 和 B 。

A 和 B 的 前缀公共数组 定义为数组 C ，其中 C[i] 是数组 A 和 B 到下标为 i 之前公共元素的数目。

请你返回 A 和 B 的 前缀公共数组 。

如果一个长度为 n 的数组包含 1 到 n 的元素恰好一次，我们称这个数组是一个长度为 n 的 排列 。

示例 1：

输入：A = [1,3,2,4], B = [3,1,2,4]
输出：[0,2,3,4]
解释：i = 0：没有公共元素，所以 C[0] = 0 。
i = 1：1 和 3 是两个数组的前缀公共元素，所以 C[1] = 2 。
i = 2：1，2 和 3 是两个数组的前缀公共元素，所以 C[2] = 3 。
i = 3：1，2，3 和 4 是两个数组的前缀公共元素，所以 C[3] = 4 。
示例 2：

输入：A = [2,3,1], B = [3,1,2]
输出：[0,1,3]
解释：i = 0：没有公共元素，所以 C[0] = 0 。
i = 1：只有 3 是公共元素，所以 C[1] = 1 。
i = 2：1，2 和 3 是两个数组的前缀公共元素，所以 C[2] = 3 。

提示：

1 <= A.length == B.length == n <= 50
1 <= A[i], B[i] <= n
题目保证 A 和 B 两个数组都是 n 个元素的排列。

法一：直接模拟即可：

func findThePrefixCommonArray(A []int, B []int) []int {
    // 用int64来记录数组A和B中出现过的数字，因为最多只有50种数字
    // 位运算用无符号数
    iA := uint64(0)
    iB := uint64(0)
    n := len(A)
    C := []int{}
    for i := 0; i < n; i++ {
        iA |= 1 << A[i]
        iB |= 1 << B[i]
        C = append(C, getCommonBitNum(iA, iB))
    }
    return C
}

func getCommonBitNum(iA, iB uint64) int {
    and := iA & iB
    ans := 0
    for and != 0 {
        ans++
        and &= (and - 1)
    }
    return ans
}

此算法时间复杂度为O(n)，空间复杂度为O(1)。

法二：在计算一个数字的位数时，可以用bits.OnesCount：

func findThePrefixCommonArray(A []int, B []int) []int {
    iA := uint(0)
    iB := uint(0)
    n := len(A)
    // 我们已知C的大小，就不初始化为空了，就像c++ vector
    C := make([]int, n)
    for i := 0; i < n; i++ {
        iA |= 1 << A[i]
        iB |= 1 << B[i]
        C[i] = bits.OnesCount(iA & iB)
    }
    return C
}

此算法时间复杂度为O(n)，空间复杂度为O(1)。