Merge pull request #12 from rihib/next_permutation

Next Permutation

Author: rihib

pullrequests/next_permutation/step1.go

@@ -0,0 +1,32 @@
+//lint:file-ignore U1000 Ignore all unused code
+package nextpermutation
+
+import "sort"
+
+/*
+時間：36分
+解法を思いつくのに時間がかかってしまいました（15分ぐらい）。
+後ろから見ていって昇順になっていない箇所を見つけたら、nums[i-1]よりも大きい中で最小の値を見つけてスワップし、後ろの要素をソートするというやり方を取りました。
+*/
+func nextPermutation_step1(nums []int) {
+	for i := len(nums) - 1; i > 0; i-- {
+		if nums[i-1] < nums[i] {
+			minN := nums[i]
+			idx := i
+			for j := i; j < len(nums); j++ {
+				if nums[j] < minN && nums[i-1] < nums[j] {
+					idx = j
+					minN = nums[j]
+				}
+			}
+			nums[i-1], nums[idx] = nums[idx], nums[i-1]
+			sort.Slice(nums[i:], func(a, b int) bool {
+				return nums[i+a] < nums[i+b]
+			})
+			return
+		}
+	}
+	sort.Slice(nums, func(a, b int) bool {
+		return nums[a] < nums[b]
+	})
+}

pullrequests/next_permutation/step2.go

@@ -0,0 +1,28 @@
+//lint:file-ignore U1000 Ignore all unused code
+package nextpermutation
+
+import "sort"
+
+/*
+変数名などをリファクタしました。
+*/
+func nextPermutation_step2(nums []int) {
+	for i := len(nums) - 1; i > 0; i-- {
+		if nums[i-1] < nums[i] {
+			candidate, candidateIndex := nums[i], i
+			for j := i; j < len(nums); j++ {
+				if nums[j] < candidate && nums[i-1] < nums[j] {
+					candidate, candidateIndex = nums[j], j
+				}
+			}
+			nums[i-1], nums[candidateIndex] = nums[candidateIndex], nums[i-1]
+			sort.Slice(nums[i:], func(a, b int) bool {
+				return nums[i+a] < nums[i+b]
+			})
+			return
+		}
+	}
+	sort.Slice(nums, func(a, b int) bool {
+		return nums[a] < nums[b]
+	})
+}

pullrequests/next_permutation/step3.go

@@ -0,0 +1,30 @@
+//lint:file-ignore U1000 Ignore all unused code
+package nextpermutation
+
+/*
+これまでの解法がソートを使っていたために時間計算量がO(n log n)になってしまったわけですが、他の人の回答を見てO(n)で解ける方法に変えました。
+nums[i-1]よりも後ろは昇順になっているのを利用し、リバースすればソートできるということに気づきませんでした。
+*/
+func nextPermutation_step3(nums []int) {
+	if len(nums) < 2 {
+		return
+	}
+	i := len(nums) - 2
+	for i >= 0 && nums[i] >= nums[i+1] {
+		i--
+	}
+	if i >= 0 {
+		j := len(nums) - 1
+		for nums[j] <= nums[i] {
+			j--
+		}
+		nums[i], nums[j] = nums[j], nums[i]
+	}
+	reverse(nums[i+1:])
+}
+
+func reverse(nums []int) {
+	for i, j := 0, len(nums)-1; i < j; i, j = i+1, j-1 {
+		nums[i], nums[j] = nums[j], nums[i]
+	}
+}