📑 题目:89. 格雷编码

🚀 本题 LeetCode 传送门

题目大意

格雷编码是一个二进制数字系统,在该系统中,两个连续的数值仅有一个位数的差异。给定一个代表编码总位数的非负整数 n,打印其格雷编码序列。格雷编码序列必须以 0 开头。

解题思路

  • 输出 n 位格雷码
  • 格雷码生成规则:以二进制为0值的格雷码为第零项,第一次改变最右边的位元,第二次改变右起第一个为1的位元的左边位元,第三、四次方法同第一、二次,如此反复,即可排列出 n 个位元的格雷码。
  • 可以直接模拟,也可以用递归求解。

代码

  1. package leetcode
  2. // 解法一 递归方法,时间复杂度和空间复杂度都较优
  3. func grayCode(n int) []int {
  4. if n == 0 {
  5. return []int{0}
  6. }
  7. res := []int{}
  8. num := make([]int, n)
  9. generateGrayCode(int(1<<uint(n)), 0, &num, &res)
  10. return res
  11. }
  12. func generateGrayCode(n, step int, num *[]int, res *[]int) {
  13. if n == 0 {
  14. return
  15. }
  16. *res = append(*res, convertBinary(*num))
  17. if step%2 == 0 {
  18. (*num)[len(*num)-1] = flipGrayCode((*num)[len(*num)-1])
  19. } else {
  20. index := len(*num) - 1
  21. for ; index >= 0; index-- {
  22. if (*num)[index] == 1 {
  23. break
  24. }
  25. }
  26. if index == 0 {
  27. (*num)[len(*num)-1] = flipGrayCode((*num)[len(*num)-1])
  28. } else {
  29. (*num)[index-1] = flipGrayCode((*num)[index-1])
  30. }
  31. }
  32. generateGrayCode(n-1, step+1, num, res)
  33. return
  34. }
  35. func convertBinary(num []int) int {
  36. res, rad := 0, 1
  37. for i := len(num) - 1; i >= 0; i-- {
  38. res += num[i] * rad
  39. rad *= 2
  40. }
  41. return res
  42. }
  43. func flipGrayCode(num int) int {
  44. if num == 0 {
  45. return 1
  46. }
  47. return 0
  48. }
  49. // 解法二 直译
  50. func grayCode1(n int) []int {
  51. var l uint = 1 << uint(n)
  52. out := make([]int, l)
  53. for i := uint(0); i < l; i++ {
  54. out[i] = int((i >> 1) ^ i)
  55. }
  56. return out
  57. }