推 PogChampLUL: 大師 01/14 23:39
→ sustainer123: 大師 01/14 23:39
推 Meaverzt: 別卷了 01/14 23:53
一開始直接不管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
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 125.229.37.69 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1736869096.A.EF2.html