一開始直接不管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