Go语言：高效生成N字符密码组合的实现指南

package main

import "fmt"

// NAryProduct 生成给定字符集input的所有n字符组合
// 参数:
//   input: 包含所有可用字符的字符串，例如 "ABCDE"
//   n: 期望生成的组合长度
// 返回值:
//   一个字符串切片，包含所有生成的n字符组合
func NAryProduct(input string, n int) []string {
    // 边界条件：如果n小于等于0，则没有有效的组合，返回nil
    if n <= 0 {
        return nil
    }

    // 初始化prod切片，存储长度为1的组合
    // 每个字符本身就是一个长度为1的组合
    prod := make([]string, len(input))
    for i, char := range input {
        prod[i] = string(char)
    }

    // 从长度2开始，迭代构建直到长度为n的组合
    for i := 1; i < n; i++ {
        // next切片用于存储当前迭代（长度为i+1）的所有组合
        // 预估容量：字符集大小 * 前一轮组合的数量
        next := make([]string, 0, len(input)*len(prod))

        // 遍历前一轮（长度为i）的所有组合
        for _, word := range prod {
            // 将前一轮的每个组合与字符集中的每个字符拼接
            for _, char := range input {
                next = append(next, word+string(char))
            }
        }
        // 更新prod为新生成的组合，用于下一轮迭代
        prod = next
    }

    return prod
}

func main() {
    charSet := "ABCDE"
    length := 2
    passwords := NAryProduct(charSet, length)
    fmt.Printf("生成所有长度为 %d 的组合，字符集为 '%s':\n", length, charSet)
    for _, p := range passwords {
        fmt.Println(p)
    }
    fmt.Printf("总共生成了 %d 个组合。\n\n", len(passwords))

    length = 3
    passwords = NAryProduct("ab", length)
    fmt.Printf("生成所有长度为 %d 的组合，字符集为 '%s':\n", length, "ab")
    for _, p := range passwords {
        fmt.Println(p)
    }
    fmt.Printf("总共生成了 %d 个组合。\n", len(passwords))
}

生成所有长度为 2 的组合，字符集为 'ABCDE':
AA
AB
AC
AD
AE
BA
BB
BC
BD
BE
CA
CB
CC
CD
CE
DA
DB
DC
DD
DE
EA
EB
EC
ED
EE
总共生成了 25 个组合。

生成所有长度为 3 的组合，字符集为 'ab':
aaa
aab
aba
abb
baa
bab
bba
bbb
总共生成了 8 个组合。

package main

import "fmt"

// GeneratePasswords 作为一个生成器，通过 channel 逐个发送生成的密码
// 参数:
//   input: 包含所有可用字符的字符串
//   n: 期望生成的组合长度
// 返回值:
//   一个只读的字符串 channel
func GeneratePasswords(input string, n int) <-chan string {
    out := make(chan string) // 创建一个无缓冲或有缓冲的channel

    go func() { // 在一个goroutine中执行生成逻辑
        defer close(out) // 确保在所有密码发送完毕后关闭channel

        if n <= 0 {
            return
        }

        // 递归辅助函数，用于生成组合并发送到channel
        var generate func(current string, k int)
        generate = func(current string, k int) {
            if k == n { // 达到目标长度
                out <- current // 将完整组合发送到channel
                return
            }
            // 递归地为下一个字符位置选择字符
            for _, char := range input {
                generate(current+string(char), k+1)
            }
        }

        generate("", 0) // 从空字符串和长度0开始生成
    }()

    return out
}

func main() {
    charSet := "ABCDE"
    length := 2

    fmt.Printf("使用生成器生成所有长度为 %d 的组合，字符集为 '%s':\n", length, charSet)
    count := 0
    for password := range GeneratePasswords(charSet, length) {
        fmt.Println(password)
        count++
    }
    fmt.Printf("总共生成了 %d 个组合。\n", count)
}