🔗 https://leetcode.com/problems/find-the-prefix-common-array-of-two-arrays

题目

• 给两个数组 A 和 B，是 n 的全排列
• 返回数组 C，表示在 index 及之前，A 和 B 有多少个相同的数

思路

• hashset ，遍历 index，判断此时 A 和 B 的共同数字有几个
• frequency 统计，遍历 index，A index 和 B index 的 frequency++，若 frequency 为 2，则计数

代码

class Solution {
public:vector<int> solution1(vector<int>& A, vector<int>& B) {unordered_set<int> s;vector<int> C(A.size());for (int i = 0; i < A.size(); i++) {s.insert(A[i]);for (int j = 0; j <= i; j++) {if (s.count(B[j])) C[i]++;}}return C;}vector<int> solution2(vector<int>& A, vector<int>& B) {unordered_map<int, int> m;vector<int> C(A.size());for (int i = 0; i < A.size(); i++) {m[A[i]]++;m[B[i]]++;if (i != 0) C[i] = C[i-1];if (m[A[i]] == 2) C[i]++;if (A[i] == B[i]) continue;if (m[B[i]] == 2) C[i]++;}return C;}vector<int> findThePrefixCommonArray(vector<int>& A, vector<int>& B) {//return solution1(A, B);return solution2(A, B);}
};