Create 46. Permutations.md

[tokuhirat]

This Problem
46. Permutations
Next Ploblem
78. Subsets
言語: Python

[oda]

私はこの部分が特に気に食わなくて、

• if else を入れ替えます。
• else のインデントを下げます。
• break を先に書いて、yield を for の外に追い出します。

とすると思います。

[tokuhirat]

私も読んでいる時 if else の部分は微妙だと思っていました。
yield を for の外に追い出すのは思いつきませんでした。while の役割が明確になってわかりやすくなると感じます。

class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)
        indices = list(range(n))
        cycles = list(range(n, 0, -1))
        results = []
        results.append([nums[i] for i in indices])
        while True:
            for i in reversed(range(n)):
                cycles[i] -= 1
                if cycles[i] > 0:
                    j = cycles[i]
                    indices[i], indices[-j] = indices[-j], indices[i]
                    break
                indices[i:] = indices[i + 1 :] + indices[i : i + 1]
                cycles[i] = n - i
            else:
                break
            results.append([nums[i] for i in indices])
        return results

[oda]

2つの append をまとめてループの先頭に回すこともできますね。

                    j = cycles[i]
                    indices[i], indices[-j] = indices[-j], indices[i]

これも for の外に追い出せます。(どちらがいいかは難しいところ。)

[tokuhirat]

そうですね。do-while のような処理をまとめる形で書くのが苦手だなと感じています。
for の仕事として適切な i を見つけるところで終わるのか、indices の変更までしてもらうのかというところですね。
cycles[i] == 0 の時 for 内で indices を整えている点、cycles[i] > 0 で break してから for の処理がやや複雑な点（else:break がある）、を考えると for 内に書くのが個人的には自然に思います。

[nodchip]

使用中の num を set に入れておくと、時間計算量は下がりそうです。ただ、 1 <= nums.length <= 6 のため、こちらのほうが速いかもしれません。