Submission #615577

#	Time	Username	Problem	Language	Result	Execution time	Memory
615577		Bungmint	Type Printer (IOI08_printer)	C++17	100 / 100	162 ms	63480 KiB

printer

// Copyright © 2022 Youngmin Park. All rights reserved.
//#pragma GCC optimize("O3")
//#pragma GCC target("avx2")
#include <bits/stdc++.h>
using namespace std;

#pragma region TEMPLATE

using ll = long long;
using vi = vector<int>;
using pii = pair<int, int>;
using vpi = vector<pii>;
using pll = pair<ll, ll>;
using vl = vector<ll>;
using vpl = vector<pll>;
using ld = long double;
template <typename T, size_t SZ>
using ar = array<T, SZ>;
template <typename T>
using pqg = priority_queue<T, vector<T>, greater<T>>;

#define all(v) (v).begin(), (v).end()
#define pb push_back
#define sz(x) (int)(x).size()
#define fi first
#define se second
#define lb lower_bound
#define ub upper_bound

constexpr int INF = 1e9;
constexpr ll LINF = 1e18;
const ld PI = acos((ld)-1.0);
constexpr int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
template <typename T>
constexpr bool ckmin(T &a, const T &b) { return b < a ? a = b, 1 : 0; }
template <typename T>
constexpr bool ckmax(T &a, const T &b) { return b > a ? a = b, 1 : 0; }
ll cdiv(ll a, ll b) { return a / b + ((a ^ b) > 0 && a % b); } // divide a by b rounded up
ll fdiv(ll a, ll b) { return a / b - ((a ^ b) < 0 && a % b); } // divide a by b rounded down

#ifdef LOCAL
#include "miscellaneous/debug.h"
#else
#define dbg(...) 42
#endif

inline namespace RecursiveLambda {
	template <typename Fun>
	struct y_combinator_result {
		Fun fun_;
		template <typename T>
		explicit constexpr y_combinator_result(T &&fun) : fun_(forward<T>(fun)) {}
		template <typename... Args>
		constexpr decltype(auto) operator()(Args &&...args) const {
			return fun_(ref(*this), forward<Args>(args)...);
		}
	};
	template <typename Fun>
	decltype(auto) y_combinator(Fun &&fun) {
		return y_combinator_result<decay_t<Fun>>(forward<Fun>(fun));
	}
};

#pragma endregion TEMPLATE

// TODO: Test imp
/**
 * Description: Trie supporting string insertion
 * Source: Folklore, Benq
 * Verification: https://oj.uz/problem/view/IOI08_printer
 * Time complexity: O(|S_i|) per insertion
 */

template <int SZ, int ALPHA>
struct TrieAlpha {
    int trie[SZ][ALPHA], sz[SZ], cnt_end[SZ], f[SZ], g[SZ], nxt_g[SZ], num;
    string ret;
	TrieAlpha() { memset(trie, 0, sizeof(trie)), memset(sz, 0, sizeof(sz)); }
    void insert(const string& s, int cnt = 1) {
        int now = 0; sz[now] += cnt;
        for (int i = 0; i < sz(s); i++) {
            int c = s[i] - 'a';
            if (!trie[now][c]) trie[now][c] = ++num;
            now = trie[now][c];
            sz[now] += cnt;
        }
		cnt_end[now]++;
    }
	void dfs(int u = 0) {
		int k = 0;
		int sum_f{}, min_gf{INF};
		for (int i = 0; i < ALPHA; i++) {
			int v = trie[u][i];
			if (!v) continue;
			dfs(v);
			k++;
			sum_f += f[v];
			if (ckmin(min_gf, g[v]-f[v])) {
				nxt_g[u] = i;
			}
		}
		if (k == 0) {
			f[u] = g[u] = 0;
			return;
		}
		f[u] = 2*k + sum_f;
		g[u] = 2*k-1 + sum_f + min_gf;
		dbg(f[u], g[u]);
	}
	void calc_string(int u = 0, bool is_g = 1) {
		dbg(cnt_end[u], sz[u]);
		for (int i = 0; i < cnt_end[u]; i++) ret += 'P';
		for (int i = 0; i < ALPHA; i++) {
			int v = trie[u][i];
			char c = 'a' + i;
			if (!v) continue;
			else if (is_g && nxt_g[u] == i) {
				continue;
			}else{
				ret += c;
				calc_string(v, 0);
				ret += '-';
			}
		}
		if (is_g && trie[u][nxt_g[u]]) {
			ret += (char)('a' + nxt_g[u]);
			calc_string(trie[u][nxt_g[u]], 1);
		}
	}
};
TrieAlpha<20*25100, 26> trie;

void solve() {
	int n; cin >> n;
	for (int i = 0; i < n; i++) {
		string s; cin >> s;
		trie.insert(s);
	}
	trie.dfs();
	trie.calc_string();
	cout << sz(trie.ret) << '\n';
	for (auto &ch : trie.ret) cout << ch << '\n';
}

int main() {
	cin.tie(0)->sync_with_stdio(0);
	cin.exceptions(cin.failbit);
	int testcase = 1;
	// cin >> testcase;
	while (testcase--) {
		solve();
	}
#ifdef LOCAL
	cerr << "Time elapsed: " << 1.0 * (double)clock() / CLOCKS_PER_SEC << " s.\n";
#endif
}

Compilation message (stderr)

printer.cpp:7: warning: ignoring '#pragma region TEMPLATE' [-Wunknown-pragmas]
    7 | #pragma region TEMPLATE
      | 
printer.cpp:65: warning: ignoring '#pragma endregion TEMPLATE' [-Wunknown-pragmas]
   65 | #pragma endregion TEMPLATE
      | 
printer.cpp: In instantiation of 'void TrieAlpha<SZ, ALPHA>::dfs(int) [with int SZ = 502000; int ALPHA = 26]':
printer.cpp:140:11:   required from here
printer.cpp:45:18: warning: statement has no effect [-Wunused-value]
   45 | #define dbg(...) 42
      |                  ^~
printer.cpp:109:3: note: in expansion of macro 'dbg'
  109 |   dbg(f[u], g[u]);
      |   ^~~
printer.cpp: In instantiation of 'void TrieAlpha<SZ, ALPHA>::calc_string(int, bool) [with int SZ = 502000; int ALPHA = 26]':
printer.cpp:141:19:   required from here
printer.cpp:45:18: warning: statement has no effect [-Wunused-value]
   45 | #define dbg(...) 42
      |                  ^~
printer.cpp:112:3: note: in expansion of macro 'dbg'
  112 |   dbg(cnt_end[u], sz[u]);
      |   ^~~

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