【Arai60】13問目 349_Intersection of Two Arrays

[shining-ai]

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

[hayashi-ay]

全般的に良いと思います。

[hayashi-ay]

変数名をnumの代わりにnum1とかにした方がnumという一般的な変数がnums由来ではなくnums1由来であることを別途覚えておく必要がなく読みやすい気がしました。

[hayashi-ay]

whileの中は素直にif elif elseで良いと思います。例外が2ケースと本命が1ケースあるというより同じレベルの処理が3ケースあるみたいな感じだと思います。

[shining-ai]

綺麗に3ケースに分かれるので、その書き方のほうがいいかもしれませんね。
自然言語で説明する場合は、そのようにするので流れとしては自然ですね。

[shining-ai]

Odaさんのコメントを転記
in list は、list を全部確認するので、O(n) の時間がかかります。

if num in nums2:

if nums1_sorted[i] not in intersection:

このあたりです。

特に後ろの、not in は何をやっているのか分からなくなる印象を受けるので、ちょっと作り直してみましょう。

私は、この場合は、結構強く if continue を勧めます。

if A:
  B
elif C:
  D
else E:
  F
G

という形は、B を読んだ後に、G があるかどうかを確認させているからです。

一般に、early return pattern などといいます。

if nums1_sorted[i] == nums2_sorted[j]:

これは書きません。
やりたかったら、コメントをつけるか、assert を書くでしょう。

ただ、nan というのがあって、nan == nan が False です。
IEEE754 を確認しておいてください。

[shining-ai]

Odaさんのコメントを転記
nums1_sorted[i] と intersection[-1] 一緒だと思うんですが、これ、普通に変数名をつけてしまいませんか。重要なやつには名前をつけましょう。関数への切り出しなども重要なやつに名前をつけるの一種ですね。