Итерация по 2 строкам

Question

Я хочу сравнить две строки за руной, чтобы увидеть, какая из них идет первой в произвольном алфавитном порядке.

Прямо сейчас у меня есть эта реализация, которая хранит в map[rune]int отображение, представляющее порядок букв в моем алфавите.

У меня есть этот рабочий код. Я хорошо осведомлен о недостатках существующего дизайна, но это не главное.

package main

import (
    "bufio"
    "log"
    "math/rand"
    "os"
    "sort"
)

type Dictionnary struct {
    content           []string
    alphaBeticalOrder map[rune]int
}

func minSize(w1, w2 []rune) int {
    if len(w1) < len(w2) {
        return len(w1)
    }
    return len(w2)
}

func (d *Dictionnary) comesFirst(a, b rune) int {

    return d.alphaBeticalOrder[a] - d.alphaBeticalOrder[b]
}

func (d Dictionnary) Less(i, j int) bool {
    wordi, wordj := []rune(d.content[i]), []rune(d.content[j])
    size := minSize(wordi, wordj)
    for index := 0; index < size; index++ {
        diff := d.comesFirst(wordi[index], wordj[index])
        switch {
        case diff < 0:
            return true
        case diff > 0:
            return false
        default:
            continue
        }
    }
    return len(wordi) < len(wordj)
}

func (d Dictionnary) Swap(i, j int) {
    d.content[i], d.content[j] = d.content[j], d.content[i]
}

func (d Dictionnary) Len() int {
    return len(d.content)
}

func main() {

    letters := []rune{'z', 'y', 'x', 'w', 'v', 'u', 't', 's', 'r', 'q', 'p', 'o', 'n', 'm', 'l', 'k', 'j', 'i', 'h', 'g', 'f', 'e', 'd', 'c', 'b', 'a'}
    aOrder := make(map[rune]int)
    perm := rand.Perm(len(letters))
    for i, v := range perm {
        aOrder[letters[i]] = v
    }

    file, err := os.Open("testdata/corpus.txt")
    if err != nil {
        log.Fatal(err)
    }

    corpus := make([]string, 0, 1000)
    scanner := bufio.NewScanner(file)
    for scanner.Scan() {
        corpus = append(corpus, scanner.Text())
    }

    if err := scanner.Err(); err != nil {
        log.Fatal(err)
    }
    file.Close()

    input := Dictionnary{content: corpus, alphaBeticalOrder: aOrder}

    sort.Sort(input)

    ofile, err := os.Create("testdata/sorted.txt")
    writer := bufio.NewWriter(ofile)
    for _, v := range input.content {
        writer.WriteString(v)
        writer.WriteString("\n")
    }
    writer.Flush()
    defer ofile.Close()
}

Мой вопрос касается Less(i,j int) bool функция. Есть ли более идиоматический способ перебрать 2 строки, чтобы сравнить их руна за руной? Я делаю копию данных здесь, которых можно было бы избежать.

РЕДАКТИРОВАТЬ: Чтобы прояснить мою проблему заключается в том, что диапазон (строка) может позволить вам перебирать строки руны за руной, но я не вижу способа перебирать 2 строки рядом друг с другом. Единственный способ, которым я вижу это, чтобы преобразовать строки в [] руну.

Answer 1

Перебирать две строки рядом в Less метод:

package main

import (
    "bufio"
    "log"
    "math/rand"
    "os"
    "sort"
    "unicode/utf8"
)

type Dictionary struct {
    content           []string
    alphaBeticalOrder map[rune]int
}

func (d Dictionary) Len() int {
    return len(d.content)
}

func (d Dictionary) Swap(i, j int) {
    d.content[i], d.content[j] = d.content[j], d.content[i]
}

func (d Dictionary) Less(i, j int) bool {
    wi, wj := d.content[i], d.content[j]
    jj := 0
    for _, ri := range wi {
        rj, size := utf8.DecodeRuneInString(wj[jj:])
        if rj == utf8.RuneError && size == 0 {
            return false
        }
        switch ao := d.alphaBeticalOrder[ri] - d.alphaBeticalOrder[rj]; {
        case ao < 0:
            return true
        case ao > 0:
            return false
        }
        jj += size
    }
    return len(wi) < len(wj)
}

func main() {

    letters := []rune{'z', 'y', 'x', 'w', 'v', 'u', 't', 's', 'r', 'q', 'p', 'o', 'n', 'm', 'l', 'k', 'j', 'i', 'h', 'g', 'f', 'e', 'd', 'c', 'b', 'a'}
    aOrder := make(map[rune]int)
    perm := rand.Perm(len(letters))
    for i, v := range perm {
        aOrder[letters[i]] = v
    }

    file, err := os.Open("testdata/corpus.txt")
    if err != nil {
        log.Fatal(err)
    }

    corpus := make([]string, 0, 1000)
    scanner := bufio.NewScanner(file)
    for scanner.Scan() {
        corpus = append(corpus, scanner.Text())
    }

    if err := scanner.Err(); err != nil {
        log.Fatal(err)
    }
    file.Close()

    input := Dictionary{content: corpus, alphaBeticalOrder: aOrder}

    sort.Sort(input)

    ofile, err := os.Create("testdata/sorted.txt")
    writer := bufio.NewWriter(ofile)
    for _, v := range input.content {
        writer.WriteString(v)
        writer.WriteString("\n")
    }
    writer.Flush()
    defer ofile.Close()
}

Answer 2

Вы можете сделать цикл немного более идиоматическим, используя диапазон для одного из двух слов. Это требует добавления проверки в цикле, но вам больше не нужно выполнять проверку при окончательном возврате.

// determines if the word indexed at i is less than the word indexed at j.
func (d Dictionnary) Less(i, j int) bool {
    wordi, wordj := []rune(d.content[i]), []rune(d.content[j])
    for i, c := range wordi {
        if i == len(wordj) {
            return false
        }

        diff := d.comesFirst(c, wordj[i])
        switch {
        case diff < 0:
            return true
        case diff > 0:
            return false
        default:
            continue
        }
    }
    return false
}