347. Top K Frequent Elements

[kazizi55]

https://leetcode.com/problems/top-k-frequent-elements/description/

Next: https://leetcode.com/problems/find-k-pairs-with-smallest-sums/description/

[h-masder]

sorted()をもとにreturnするとすっきりすると思います。

        num_to_frequency_ordered_by_desc = sorted(num_to_frequency.items(), key=lambda item: item[1], reverse=True)
        return [num for num, _ in num_to_frequency_ordered_by_desc[:k]]

[kazizi55]

ありがとうございます！
確かにスッキリしました。余計な dict を消すことができたので、num_to_frequency.items() の num を使うという意図がよりはっきり表現できるようになった感じがあります。
49a93eb

[oda]

sorted(num_to_frequency, key=num_to_frequency.get, reverse=True)[:k]
みたいなのは一応ありますね。
sorted は iterable を取るので、dict を入れると dict key がソートされます。

https://docs.google.com/document/d/11HV35ADPo9QxJOpJQ24FcZvtvioli770WWdZZDaLOfg/edit?tab=t.0#heading=h.dkkbub5o1tvz

[kazizi55]

こう書けるのですね。
https://discord.com/channels/1084280443945353267/1201211204547383386/1207251531041210408

class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
        num_to_frequency = defaultdict(int)
        for num in nums:
            num_to_frequency[num] += 1
        return sorted(num_to_frequency, key=num_to_frequency.get, reverse=True)[:k]

この dict key の配列を sort して返すという方が、tuple の解法よりも寄り道しないで解いている感じがあり、個人的には好みです。ご教示いただき、ありがとうございます。

[h-masder]

単純な興味ですが、見え方とか変わりましたか？

[kazizi55]

はい、明確に変わったと感じます。

1周目の時点ではアルゴリズムほぼ初心者みたいな感じだったので周りを見るのにかなり時間がかかってしまっており余裕もありませんでしたが、2周目になって色んな人のコメントを見るのに前ほど時間をかけなくても良くなり余裕が出てきた感じがあります。解法をただ覚えるのではなく、それぞれの解法を抽象化して理解するのに慣れたからなのではないかなと考えています。

もちろんまだまだ周りを見るのに時間はかかるので精進しないとですが、、

[MA-yo-TA]

やや手順増えますが一度 [(freq, num), ...] というタプルのリストにしてからソートしてそのあと num だけ取り出すとかもあり得ますかね？

frequency_and_nums = [(frequency, num) for num, frequency in num_to_frequency.items()]

[kazizi55]

ありがとうございます！
確かにその方が記述量が減って、やりたいことがより明確に表現できる感じがあって良いですね！
step 4 として追加しました。
49a93eb