一开始直接不管permutation这个设定直接干下去了== 写的超臭超长
不过其实有permutation这个设定
好像可以只用一个count array
count[x]==2的时候就代表有进交集了
哀
def findThePrefixCommonArray(self, A: List[int], B: List[int]) -> List[int]:
a_cnt = [0 for _ in range(51)]
b_cnt = [0 for _ in range(51)]
inter = [0 for _ in range(51)]
cur_ans = 0
ans = []
for i in range(len(A)):
a_cnt[A[i]] += 1
b_cnt[B[i]] += 1
if A[i]==B[i]:
cur_ans += 1
inter[A[i]] += 1
else:
cur_ans, inter[A[i]] = cur_ans+min(a_cnt[A[i]],
b_cnt[A[i]])-inter[A[i]], min(a_cnt[A[i]], b_cnt[A[i]])
cur_ans, inter[B[i]] = cur_ans+min(a_cnt[B[i]],
b_cnt[B[i]])-inter[B[i]], min(a_cnt[B[i]], b_cnt[B[i]])
ans.append(cur_ans)
return ans