Submission #739895

#	Time	Username	Problem	Language	Result	Execution time	Memory
739895		Desh03	Type Printer (IOI08_printer)	C++17	30 / 100	186 ms	83260 KiB

printer

#include <bits/stdc++.h>
using namespace std;

struct node {
    node* c[26] = {NULL};
    bool endofword;
    int sz;
    node(bool x) : endofword(x) { };
};

vector<char> ans;

struct trie {
    node* root;
    trie() : root(NULL) { };
    void insert(node* &u, int id, string &s) {
        if (u == NULL) u = new node(false);
        if (id == s.size()) {
            u->endofword = true;
            return;
        }
        insert(u->c[s[id] - 'a'], id + 1, s);
    }
    void dfs1(node* &u) {
        u->sz = 1;
        for (int i = 0; i < 26; i++)
            if (u->c[i] != NULL)
                dfs1(u->c[i]), u->sz += u->c[i]->sz;
    }
    void dfs2(node* u) {
        if (u->endofword) ans.push_back('P');
        vector<int> f(26);
        for (int i = 0; i < 26; i++) f[i] = i;
        sort(f.begin(), f.end(), [&](int a, int b) {
            int x = u->c[a] == NULL ? 0 : u->c[a]->sz, y = u->c[b] == NULL ? 0 : u->c[b]->sz;
            return x < y;
        });
        for (int i : f)
            if (u->c[i] != NULL) {
                ans.push_back('a' + i);
                dfs2(u->c[i]);
            }
        ans.push_back('-');
    }
    void insert(string s) {
        insert(root, 0, s);
    }
};

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    int n;
    cin >> n;
    trie tr;
    for (int i = 0; i < n; i++) {
        string s;
        cin >> s;
        tr.insert(s);
    }
    tr.dfs1(tr.root);
    tr.dfs2(tr.root);
    while (ans.back() == '-') ans.pop_back();
    cout << ans.size() << '\n';
    for (char c : ans) cout << c << '\n';
}

Compilation message (stderr)

printer.cpp: In member function 'void trie::insert(node*&, int, std::string&)':
printer.cpp:18:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   18 |         if (id == s.size()) {
      |             ~~~^~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 10 / 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