Pagini recente » Cod sursa (job #1967577) | Cod sursa (job #797912) | Cod sursa (job #569123) | Cod sursa (job #2726559) | Cod sursa (job #1005263)
#include <iostream>
#include <fstream>
#include <cstdio>
#include <sstream>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdlib>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <utility>
#include <string>
#include <vector>
using namespace std;
typedef vector<int> vi;
typedef pair<int, int> ii;
typedef vector<ii> vii;
typedef vector<vi> vvi;
typedef vector<string> vs;
typedef map<string, int> msi;
typedef long long ll;
typedef unsigned long long ull;
typedef vector<int>::iterator vit;
typedef vector<ii>::iterator viit;
typedef vector<string>::iterator vst;
#define REP(i, n) for (int i = 0; i < n; ++i)
#define FOR(i, a, b) for (int i = a; i < b; ++i)
#define FORD(i, a, b) for (int i = a-1; i >= b; --i)
#define MP make_pair
#define PB push_back
#define ALL(x) (x).begin(), x.end()
#define SIZE(x) (int)(x).size()
#define FOREACH(it, c) for (__typeof(c.begin()) it = c.begin(); it != c.end(); ++it)
#define INF 1023456789
#define DEBUG(x) cerr << #x << ": " << x << endl;
#define ERR(...) fprintf(stderr, __VA_ARGS__);
ifstream f("trie.in");
ofstream g("trie.out");
class TrieNode {
TrieNode* children[30];
public:
int count;
int nchildren;
TrieNode() {
count = 0;
nchildren = 0;
for (int i = 0; i < 30; i++)
children[i] = NULL;
}
TrieNode* getChild(char c) {
return this->children[c - 'a'];
}
void setChild(char c, TrieNode* node) {
this->children[c - 'a'] = node;
if (node == NULL) {
nchildren--;
} else {
nchildren++;
}
}
int nChildren() {
return nchildren;
}
};
class Trie {
TrieNode* root;
public:
Trie() {
root = new TrieNode();
}
void add(string s) {
TrieNode* node = root;
for (int i = 0; i < s.length(); i++) {
TrieNode *nxt = node->getChild(s[i]);
if (nxt != NULL) {
node = nxt;
if (i == s.length() - 1)
node->count++;
} else {
nxt = new TrieNode();
node->setChild(s[i], nxt);
node = nxt;
if (i == s.length() - 1)
node->count++;
}
}
}
int get(string s) {
TrieNode* node = root;
for (int i = 0; i < s.length(); i++) {
TrieNode *nxt = node->getChild(s[i]);
if (nxt != NULL) {
node = nxt;
if (i == s.length() - 1)
return node->count;
} else {
break;
}
}
return 0;
}
int lpm(string s) {
TrieNode* node = root;
int i = 0;
for (; i < s.length(); i++) {
TrieNode *nxt = node->getChild(s[i]);
if (nxt == NULL)
break;
node = nxt;
}
return i;
}
void del(string s) {
del_rec(s, root);
}
TrieNode* del_rec(string s, TrieNode* node) {
if (s.size() == 0) {
node->count--;
if (node->count == 0 && node->nChildren() == 0) {
return NULL;
}
return node;
} else {
TrieNode* nxt = node->getChild(s[0]);
if (nxt == NULL) {
node->count--;
return node;
}
node->setChild(s[0], del_rec(s.substr(1), nxt));
return node;
}
}
};
int main() {
int op;
string w;
Trie* trie = new Trie();
while (!f.eof()) {
f >> op >> w;
if (op == 0) {
trie->add(w);
} else if (op == 1) {
trie->del(w);
} else if (op == 2) {
g << trie->get(w) << endl;
} else if (op == 3) {
g << trie->lpm(w) << endl;
}
}
}