Go regexp 包中 FindAll 方法的 n 参数详解

package main

import (
    "fmt"
    "regexp"
)

func main() {
    // 编译一个简单的正则表达式，匹配字符 'a'
    reA, err := regexp.Compile("a")
    if err != nil {
        fmt.Println("Error compiling regex reA:", err)
        return
    }

    testString := "aaaaa, bbb aaa ccc"

    fmt.Println("--- 使用 FindAllString 示例 ---")

    // 情况 1: n = -1 (返回所有匹配)
    fmt.Println("\n匹配所有 'a' (n = -1):")
    matchesAll := reA.FindAllString(testString, -1)
    fmt.Printf("找到 %d 个匹配项: %v\n", len(matchesAll), matchesAll)
    // 预期输出: 找到 8 个匹配项: [a a a a a a a a]

    // 情况 2: n = 1 (返回前 1 个匹配)
    fmt.Println("\n匹配前 1 个 'a' (n = 1):")
    matchesOne := reA.FindAllString(testString, 1)
    fmt.Printf("找到 %d 个匹配项: %v\n", len(matchesOne), matchesOne)
    // 预期输出: 找到 1 个匹配项: [a]

    // 情况 3: n = 3 (返回前 3 个匹配)
    fmt.Println("\n匹配前 3 个 'a' (n = 3):")
    matchesThree := reA.FindAllString(testString, 3)
    fmt.Printf("找到 %d 个匹配项: %v\n", len(matchesThree), matchesThree)
    // 预期输出: 找到 3 个匹配项: [a a a]

    // 情况 4: n = 10 (返回前 10 个匹配，但实际只有 8 个)
    fmt.Println("\n匹配前 10 个 'a' (n = 10，超过实际数量):")
    matchesTen := reA.FindAllString(testString, 10)
    fmt.Printf("找到 %d 个匹配项: %v\n", len(matchesTen), matchesTen)
    // 预期输出: 找到 8 个匹配项: [a a a a a a a a] (仍然返回所有 8 个)

    // --- 使用 FindAllStringSubmatch 示例 ---
    fmt.Println("\n--- 使用 FindAllStringSubmatch 示例 ---")

    // 编译一个更复杂的正则表达式，包含子匹配组
    // 匹配一个或多个 'a'，后面跟零个或多个 'b'
    reSubmatch, err := regexp.Compile("(a+)(b*)")
    if err != nil {
        fmt.Println("Error compiling regex reSubmatch:", err)
        return
    }

    testStringSub := "aaabbbaabcc"

    // 使用 n = -1 (返回所有匹配及其子匹配)
    fmt.Println("\nFindAllStringSubmatch (n = -1):")
    submatchesAll := reSubmatch.FindAllStringSubmatch(testStringSub, -1)
    for i, match := range submatchesAll {
        fmt.Printf("第 %d 个匹配: %v (完整匹配: %s, 子匹配1: %s, 子匹配2: %s)\n", i+1, match, match[0], match[1], match[2])
    }
    // 预期输出:
    // 第 1 个匹配: [aaabbb aaabbb bbb] (完整匹配: aaabbb, 子匹配1: aaabbb, 子匹配2: bbb)
    // 第 2 个匹配: [aa aa ] (完整匹配: aa, 子匹配1: aa, 子匹配2: )

    // 使用 n = 1 (返回第一个匹配及其子匹配)
    fmt.Println("\nFindAllStringSubmatch (n = 1):")
    submatchesOne := reSubmatch.FindAllStringSubmatch(testStringSub, 1)
    for i, match := range submatchesOne {
        fmt.Printf("第 %d 个匹配: %v (完整匹配: %s, 子匹配1: %s, 子匹配2: %s)\n", i+1, match, match[0], match[1], match[2])
    }
    // 预期输出:
    // 第 1 个匹配: [aaabbb aaabbb bbb] (完整匹配: aaabbb, 子匹配1: aaabbb, 子匹配2: bbb)
}