SimSearcher/GramSet.cpp at master · LittleJiafang/SimSearcher · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#include <cstdio>
#include <cstring>

#include "Const.h"
#include "GramSet.h"

int GramSet::Z = 0;
int * GramSet::asciiId = NULL;

GramSet::GramSet() {
    root = new GramSetNode();
}

GramSet::~GramSet() {
    clear(root);
}

void GramSet::clear(GramSetNode * cur) {
    if (cur == NULL) {
        return;
    }
    for (int i = 0; i < Z; i ++) {
        GramSetNode * son = cur -> getSonById(i);
        if (son != NULL) {
            clear(son);
        }
    }
    delete cur;
}

void GramSet::printVisitor(GramSetNode * cur, char * gram, int gramLen) {
    std::vector < int > * lineSet = cur -> getLineSet();
    if (lineSet != NULL) {
        int n = lineSet -> size();
        printf("%s:{", gram);
        for (int i = 0; i < n; i ++) {
            printf("%d,", lineSet -> operator [] (i));
        }
        printf("}\n");
    }
    for (int ch = 0; ch < MAX_ASCII; ch ++) {
        int i = asciiId[ch];
        if (i != -1) {
            GramSetNode * son = cur -> getSonById(i);
            if (son != NULL) {
                gram[gramLen] = ch;
                printVisitor(son, gram, gramLen + 1);
            }
        }
    }
    gram[gramLen] = 0;
}

void GramSet::insertGram(const std::string & gram, int lineId) {
    GramSetNode * cur = root;
    for (int i = 0; i < (int) gram.length(); i ++) {
        cur = cur -> getOrCreateSonByAscii(gram[i]);
    }
    std::vector < int > * lineSet = cur -> getOrCreateLineSet();
    if (lineSet -> size() == 0 || lineSet -> back() != lineId) {
        lineSet -> push_back(lineId);
    }
}

std::vector < int > * GramSet::getLineSet(const std::string & gram) {
    GramSetNode * cur = root;
    for (int i = 0; i < (int) gram.length(); i ++) {
        cur = cur -> getSonByAscii(gram[i]);
        if (cur == NULL) {
            return NULL;
        }
    }
    return cur -> getLineSet();
}

void GramSet::print() {
    static char gram[MAX_LINE_LENGTH];
    memset(gram, 0, sizeof(gram));
    printVisitor(root, gram, 0);

}

void GramSet::init(int Z_, int * asciiId_) {
    Z = Z_;
    asciiId = asciiId_;
    GramSetNode::init(Z_, asciiId_);
}