Submission #838537

#	Time	Username	Problem	Language	Result	Execution time	Memory
838537		VMaksimoski008	Type Printer (IOI08_printer)	C++14	20 / 100	45 ms	21348 KiB

printer

#include <bits/stdc++.h>

#define pb push_back
#define eb emplace_back
#define sz(x) (int)x.size()
#define all(x) x.begin(), x.end()
#define uniq(x) x.erase(unique(all(x)), x.end())
#define rall(x) x.rbegin(), x.rend()
//#define int long long

using namespace std;

using ll = long long;
using ull = unsigned long long;
using ld = long double;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

const int mod = 1e9 + 7;
const int LOG = 20;
const int maxn = 1e5 + 5;
const double eps = 1e-9;

void setIO() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
}

struct Node {
    bool isWord;
    int words;
    map<char, Node*> child;

    Node() : isWord(false), words(0) {}
};

bool cmp(pair<Node*, char> a, pair<Node*, char> b) {
    if(a.first->words == b.first->words)
        return a.second < b.second;
    return a.first->words < b.first->words;
}

struct Trie {
    Node* root;
    int totalWords;
    string dfsAns;
    int totalOp;
    int foundSoFar;

    Trie() {
        root = new Node();
        root->isWord = false;
        totalWords = 0;
        dfsAns = "";
        totalOp = 0;
    }

    ~Trie() {
        delete root;
    }

    void insert(string &word) {
        Node* curr = root;
        for(char &ch : word) {
            if(!curr->child.count(ch))
                curr->child[ch] = new Node();
            curr = curr->child[ch];
            curr->words++;
        }

        curr->isWord = true;
        totalWords++;
    }

    bool search(string &word) {
        Node* curr = root;
        for(char &ch : word) {
            if(!curr->child.count(ch)) return false;
            curr = curr->child[ch];
        }

        return curr->isWord;
    }

    void dfs(Node* curr, char ch) {
        if(ch != ' ') dfsAns += ch, totalOp++;
        if(curr->isWord == true) dfsAns += 'P', totalOp++, foundSoFar++;

        vector<pair<Node*, char> > nextVec;
        for(auto &next : curr->child)
            nextVec.push_back({ next.second, next.first });
        sort(all(nextVec), cmp);

        for(auto &next : nextVec)
            dfs(next.first, next.second);

        if(ch != ' ' && foundSoFar < totalWords) dfsAns += '-', totalOp++;
    }

    void callDFS() {
        foundSoFar = 0;
        dfs(root, ' ');
        cout << totalOp << '\n';
        for(char &ch : dfsAns)
            cout << ch << '\n';
    }
};

int32_t main() {
    setIO();

    int n;
    cin >> n;
    Trie trie;

    for(int i=0; i<n; i++) {
        string s;
        cin >> s;
        trie.insert(s);
    }

    trie.callDFS();
    return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 0 / 10
• Subtask 3: 0 / 10
• Subtask 4: 10 / 10
• Subtask 5: 0 / 10
• Subtask 6: 0 / 10
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 10
• Subtask 10: 0 / 10