349. Intersection of Two Arrays

[tarinaihitori]

https://leetcode.com/problems/intersection-of-two-arrays/description/

[liquo-rice]

空間計算量がO(min(n, m))の方法でできますか？

[tarinaihitori]

nums1とnums2の長さを比較して、短い方だけをsetに格納すれば
O(min(n, m))の空間計算量で書くことができると思います。以下のような感じでしょうか。

class Solution:
    def intersection(nums1, nums2):
        if len(nums1) > len(nums2):
            nums1, nums2 = nums2, nums1

        set_nums = set(nums1)  # 小さい方の配列をセットに変換
        result = set()

        for num in nums2:
            if num in set_nums:
                result.add(num)

        return list(result)

[liquo-rice]

setを一つだけ（set_numsのみ）使ってできますか？

[tarinaihitori]

結果を返すresultをsetからリストに変えて、重複を見つけたら、set_numsからremoveするようにしました。
これ以外だと、時間計算量はO(n * m)になりますが、二重ループでそれぞれ値を取り出して値が同じならsetに格納する方法が思いつきました。

class Solution:
    def intersection(self, nums1, nums2):
        if len(nums1) > len(nums2):
            nums1, nums2 = nums2, nums1
        
        set_nums = set(nums1) 
        result = [] 

        for num in nums2:
            if num in set_nums:
                result.append(num)
                set_nums.remove(num)  

        return result

[liquo-rice]

良さそうです！

[hroc135]

Pythonは引数が値渡しなので気にしなくてもいいと思いますが、参照渡しの言語なら下のようにすると参照透過性が失われないと思いました

if len(nums1) > len(nums2):
    return self.intersection(nums2, nums1)

[seal-azarashi]

好みですが、それぞれの set を変数化する必要もないのかなと思いました。そのぐらい読みやすいです。

[hayashi-ay]

良いと思います

[oda]

他、両方を sort するという手もあるでしょう。
そこから、merge sort ようなことをする作戦や、片方の値でバイナリサーチをする作戦などがあるでしょう。

あと、他の人のコードを読むほうが練習として大事なので、Discord サーバー内を漁るといいと思います。