Добавление пользовательского типа в RedBlackTree - PullRequest
Новая
       69

Добавление пользовательского типа в RedBlackTree

1 голос
/ 14 апреля 2020

Я хочу сохранить упорядоченный набор записей, и стандарт предоставляет мне RedBlackTree. Запись имеет тип Tuple! (String, uint). Вот как это выглядит: 

import std.json : parseJSON;

    uint[string] wordTable;
    import std.datetime.stopwatch : StopWatch, AutoStart;

    auto sw = StopWatch(AutoStart.yes);
    const auto j = parseJSON(get(link));
    const long downloadTime = sw.peek.total!"msecs";

    import std.typecons : Tuple, tuple;
    import std.container.rbtree : RedBlackTree;
    import std.functional : binaryFun;

    RedBlackTree!(Tuple!(string, uint), binaryFun!("a[1] > b[1]")) records;

    foreach (node; j["posts"].array()) {
        import std.stdio : writeln;
        import std.utf : decode;

        if ("com" in node) {
            import std.algorithm : splitter;

            foreach (word; getStr(node["com"].str()).splitter(' ')) {
                import std.string : strip;

                if (word.strip().length > 0)
                    wordTable.require(word, 0)++;

                records ~= (tuple(word, wordTable[word])); // error
            }
        }
    }

Теперь я в основном использовал метод insert() для добавления записи в записи, но это вызывает segfault во время выполнения. Поэтому я решил использовать ~ = в надежде на лучшие сообщения об ошибках. И вот что говорит компилятор:

Ошибка: невозможно добавить тип Tuple! (String, uint) к типу std.container.rbtree.RedBlackTree! (Tuple! (String, uint), binaryFun, false )

Согласно https://dlang.org/phobos/std_container_rbtree.html# .RedBlackTree Я должен предоставить такой тип, что вызов less(a, b) для него возвращает логическое значение. Поэтому я продолжил и создал для него тип: 

struct Record {
    string key;
    uint value;

    int opCmp(ref const Record other) const {
        return value - other.value;
    }
}

// bool less(Record a, Record b) {
//  return a < b;
// }

void main(string[] args) {
    import std.stdio : writeln, writefln;

    if (args.length < 3) {
        writeln("Must have 2 arguments " ~ "first argument is the link, "
                ~ "the second one is for minimum repeatation threshold. Exiting.");
        import core.stdc.stdlib : exit;

        exit(-1);
    }

    const auto link = parseLink(args[1]);
    const auto threshold = atoui(args[2]);

    import std.json : parseJSON;

    uint[string] wordTable;
    import std.datetime.stopwatch : StopWatch, AutoStart;

    auto sw = StopWatch(AutoStart.yes);
    const auto j = parseJSON(get(link));
    const long downloadTime = sw.peek.total!"msecs";

    import std.container.rbtree : RedBlackTree;
    import std.functional : binaryFun;

    RedBlackTree!Record records;

    foreach (node; j["posts"].array()) {
        import std.utf : decode;

        if ("com" in node) {
            import std.algorithm : splitter;

            foreach (word; getStr(node["com"].str()).splitter(' ')) {
                import std.string : strip;

                if (word.strip().length > 0)
                    wordTable.require(word, 0)++;

                records ~= (Record(word, wordTable[word]));
            }
        }
    }

На этот раз компилятор жалуется:

Ошибка: невозможно добавить тип записи в тип std.container.rbtree.RedBlackTree ! (Запись, "a

Итак, суть вопроса в том, что если у меня RedBlackTree с пользовательским binaryFun, как я могу добавить экземпляр кортежа? или пользовательский тип к нему?

...