oj.uz

Submission #640180

# Submission time Handle Problem Language Result Execution time Memory
640180 2022-09-13T21:10:11 Z ojoConmigo Type Printer (IOI08_printer) C++17
90 / 100
 1000 ms 58956 KB 
#include <bits/stdc++.h>
using namespace std;

struct Trie{
    //struct Trie *child[26];
    map<char,Trie*> child;
    char letra;
    bool endOfWord,longest;
};

struct Trie *createNode(char letra){
    struct Trie *nodo = new Trie;
    /*
    for(int i=0; i<26; i++){
        nodo->child[i] = NULL;
    }
    */
    nodo->endOfWord = false;
    nodo->longest = false;
    nodo->letra = letra;
    return nodo;
}

void insert(struct Trie *root, string key, bool masLarga){
    struct Trie *curr = root;
    for(int i=0; i<(int)key.size(); i++){
        /*
        int index = key[i] - 'a';
        if(curr->child[index] == NULL){
            curr->child[index] = createNode(key[i]);
        }
        */
        if(curr->child.find(key[i]) == curr->child.end()){
            curr->child[key[i]] = createNode(key[i]);
        }
        //curr = curr->child[index];
        curr = curr->child[key[i]];
        if(masLarga) curr->longest = true;
    }
    curr->endOfWord = true;
}

int counting(struct Trie *root){
    struct Trie *curr = root;
    int cont = 2;
    if(curr->longest){
        cont--;
    }
    /*
    for(int i=0; i<26; i++){
        if(curr->child[i] != NULL){
            struct Trie *sig = curr->child[i];
            cont += counting(sig);
        }
    }
    */
    for(auto it = curr->child.begin(); it != curr->child.end(); it++){
        struct Trie *sig = it->second;
        cont += counting(sig);
    }
    return cont;
}

void printing(struct Trie *root){
    struct Trie *curr = root;
    //cout << curr->letra << endl;
    struct Trie *seguimos = NULL;
    /*
    for(int i=0; i<26; i++){
        if(curr->child[i] != NULL){
        //cout << sizeof(*(curr->child[i])) << " " << sizeof(Trie) << endl;
        //if(sizeof(*(curr->child[i])) == sizeof(Trie)){
            struct Trie *sig = curr->child[i];
            if(sig->longest){
                seguimos = sig;
            }else{
                cout << sig->letra << endl;
                printing(sig);
            }
        }
    }
    */
    for(auto it = curr->child.begin(); it != curr->child.end(); it++){
        struct Trie *sig = it->second;
        if(sig->longest){
            seguimos = sig;
        }else{
            cout << sig->letra << endl;
            printing(sig);
        }
    }

    if(curr->endOfWord){
        cout << 'P' << endl;
    }

    if(seguimos != NULL){
    //cout << sizeof(*seguimos) << " " << sizeof(Trie) << endl;
    //if(sizeof(*seguimos) == sizeof(Trie)){
        cout << seguimos->letra << endl;
        printing(seguimos);
        return;
    }

    if(!curr->longest) cout << '-' << endl;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n;
    cin >> n;
    struct Trie *root = createNode('0');
    root->longest = true;
    vector<string> v(n);
    pair<int,int> longest;
    longest.first = 1;
    longest.second = 0;
    for(int i=0; i<n; i++){
        string s;
        cin >> s;
        v[i] = s;
        if((int)s.size() > longest.first){
            longest.first = s.size();
            longest.second = i;
        }
    }

    for(int i=0; i<n; i++){
        if(i == longest.second){
            insert(root,v[i],true);
        }else insert(root,v[i],false);
    }
    //cout << sizeof(Trie) << endl;
    //cout << sizeof(*root) << endl;
    cout << counting(root)+n-1 << endl;
    printing(root);
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 340 KB Output is correct
2 Correct 10 ms 724 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 1108 KB Output is correct
2 Correct 22 ms 1456 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 71 ms 3580 KB Output is correct
2 Correct 140 ms 7436 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 161 ms 8772 KB Output is correct
2 Correct 47 ms 2508 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 427 ms 21704 KB Output is correct
2 Correct 914 ms 49616 KB Output is correct
3 Correct 494 ms 26316 KB Output is correct

Subtask #10
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 333 ms 17208 KB Output is correct
2 Execution timed out 1062 ms 58956 KB Time limit exceeded
3 Halted 0 ms 0 KB -