349. Intersection of Two Arrays

[rimokem]

今回の問題

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

次回の問題

https://leetcode.com/problems/unique-email-addresses/

[h-masder]

こちらでは、どのような「大変さ」を想定していますか？

set への変換に伴うメモリ使用量を問題とするのであれば、「小さい方を set にする」という解法とは結びつかないので、実行時間ですかね？

もし、大きい配列を set に変換する際の実行時間を問題とするのであれば、大きい配列に対して線形探索を行うことは避けたくなります。たしかに早期終了の可能性はありますが、小さい配列をイテレーションすれば確実に早期終了します。

[rimokem]

「大変さ」としてはsetへの変換に伴うメモリ使用量を想定しています。

"片方がとても大きくて、片方がとても小さい"という状況なので、小さい方だけsetに変換し、大きい方はそのままにしています。
これで、空間計算量をO(n+m)からO(min(n,m))に減らせると考えて実装しました。

min(n,m) << n+m なので、上記の状況だと空間計算量の問題を解決できていると考えています。

[h-masder]

ありがとうございます。よくわかりました。

あとは、katataku/leetcode#12 (comment)
にある二分探索を使う方法と比べて、どういうときにどちらが実行時間が短いかなど考えてみてもいいかもしれません。

[h-masder]

今回の問題はインターセクションを求めること自体がテーマだと思うので、intersection をそのまま用いる解法は、やや問題の意図とは異なる可能性もあるかなと感じました。
そのため、別のアプローチ（例えば基本的な操作で解く方法）も併せて示せると、より理解が深まると思います。
一方で、intersection の仕様や挙動に触れている点はとても良いと思います。

[rimokem]

レビューありがとうございます。

元はコーディング面接の問題であることが、頭から抜けていました。
いざ出題されたときにこれを出すのは、流石に意図を無視していますね。

後に色々聞かれるのでしょうが、一発目に出すコードとしては以下を思い付きました。

 def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        nums1_set = set(nums1)
        nums2_set = set(nums2)
        result = []
        for n in nums1_set:
            if n in nums2_set:
                result.append(n)
        return result